polos-sdk 0.1.4__py3-none-any.whl → 0.1.5__py3-none-any.whl

polos/llm/stream.py

@@ -86,7 +86,7 @@ async def _stream_from_provider(
     Returns:
         Dictionary with chunk_index, response_content, response_tool_calls, usage, raw_output
     """
-    from ..features.events import publish as publish_event
+    from ..features.events import publish as publish_event, EventData
 
     chunk_index = 0
     response_content = None
@@ -103,8 +103,10 @@ async def _stream_from_provider(
     await publish_event(
         client=polos_client,
         topic=topic,
-        event_type="stream_start",
-        data={"step": agent_step},
+        event_data=EventData(
+            event_type="stream_start",
+            data={"step": agent_step},
+        ),
     )
 
     # Stream from provider
@@ -157,18 +159,20 @@ async def _stream_from_provider(
             await publish_event(
                 client=polos_client,
                 topic=topic,
-                event_type=event_type,
-                data={
-                    "step": agent_step,
-                    "chunk_index": chunk_index,
-                    "content": normalized_chunk["data"].get("content"),
-                    "tool_call": normalized_chunk["data"].get("tool_call"),
-                    "usage": normalized_chunk["data"].get("usage"),
-                    "_metadata": {
-                        "execution_id": agent_run_id,
-                        "workflow_id": ctx.workflow_id,
-                    },
-                },
+                event_data=EventData(
+                    event_type=event_type,
+                    data={
+                        "step": agent_step,
+                        "chunk_index": chunk_index,
+                        "content": normalized_chunk.get("data", {}).get("content"),
+                        "tool_call": normalized_chunk.get("data", {}).get("tool_call"),
+                        "usage": normalized_chunk.get("data", {}).get("usage"),
+                        "_metadata": {
+                            "execution_id": agent_run_id,
+                            "workflow_id": ctx.workflow_id,
+                        },
+                    }
+                ),
             )
             chunk_index += 1
 

{polos_sdk-0.1.4.dist-info → polos_sdk-0.1.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: polos-sdk
-Version: 0.1.4
+Version: 0.1.5
 Summary: Polos SDK for Python - Durable Agent Execution
 Project-URL: Homepage, https://github.com/polos-dev/polos
 Project-URL: Repository, https://github.com/polos-dev/polos

{polos_sdk-0.1.4.dist-info → polos_sdk-0.1.5.dist-info}/RECORD

@@ -16,7 +16,7 @@ polos/features/tracing.py,sha256=xOaj9WLqK7B9IwRUDyCDVZ7IGhVPDxQ1YB1UQ3aFK2w,227
 polos/features/wait.py,sha256=c2NUz1BPG56EGIdxQwuDhydnyztqD5B9KBTlaimdofE,2504
 polos/llm/__init__.py,sha256=SYLl5KdW3ah2lF5kpNsAuT-86MrNtsUDKXmznNsW8Fk,170
 polos/llm/generate.py,sha256=FuuY2vPeka7q8cpQ4pHzygmkO_pnvAadLP65xyFgESI,6491
-polos/llm/stream.py,sha256=oBPDL2t5srTMYKeKTm7dhVmTjWRgV-ynPhPKGM6ud98,6732
+polos/llm/stream.py,sha256=XUH3_AF2fCgAEIeu5DNxaKDWePIQfC00HkzX6u-YHEg,6920
 polos/llm/providers/__init__.py,sha256=xFiGlL4P1_3IxVXYFIphXWLBEnillFcRVmpa34YEysk,192
 polos/llm/providers/anthropic.py,sha256=1aAbDIG8pYhTD2aOFBKK87Qh_LZDmZU5nGqqDh29opQ,26488
 polos/llm/providers/azure.py,sha256=-_IxZhu6tCgoQ9TdmltUrimKVC6kw4DsU6eDIsitzOQ,1487
@@ -50,6 +50,6 @@ polos/utils/retry.py,sha256=Igz4degC0IUWnsJsRKVIrkNjYm57DG467EcLbmMOMuM,1398
 polos/utils/serializer.py,sha256=XfnZ7vIbTk_xly116De1xJiNncJdb1GIjAERys7ZV10,6554
 polos/utils/tracing.py,sha256=SkBktU12KyIQcl8q1wjyTX7QfGwyzdzEcBY8QTX_t6Y,1021
 polos/utils/worker_singleton.py,sha256=7-MOx9VFlyC-8waB7OMg6xER5NB7Ld7ofMku9yoitDo,1081
-polos_sdk-0.1.4.dist-info/METADATA,sha256=k1ggHONyfKT0v8bi7c2skwKsVoYikv4IrYc4dhwcSG8,3870
-polos_sdk-0.1.4.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-polos_sdk-0.1.4.dist-info/RECORD,,
+polos_sdk-0.1.5.dist-info/METADATA,sha256=TTWW8dETmBvCjhJptx-lEUrnmAb5Q3cje3E8Q5GmiNs,3870
+polos_sdk-0.1.5.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+polos_sdk-0.1.5.dist-info/RECORD,,