simile 0.4.10__py3-none-any.whl → 0.4.12__py3-none-any.whl

simile/client.py

@@ -1,5 +1,5 @@
 import httpx
-from httpx import AsyncClient
+from httpx import AsyncClient, Limits
 from typing import List, Dict, Any, Optional, Union, Type, AsyncGenerator
 import uuid
 from pydantic import BaseModel
@@ -24,6 +24,7 @@ from .models import (
     SurveySessionCreateResponse,
     SurveySessionDetailResponse,
     MemoryStream,
+    UpdateAgentInfoPayload,
 )
 from .resources import Agent, SurveySession
 from .exceptions import (
@@ -34,7 +35,7 @@ from .exceptions import (
 )
 
 DEFAULT_BASE_URL = "https://api.simile.ai/api/v1"
-TIMEOUT_CONFIG = httpx.Timeout(5.0, read=30.0, write=30.0, pool=30.0)
+TIMEOUT_CONFIG = httpx.Timeout(5.0, read=60.0, write=30.0, pool=30.0)
 
 
 class Simile:
@@ -43,13 +44,26 @@ class Simile:
     NotFoundError = SimileNotFoundError
     BadRequestError = SimileBadRequestError
 
-    def __init__(self, api_key: str, base_url: str = DEFAULT_BASE_URL):
+    def __init__(
+            self,
+            api_key: str,
+            base_url: str = DEFAULT_BASE_URL,
+            max_connections: int = 5000,
+            max_keepalive_connections: int = 2000,
+            keepalive_expiry: float = 300.0,
+    ):
         if not api_key:
             raise ValueError("API key is required.")
         self.api_key = api_key
         self.base_url = base_url.rstrip("/")
+
+        limits = Limits(
+            max_connections=max_connections,
+            max_keepalive_connections=max_keepalive_connections,
+            keepalive_expiry=keepalive_expiry,
+        )
         self._client = AsyncClient(
-            headers={"X-API-Key": self.api_key}, timeout=TIMEOUT_CONFIG
+            headers={"X-API-Key": self.api_key}, timeout=TIMEOUT_CONFIG, limits=limits,
         )
 
     async def _request(
@@ -288,6 +302,27 @@ class Simile:
             "GET", f"agents/get/{str(agent_id)}", response_model=AgentModel
         )
         return response_data
+    
+    async def update_agent_info(
+        self,
+        agent_id: Union[str, uuid.UUID],
+        name: str,
+    ) -> AgentModel:
+        """
+        Updates agent information (name).
+        Name must be provided.
+        Requires write access to the agent.
+        """
+        payload = UpdateAgentInfoPayload(
+            name=name,
+        )
+        response_data = await self._request(
+            "PUT",
+            f"agents/update/{str(agent_id)}",
+            json=payload.model_dump(mode="json", exclude_none=True),
+            response_model=AgentModel,
+        )
+        return response_data
 
     async def delete_agent(self, agent_id: Union[str, uuid.UUID]) -> DeletionResponse:
         response_data = await self._request(

simile/models.py

@@ -58,6 +58,11 @@ class UpdatePopulationInfoPayload(BaseModel):
     description: Optional[str] = None
     metadata: Optional[Dict[str, Any]] = None
 
+
+class UpdateAgentInfoPayload(BaseModel):
+    name: str
+
+
 class InitialDataItemPayload(BaseModel):
     data_type: str
     content: Any

{simile-0.4.10.dist-info → simile-0.4.12.dist-info}/METADATA

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

simile-0.4.12.dist-info/RECORD

@@ -0,0 +1,11 @@
+simile/__init__.py,sha256=JAYtieyGg6YYiCackJ6YNlGJkFWmhryzbwwVt4K67uI,1360
+simile/auth_client.py,sha256=ICImmaA5fZX9ADbIPIUh4RED3hBZvLf3XSiaqELDAME,7923
+simile/client.py,sha256=9nZo7y_umY52oxF-YszPlbDXEbSI5MfJu0Aq0jYy7CU,30635
+simile/exceptions.py,sha256=Q1lbfwR7mEn_LYmwjAnsMc8BW79JNPvmCmVoPibYisU,1502
+simile/models.py,sha256=qDA768d5DtuIXUgHJ8T2dgY3Ypy7IvDDUEumMzX3__Y,12536
+simile/resources.py,sha256=LSYZSzx1YO69xvShGxxLFVPjQHw1WukHXhdYc1EyuOs,10555
+simile-0.4.12.dist-info/licenses/LICENSE,sha256=tpxX3bpODfyOQVyEM6kCMvPHFCpkjFDj0AICRqKqOFA,1066
+simile-0.4.12.dist-info/METADATA,sha256=fHtzZuJGxOk2W_ATBganwqcZZSr0EIKYIE8z5HDhQx8,1599
+simile-0.4.12.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+simile-0.4.12.dist-info/top_level.txt,sha256=41lJneubAG4-ZOAs5qn7iDtDb-MDxa6DdvgBKwNX84M,7
+simile-0.4.12.dist-info/RECORD,,

simile-0.4.10.dist-info/RECORD

@@ -1,11 +0,0 @@
-simile/__init__.py,sha256=JAYtieyGg6YYiCackJ6YNlGJkFWmhryzbwwVt4K67uI,1360
-simile/auth_client.py,sha256=ICImmaA5fZX9ADbIPIUh4RED3hBZvLf3XSiaqELDAME,7923
-simile/client.py,sha256=7QhbMCiBLF6loYZdWSpnbE2HjVEVWgLeHQv8XW-WiB0,29604
-simile/exceptions.py,sha256=Q1lbfwR7mEn_LYmwjAnsMc8BW79JNPvmCmVoPibYisU,1502
-simile/models.py,sha256=kD1zwLW4N1sbrfGggipSBTew--uK2Gc6OLrQb2ZICF4,12478
-simile/resources.py,sha256=LSYZSzx1YO69xvShGxxLFVPjQHw1WukHXhdYc1EyuOs,10555
-simile-0.4.10.dist-info/licenses/LICENSE,sha256=tpxX3bpODfyOQVyEM6kCMvPHFCpkjFDj0AICRqKqOFA,1066
-simile-0.4.10.dist-info/METADATA,sha256=qt6h9YEGRSv1s2lCJFvPO9lDk64RDsq983DwwS6qKyc,1599
-simile-0.4.10.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-simile-0.4.10.dist-info/top_level.txt,sha256=41lJneubAG4-ZOAs5qn7iDtDb-MDxa6DdvgBKwNX84M,7
-simile-0.4.10.dist-info/RECORD,,