format

Author: hfhoffman1144

python-langgraph/chains/notice_extraction.py

@@ -1,7 +1,7 @@
-from datetime import datetime, date
+from datetime import date, datetime
 from langchain_core.prompts import ChatPromptTemplate
 from langchain_openai import ChatOpenAI
-from pydantic import BaseModel, Field, computed_field
+from pydantic import BaseModel, computed_field, Field
 
 
 class NoticeEmailExtract(BaseModel):
@@ -64,7 +64,9 @@ class NoticeEmailExtract(BaseModel):
     @property
     def date_of_notice(self) -> date | None:
         try:
-            return datetime.strptime(self.date_of_notice_str, "%Y-%m-%d").date()
+            return datetime.strptime(
+                self.date_of_notice_str, "%Y-%m-%d"
+            ).date()
         except Exception as e:
             print(e)
             return None
@@ -73,7 +75,9 @@ def date_of_notice(self) -> date | None:
     @property
     def compliance_deadline(self) -> date | None:
         try:
-            return datetime.strptime(self.compliance_deadline_str, "%Y-%m-%d").date()
+            return datetime.strptime(
+                self.compliance_deadline_str, "%Y-%m-%d"
+            ).date()
         except Exception as e:
             print(e)
             return None
@@ -99,6 +103,7 @@ def compliance_deadline(self) -> date | None:
 
 notice_parser_model = ChatOpenAI(model="gpt-4o-mini", temperature=0)
 
-NOTICE_PARSER_CHAIN = info_parse_prompt | notice_parser_model.with_structured_output(
-    NoticeEmailExtract
+NOTICE_PARSER_CHAIN = (
+    info_parse_prompt
+    | notice_parser_model.with_structured_output(NoticeEmailExtract)
 )

python-langgraph/graphs/email_agent.py

@@ -1,11 +1,11 @@
-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
+import time
 from utils.logging_config import LOGGER
 
 

python-langgraph/graphs/notice_extraction.py

@@ -1,9 +1,9 @@
-from typing import TypedDict
 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 typing import TypedDict
 from utils.graph_utils import create_legal_ticket, send_escalation_email
 from utils.logging_config import LOGGER
 

python-langgraph/utils/graph_utils.py

@@ -1,7 +1,7 @@
+from chains.notice_extraction import NoticeEmailExtract
+from pydantic import EmailStr
 import random
 import time
-from pydantic import EmailStr
-from chains.notice_extraction import NoticeEmailExtract
 from utils.logging_config import LOGGER