simile 0.3.6__py3-none-any.whl → 0.3.8__py3-none-any.whl

simile/client.py

@@ -1,6 +1,6 @@
 import httpx
 from httpx import AsyncClient
-from typing import List, Dict, Any, Optional, Union, Type
+from typing import List, Dict, Any, Optional, Union, Type, AsyncGenerator
 import uuid
 from pydantic import BaseModel
 
@@ -312,6 +312,68 @@ class Simile:
         )
         return response_data
 
+    async def stream_open_response(
+        self,
+        agent_id: uuid.UUID,
+        question: str,
+        data_types: Optional[List[str]] = None,
+        exclude_data_types: Optional[List[str]] = None,
+        images: Optional[Dict[str, str]] = None,
+    ) -> AsyncGenerator[str, None]:
+        """Streams an open response from an agent."""
+        endpoint = f"/generation/open-stream/{str(agent_id)}"
+        request_payload = OpenGenerationRequest(
+            question=question,
+            data_types=data_types,
+            exclude_data_types=exclude_data_types,
+            images=images,
+        )
+
+        url = self.base_url + endpoint  # assuming self.base_url is defined
+
+        async with httpx.AsyncClient(timeout=None) as client:
+            async with client.stream("POST", url, json=request_payload.model_dump()) as response:
+                response.raise_for_status()
+                async for line in response.aiter_lines():
+                    if line.strip():  # skip empty lines
+                        if line.startswith("data: "):  # optional, if using SSE format
+                            yield line.removeprefix("data: ").strip()
+                        else:
+                            yield line.strip()
+
+    async def stream_closed_response(
+        self,
+        agent_id: uuid.UUID,
+        question: str,
+        options: List[str],
+        data_types: Optional[List[str]] = None,
+        exclude_data_types: Optional[List[str]] = None,
+        images: Optional[Dict[str, str]] = None,
+    ) -> AsyncGenerator[str, None]:
+        """Streams a closed response from an agent."""
+        endpoint = f"/generation/closed-stream/{str(agent_id)}"
+        
+        request_payload = {
+            "question": question,
+            "options": options,
+            "data_types": data_types,
+            "exclude_data_types": exclude_data_types,
+            "images": images,
+        }
+
+        url = self.base_url + endpoint  # assuming self.base_url is defined
+
+        async with httpx.AsyncClient(timeout=None) as client:
+            async with client.stream("POST", url, json=request_payload) as response:
+                response.raise_for_status()
+                async for line in response.aiter_lines():
+                    if line.strip():  # skip empty lines
+                        if line.startswith("data: "):  # optional, if using SSE format
+                            yield line.removeprefix("data: ").strip()
+                        else:
+                            yield line.strip()
+
+
     async def generate_open_response(
         self,
         agent_id: uuid.UUID,

simile/models.py

@@ -46,6 +46,7 @@ class CreatePopulationPayload(BaseModel):
 class InitialDataItemPayload(BaseModel):
     data_type: str
     content: Any
+    metadata: Optional[Dict[str, Any]] = None
 
 
 class CreateAgentPayload(BaseModel):

{simile-0.3.6.dist-info → simile-0.3.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: simile
-Version: 0.3.6
+Version: 0.3.8
 Summary: Package for interfacing with Simile AI agents for simulation
 Author-email: Simile AI <cqz@simile.ai>
 License: MIT

simile-0.3.8.dist-info/RECORD

@@ -0,0 +1,11 @@
+simile/__init__.py,sha256=GgbhCv5G5F8X_DIYw0H72D8fRuVHW7SLezGJykALpd0,1036
+simile/auth_client.py,sha256=ICImmaA5fZX9ADbIPIUh4RED3hBZvLf3XSiaqELDAME,7923
+simile/client.py,sha256=bV2ih_z6c7ztZooUJetPTTTlbIFOmgx5F6TUruIKA1M,16332
+simile/exceptions.py,sha256=-rJ3KZcpvNRi9JXbDpxWDSL2lU1mEJX2piwYRZvhKmg,1406
+simile/models.py,sha256=GG7hOWGv5sIhpg7tzRzeLR4NR1DLMCz99PPVqcnCDLc,5533
+simile/resources.py,sha256=Juoa32SlN8LdvzCFaobq7uV2itmcwbk5dM_mlc4bCVk,10420
+simile-0.3.8.dist-info/licenses/LICENSE,sha256=tpxX3bpODfyOQVyEM6kCMvPHFCpkjFDj0AICRqKqOFA,1066
+simile-0.3.8.dist-info/METADATA,sha256=z49F7MZ6BcwCH5wp8Q63rDlqs23usTC1KSMJk7W8JQw,1276
+simile-0.3.8.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+simile-0.3.8.dist-info/top_level.txt,sha256=41lJneubAG4-ZOAs5qn7iDtDb-MDxa6DdvgBKwNX84M,7
+simile-0.3.8.dist-info/RECORD,,

simile-0.3.6.dist-info/RECORD

@@ -1,11 +0,0 @@
-simile/__init__.py,sha256=GgbhCv5G5F8X_DIYw0H72D8fRuVHW7SLezGJykALpd0,1036
-simile/auth_client.py,sha256=ICImmaA5fZX9ADbIPIUh4RED3hBZvLf3XSiaqELDAME,7923
-simile/client.py,sha256=zJcB2l7pqrgCK7FlY_9eG3ATe39eMRpawEASwTwKXn0,13779
-simile/exceptions.py,sha256=-rJ3KZcpvNRi9JXbDpxWDSL2lU1mEJX2piwYRZvhKmg,1406
-simile/models.py,sha256=vbWrK5VKDPsiI9K4n2dCC41Y9who7cgTHSL9ARs25K0,5487
-simile/resources.py,sha256=Juoa32SlN8LdvzCFaobq7uV2itmcwbk5dM_mlc4bCVk,10420
-simile-0.3.6.dist-info/licenses/LICENSE,sha256=tpxX3bpODfyOQVyEM6kCMvPHFCpkjFDj0AICRqKqOFA,1066
-simile-0.3.6.dist-info/METADATA,sha256=-LjtcX15bPSTz4t8TjEgsZ6XOnfOdmkhLff75hJIBPY,1276
-simile-0.3.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-simile-0.3.6.dist-info/top_level.txt,sha256=41lJneubAG4-ZOAs5qn7iDtDb-MDxa6DdvgBKwNX84M,7
-simile-0.3.6.dist-info/RECORD,,