tech review updates

Author: hfhoffman1144

python-langgraph/chains/notice_extraction.py

@@ -1,13 +1,14 @@
-from datetime import date
-
+from datetime import datetime, date
 from langchain_core.prompts import ChatPromptTemplate
 from langchain_openai import ChatOpenAI
-from pydantic import BaseModel, EmailStr, Field
+from pydantic import BaseModel, Field, computed_field
 
 
 class NoticeEmailExtract(BaseModel):
-    date_of_notice: date | None = Field(
+    date_of_notice_str: str | None = Field(
         default=None,
+        exclude=True,
+        repr=False,
         description="""The date of the notice (if any) reformatted
         to match YYYY-mm-dd""",
     )
@@ -21,7 +22,7 @@ class NoticeEmailExtract(BaseModel):
         description="""The phone number of the entity sending the notice
         (if present in the message)""",
     )
-    entity_email: EmailStr | None = Field(
+    entity_email: str | None = Field(
         default=None,
         description="""The email of the entity sending the notice
         (if present in the message)""",
@@ -46,8 +47,10 @@ class NoticeEmailExtract(BaseModel):
         description="""The required changes specified by the entity
         (if present in the message)""",
     )
-    compliance_deadline: date | None = Field(
+    compliance_deadline_str: str | None = Field(
         default=None,
+        exclude=True,
+        repr=False,
         description="""The date that the company must comply (if any)
         reformatted to match YYYY-mm-dd""",
     )
@@ -57,6 +60,28 @@ class NoticeEmailExtract(BaseModel):
         (if any)""",
     )
 
+    @computed_field
+    @property
+    def date_of_notice(self) -> date | None:
+        try:
+            return datetime.strptime(
+                self.date_of_notice_str, "%Y-%m-%d"
+            ).date()
+        except Exception as e:
+            print(e)
+            return None
+
+    @computed_field
+    @property
+    def compliance_deadline(self) -> date | None:
+        try:
+            return datetime.strptime(
+                self.compliance_deadline_str, "%Y-%m-%d"
+            ).date()
+        except Exception as e:
+            print(e)
+            return None
+
 
 info_parse_prompt = ChatPromptTemplate.from_messages(
     [

python-langgraph/graphs/email_agent.py

@@ -1,18 +1,12 @@
-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
-
-logging.basicConfig(
-    level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s"
-)
-LOGGER = logging.getLogger(__name__)
+from utils.logging_config import LOGGER
 
 
 @tool

python-langgraph/graphs/notice_extraction.py

@@ -1,17 +1,11 @@
-import logging
 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 utils.graph_utils import create_legal_ticket, send_escalation_email
-
-logging.basicConfig(
-    level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s"
-)
-LOGGER = logging.getLogger(__name__)
+from utils.logging_config import LOGGER
 
 
 class GraphState(TypedDict):

python-langgraph/utils/graph_utils.py

@@ -1,14 +1,8 @@
-import logging
 import random
 import time
-
-from chains.notice_extraction import NoticeEmailExtract
 from pydantic import EmailStr
-
-logging.basicConfig(
-    level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s"
-)
-LOGGER = logging.getLogger(__name__)
+from chains.notice_extraction import NoticeEmailExtract
+from utils.logging_config import LOGGER
 
 
 def send_escalation_email(

python-langgraph/utils/logging_config.py

@@ -0,0 +1,7 @@
+import logging
+
+logging.getLogger("httpx").setLevel(logging.WARNING)
+logging.basicConfig(
+    level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s"
+)
+LOGGER = logging.getLogger(__name__)