fix: type checks passing

Author: apkostka

backend/tenantfirstaid/chat.py

@@ -3,7 +3,7 @@
 from flask.views import View
 import os
 
-from .session import TenantSessionData, TenantSessionMessage
+from .session import TenantSessionMessage
 
 API_KEY = os.getenv("OPENAI_API_KEY", os.getenv("GITHUB_API_KEY"))
 BASE_URL = os.getenv("MODEL_ENDPOINT", "https://api.openai.com/v1")
@@ -31,12 +31,11 @@
 
 
 class ChatManager:
-    def __init__(self, tenant_session: TenantSessionData = None):
+    def __init__(self):
         self.client = OpenAI(
             api_key=API_KEY,
             base_url=BASE_URL,
         )
-        self.tenant_session = tenant_session
 
     def get_client(self):
         return self.client

backend/tenantfirstaid/session.py

@@ -70,11 +70,13 @@ def set(self, value: TenantSessionData):
         self.db_con.set(session_id, json.dumps(value))
 
     def getNewSessionData(self) -> TenantSessionData:
-        return {
-            "city": "",
-            "state": "",
-            "messages": [],
-        }
+        return TenantSessionData(
+            {
+                "city": "",
+                "state": "",
+                "messages": [],
+            }
+        )
 
 
 # The Flask view to initialize a session
@@ -90,7 +92,7 @@ def dispatch_request(self):
         state = data["state"]
 
         # Initialize the session with city and state
-        initial_data: TenantSessionData = {"city": city, "state": state, "messages": []}
+        initial_data = TenantSessionData({"city": city, "state": state, "messages": []})
         self.tenant_session.set(initial_data)
 
         return Response(