handle stream errors (#63)

* handle stream errors

* adjust tests

* update version

Author: augustas1

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "weaviate-agents"
-version = "0.12.0"
+version = "0.12.1"
 description = "The official sub-package for the Weaviate Agents project."
 readme = "README.md"
 requires-python = ">=3.9"

test/query_agent/test_query_model.py

@@ -370,6 +370,8 @@ async def test_async_run_success(monkeypatch):
 
 
 class MockIterSSESuccess:
+    response = FakeResponse(200, {})
+
     def iter_sse(self):
         yield ServerSentEvent(
             event="progress_message",
@@ -514,6 +516,8 @@ async def test_async_run_failure(monkeypatch):
 
 
 class MockIterSSEFailure:
+    response = FakeResponse(200, {})
+
     def iter_sse(self):
         yield ServerSentEvent(
             event="progress_message",

weaviate_agents/query/query_agent.py

@@ -287,6 +287,10 @@ def stream(
                 headers=self._headers,
                 timeout=self._timeout,
             ) as events:
+                if events.response.is_error:
+                    events.response.read()
+                    raise Exception(events.response.text)
+
                 for sse in events.iter_sse():
                     output = _parse_sse(sse)
                     if isinstance(output, ProgressMessage):
@@ -403,6 +407,10 @@ async def stream(
                 headers=self._headers,
                 timeout=self._timeout,
             ) as events:
+                if events.response.is_error:
+                    events.response.read()
+                    raise Exception(events.response.text)
+
                 async for sse in events.aiter_sse():
                     output = _parse_sse(sse)
                     if isinstance(output, ProgressMessage):