this is insane

Author: hfhoffman1144

python-langgraph/chains/binary_questions.py

@@ -26,6 +26,5 @@ 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,6 +28,5 @@ 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

@@ -64,9 +64,7 @@ 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
@@ -75,9 +73,7 @@ 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
@@ -103,7 +99,6 @@ 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

@@ -26,9 +26,7 @@ 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
@@ -43,9 +41,7 @@ def send_wrong_email_notification_to_sender(
 
 
 @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
@@ -109,9 +105,7 @@ 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

@@ -32,9 +32,7 @@ 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)