From 1db655fd12c27d52a37f16ca6524afebc5ee6cd2 Mon Sep 17 00:00:00 2001 From: martin-martin Date: Mon, 3 Feb 2025 14:58:38 +0100 Subject: [PATCH] =?UTF-8?q?Ruff=20rules=20over=20isort=20=F0=9F=98=85?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- python-langgraph/graphs/email_agent.py | 5 ++--- python-langgraph/graphs/notice_extraction.py | 5 ++--- python-langgraph/utils/graph_utils.py | 3 +-- 3 files changed, 5 insertions(+), 8 deletions(-) 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"