simile 0.3.5__tar.gz → 0.3.7__tar.gz

{simile-0.3.5 → simile-0.3.7}/PKG-INFO

@@ -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 → simile-0.3.7}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "simile"
-version = "0.3.5"
+version = "0.3.7"
 authors = [
   { name="Simile AI", email="cqz@simile.ai" },
 ]

{simile-0.3.5 → simile-0.3.7}/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 → simile-0.3.7}/simile.egg-info/PKG-INFO

@@ -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