Attempt quick fix

Author: jamesbaskerville

src/humanloop/prompts/raw_client.py

@@ -2,6 +2,7 @@
 
 import contextlib
 import datetime as dt
+import json
 import typing
 from json.decoder import JSONDecodeError
 
@@ -686,7 +687,13 @@ def _iter():
                                 if _sse.data == None:
                                     return
                                 try:
-                                    yield _sse.data
+                                    yield typing.cast(
+                                        PromptCallStreamResponse,
+                                        construct_type(
+                                            type_=PromptCallStreamResponse,  # type: ignore
+                                            object_=json.loads(_sse.data),
+                                        ),
+                                    )
                                 except Exception:
                                     pass
                             return

tests/integration/test_prompts.py

@@ -1,5 +1,3 @@
-import json
-
 from humanloop.client import Humanloop
 
 from tests.integration.conftest import TestIdentifiers
@@ -39,7 +37,6 @@ def test_prompts_call_stream(
     output = ""
     for chunk in response:
         assert chunk is not None
-        chunk = json.loads(chunk)
         assert chunk.output is not None
         assert chunk.id is not None
         assert chunk.prompt_id is not None