simile 0.3.5__py3-none-any.whl → 0.3.7__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
 
@@ -233,6 +233,31 @@ class Simile:
         )
         return response_data
 
+    async def add_agent_to_population(
+        self, agent_id: Union[str, uuid.UUID], population_id: Union[str, uuid.UUID]
+    ) -> Dict[str, str]:
+        """Add an agent to an additional population."""
+        raw_response = await self._request(
+            "POST", f"agents/{str(agent_id)}/populations/{str(population_id)}"
+        )
+        return raw_response.json()
+
+    async def remove_agent_from_population(
+        self, agent_id: Union[str, uuid.UUID], population_id: Union[str, uuid.UUID]
+    ) -> Dict[str, str]:
+        """Remove an agent from a population."""
+        raw_response = await self._request(
+            "DELETE", f"agents/{str(agent_id)}/populations/{str(population_id)}"
+        )
+        return raw_response.json()
+
+    async def get_populations_for_agent(
+        self, agent_id: Union[str, uuid.UUID]
+    ) -> Dict[str, Any]:
+        """Get all populations an agent belongs to."""
+        raw_response = await self._request("GET", f"agents/{str(agent_id)}/populations")
+        return raw_response.json()
+
     async def create_data_item(
         self, agent_id: Union[str, uuid.UUID], data_type: str, content: Any, metadata: Optional[Dict[str, Any]] = None
     ) -> DataItem:
@@ -287,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-0.3.5.dist-info → simile-0.3.7.dist-info}/METADATA

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

{simile-0.3.5.dist-info → simile-0.3.7.dist-info}/RECORD

@@ -1,11 +1,11 @@
 simile/__init__.py,sha256=GgbhCv5G5F8X_DIYw0H72D8fRuVHW7SLezGJykALpd0,1036
 simile/auth_client.py,sha256=ICImmaA5fZX9ADbIPIUh4RED3hBZvLf3XSiaqELDAME,7923
-simile/client.py,sha256=7iXw7mF4SVWi8fm4VUnfkoRMFdi4qLm8S6ulWgvQLLA,12741
+simile/client.py,sha256=bV2ih_z6c7ztZooUJetPTTTlbIFOmgx5F6TUruIKA1M,16332
 simile/exceptions.py,sha256=-rJ3KZcpvNRi9JXbDpxWDSL2lU1mEJX2piwYRZvhKmg,1406
 simile/models.py,sha256=vbWrK5VKDPsiI9K4n2dCC41Y9who7cgTHSL9ARs25K0,5487
 simile/resources.py,sha256=Juoa32SlN8LdvzCFaobq7uV2itmcwbk5dM_mlc4bCVk,10420
-simile-0.3.5.dist-info/licenses/LICENSE,sha256=tpxX3bpODfyOQVyEM6kCMvPHFCpkjFDj0AICRqKqOFA,1066
-simile-0.3.5.dist-info/METADATA,sha256=Dli_UkcMUpWx7C85L3cjFyqqbsmToT70CwXc2BqfcN8,1276
-simile-0.3.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-simile-0.3.5.dist-info/top_level.txt,sha256=41lJneubAG4-ZOAs5qn7iDtDb-MDxa6DdvgBKwNX84M,7
-simile-0.3.5.dist-info/RECORD,,
+simile-0.3.7.dist-info/licenses/LICENSE,sha256=tpxX3bpODfyOQVyEM6kCMvPHFCpkjFDj0AICRqKqOFA,1066
+simile-0.3.7.dist-info/METADATA,sha256=8LMpcxIGGEXn1CNlgRgJThMWkyhMCWwcl_1G3xAL85g,1276
+simile-0.3.7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+simile-0.3.7.dist-info/top_level.txt,sha256=41lJneubAG4-ZOAs5qn7iDtDb-MDxa6DdvgBKwNX84M,7
+simile-0.3.7.dist-info/RECORD,,