fix: review changes from @yangm2

Author: apkostka

backend/scripts/generate_conversation/chat.py

@@ -3,6 +3,7 @@
 # dependencies = [
 #     "openai",
 #     "pandas",
+#     "python-dotenv",
 # ]
 # ///
 import os
@@ -11,15 +12,16 @@
 from pathlib import Path
 import pandas as pd
 from typing import Self
-
 from openai.types.responses.response_input_param import Message
 
-if Path("../../.env").exists():
-    from dotenv import load_dotenv
+from tenantfirstaid.chat import DEFAULT_INSTRUCTIONS, ChatManager
 
-    load_dotenv(override=True)
+dot_env_path = Path(__file__).parent.parent.parent / ".env"
+print(f"Loading environment variables from {dot_env_path}")
+if dot_env_path.exists():
+    from dotenv import load_dotenv
 
-from tenantfirstaid.chat import DEFAULT_INSTRUCTIONS, ChatManager
+    load_dotenv(dotenv_path=dot_env_path, override=True)
 
 BOT_INSTRUCTIONS = DEFAULT_INSTRUCTIONS
 

backend/tenantfirstaid/chat.py

@@ -2,6 +2,7 @@
 from openai.types.shared import Reasoning
 from openai.types.responses import (
     FileSearchToolParam,
+    Response as ResponseEvent,
     ResponseStreamEvent,
 )
 from openai.types.responses.response_input_param import Message
@@ -126,7 +127,7 @@ def prepare_openai_tools(self, city: str, state: str):
 
     def generate_chat_response(
         self, messages: list[Message], city: str, state: str, stream=False
-    ) -> ResponseStreamEvent:
+    ) -> ResponseStreamEvent | ResponseEvent:
         instructions = self.prepare_developer_instructions(city, state)
         tools = self.prepare_openai_tools(city, state)
 

backend/tenantfirstaid/session.py

@@ -15,11 +15,7 @@ class TenantSessionData(TypedDict):
     messages: list[Message]  # List of messages with role and content
 
 
-new_session_data = {
-    "city": "null",
-    "state": "or",
-    "messages": [],
-}
+NEW_SESSION_DATA = TenantSessionData(city="null", state="or", messages=[])
 
 
 # The class to manage tenant sessions using Valkey and Flask sessions
@@ -73,7 +69,7 @@ def set(self, value: TenantSessionData):
         self.db_con.set(session_id, json.dumps(value))
 
     def getNewSessionData(self) -> TenantSessionData:
-        return TenantSessionData(new_session_data)
+        return TenantSessionData(NEW_SESSION_DATA.copy())
 
 
 # The Flask view to initialize a session