idk

Author: hfhoffman1144

python-langgraph/chains/binary_questions.py

@@ -26,5 +26,6 @@ class BinaryAnswer(BaseModel):
 binary_question_model = ChatOpenAI(model="gpt-4o-mini", temperature=0)
 
 BINARY_QUESTION_CHAIN = (
-    binary_question_prompt | binary_question_model.with_structured_output(BinaryAnswer)
+    binary_question_prompt
+    | binary_question_model.with_structured_output(BinaryAnswer)
 )

python-langgraph/chains/escalation_check.py

@@ -28,5 +28,6 @@ class EscalationCheck(BaseModel):
 escalation_check_model = ChatOpenAI(model="gpt-4o-mini", temperature=0)
 
 ESCALATION_CHECK_CHAIN = (
-    escalation_prompt | escalation_check_model.with_structured_output(EscalationCheck)
+    escalation_prompt
+    | escalation_check_model.with_structured_output(EscalationCheck)
 )

python-langgraph/chains/notice_extraction.py

@@ -65,7 +65,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
@@ -74,7 +76,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
@@ -100,6 +104,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

@@ -27,7 +27,9 @@ def forward_email(email_message: str, send_to_email: str) -> bool:
 
 
 @tool
-def send_wrong_email_notification_to_sender(sender_email: str, correct_department: str):
+def send_wrong_email_notification_to_sender(
+    sender_email: str, correct_department: str
+):
     """
     Send an email back to the sender informing them that
     they have the wrong address. The email should be sent
@@ -42,7 +44,9 @@ def send_wrong_email_notification_to_sender(sender_email: str, correct_departmen
 
 
 @tool
-def extract_notice_data(email: str, escalation_criteria: str) -> NoticeEmailExtract:
+def extract_notice_data(
+    email: str, escalation_criteria: str
+) -> NoticeEmailExtract:
     """
     Extract structured fields from a regulatory notice.
     This should be used when the email message comes from
@@ -106,7 +110,9 @@ def determine_email_action(email: str) -> str:
 ]
 tool_node = ToolNode(tools)
 
-EMAIL_AGENT_MODEL = ChatOpenAI(model="gpt-4o-mini", temperature=0).bind_tools(tools)
+EMAIL_AGENT_MODEL = ChatOpenAI(model="gpt-4o-mini", temperature=0).bind_tools(
+    tools
+)
 
 
 def call_agent_model_node(state: MessagesState) -> dict[str, list[AIMessage]]:

python-langgraph/utils/graph_utils.py

@@ -33,7 +33,9 @@ def create_legal_ticket(
     ]
 
     if current_follow_ups:
-        follow_ups = [f for f in follow_ups if f not in current_follow_ups.keys()]
+        follow_ups = [
+            f for f in follow_ups if f not in current_follow_ups.keys()
+        ]
 
     follow_up = random.choice(follow_ups)