Fix frontend and api state handling (#11)

* Fix frontend and api state handling

* Increase timeout to 6 mins

* Fix test

src/geo_assistant/api/app.py

@@ -94,6 +94,8 @@ async def stream_chat(
             payload = update[agent]
             if "place" not in payload:  # TODO: why is this needed?
                 payload["place"] = None
+            if "search_area" not in payload:  # TODO: why is this needed?
+                payload["search_area"] = None
             state_payload = GeoAssistantState(**payload)
 
             resp = ChatResponse(thread_id=str(thread_id), state=state_payload)

src/geo_assistant/frontend/app.py

@@ -31,6 +31,7 @@ def stream_chat(user_message: str):
         "agent_state_input": {
             "messages": [{"type": "human", "content": user_message}],
             "place": None,
+            "search_area": None,
         },
     }
 
@@ -38,7 +39,7 @@ def stream_chat(user_message: str):
         "POST",
         f"{API_BASE_URL}/chat",
         json=request_body,
-        timeout=60.0,
+        timeout=360.0,
     ) as response:
         response.raise_for_status()
 

tests/test_api.py

@@ -31,6 +31,7 @@ async def test_hello_world(initialized_app):
                 "agent_state_input": {
                     "messages": [{"content": "Hello, world!", "type": "human"}],
                     "place": None,
+                    "search_area": None,
                 },
                 "thread_id": str(thread_id),
             },