tech review changes

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

@@ -35,7 +35,7 @@ class NoticeEmailExtract(BaseModel):
     site_location: str | None = Field(
         default=None,
         description="""The site location of the project (if present
-        in the message)""",
+        in the message). Use the full address if possible.""",
     )
     violation_type: str | None = Field(
         default=None,

python-langgraph/example_emails.py

@@ -1,11 +1,39 @@
 EMAILS = [
     # Email 1
     """
+    Date: October 15, 2024
+    From: Occupational Safety and Health Administration (OSHA)
+    To: Blue Ridge Construction, project 111232345 - Downtown Office Complex
+    Location: Dallas, TX
+
+    During a recent inspection of your construction site at 123 Main Street,
+    the following safety violations were identified:
+
+    Lack of fall protection: Workers on scaffolding above 10 feet were
+    without required harnesses or other fall protection equipment.
+    Unsafe scaffolding setup: Several scaffolding structures were noted as
+    lacking secure base plates and bracing, creating potential collapse risks.
+    Inadequate personal protective equipment (PPE): Multiple workers were found
+    without proper PPE, including hard hats and safety glasses.
+    Required Corrective Actions:
+
+    Install guardrails and fall arrest systems on all scaffolding over 10 feet.
+    Conduct an inspection of all scaffolding structures and reinforce unstable
+    sections. Ensure all workers on-site are provided with necessary PPE and conduct
+    safety training on proper usage.
+    Deadline for Compliance: All violations must be rectified by November 10, 2024.
+    Failure to comply may result in fines of up to $25,000 per violation.
+
+    Contact: For questions or to confirm compliance, please reach out to the OSHA
+    regional office at (555) 123-4567 or email compliance.osha@osha.gov.
+    """,
+    # Email 2
+    """
     From: debby@stack.com
     Hey Betsy,
     Here's your invoice for $1000 for the cookies you ordered.
     """,
-    # Email 2
+    # Email 3
     """
     From: tdavid@companyxyz.com
     Hi Paul,
@@ -14,7 +42,7 @@
     Thanks,
     Terrance
     """,
-    # Email 3
+    # Email 4
     """
     Date: January 10, 2025
     From: City of Los Angeles Building and Safety Department

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/graphs/notice_extraction.py

@@ -89,17 +89,14 @@ def answer_follow_up_question_node(state: GraphState) -> GraphState:
     BINARY_QUESTION_CHAIN"""
 
     if state["current_follow_up"]:
-
         question = state["current_follow_up"] + " " + state["notice_message"]
 
         answer = BINARY_QUESTION_CHAIN.invoke({"question": question})
 
         if state.get("follow_ups"):
-
             state["follow_ups"][state["current_follow_up"]] = answer
 
         else:
-
             state["follow_ups"] = {state["current_follow_up"]: answer}
 
     return state
@@ -122,7 +119,6 @@ def route_follow_up_edge(state: GraphState) -> str:
     """Determine whether a follow-up question is required"""
 
     if state.get("current_follow_up"):
-
         return "answer_follow_up_question"
 
     return END

python-langgraph/utils/graph_utils.py

@@ -32,15 +32,11 @@ 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)
 
     if not follow_up:
-
         LOGGER.info("Legal ticket created!")
         return follow_up