diff --git a/python-langgraph/graphs/email_agent.py b/python-langgraph/graphs/email_agent.py index 65c50fa38d..bdf8e01c48 100644 --- a/python-langgraph/graphs/email_agent.py +++ b/python-langgraph/graphs/email_agent.py @@ -1,15 +1,14 @@ import logging import time +from chains.notice_extraction import NoticeEmailExtract +from graphs.notice_extraction import NOTICE_EXTRACTION_GRAPH from langchain_core.messages import AIMessage from langchain_core.tools import tool from langchain_openai import ChatOpenAI from langgraph.graph import END, START, MessagesState, StateGraph from langgraph.prebuilt import ToolNode -from chains.notice_extraction import NoticeEmailExtract -from graphs.notice_extraction import NOTICE_EXTRACTION_GRAPH - logging.basicConfig( level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s" ) diff --git a/python-langgraph/graphs/notice_extraction.py b/python-langgraph/graphs/notice_extraction.py index ead4df20c1..ebacf62a3f 100644 --- a/python-langgraph/graphs/notice_extraction.py +++ b/python-langgraph/graphs/notice_extraction.py @@ -1,12 +1,11 @@ import logging from typing import TypedDict -from langgraph.graph import END, START, StateGraph -from pydantic import EmailStr - from chains.binary_questions import BINARY_QUESTION_CHAIN from chains.escalation_check import ESCALATION_CHECK_CHAIN from chains.notice_extraction import NOTICE_PARSER_CHAIN, NoticeEmailExtract +from langgraph.graph import END, START, StateGraph +from pydantic import EmailStr from utils.graph_utils import create_legal_ticket, send_escalation_email logging.basicConfig( diff --git a/python-langgraph/utils/graph_utils.py b/python-langgraph/utils/graph_utils.py index 8a4e37c6e5..ed16abfbbc 100644 --- a/python-langgraph/utils/graph_utils.py +++ b/python-langgraph/utils/graph_utils.py @@ -2,9 +2,8 @@ import random import time -from pydantic import EmailStr - from chains.notice_extraction import NoticeEmailExtract +from pydantic import EmailStr logging.basicConfig( level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s"