simile 0.3.1__py3-none-any.whl → 0.3.3__py3-none-any.whl

simile/client.py

@@ -6,6 +6,7 @@ from pydantic import BaseModel
 
 from .models import (
     Population,
+    PopulationInfo,
     Agent as AgentModel,
     DataItem,
     DeletionResponse,
@@ -168,6 +169,13 @@ class Simile:
         )
         return response_data
 
+    async def get_population_info(self, population_id: Union[str, uuid.UUID]) -> PopulationInfo:
+        """Gets basic population info (name and agent count) without full population data."""
+        response_data = await self._request(
+            "GET", f"populations/info/{str(population_id)}", response_model=PopulationInfo
+        )
+        return response_data
+
     async def delete_population(
         self, population_id: Union[str, uuid.UUID]
     ) -> DeletionResponse:

simile/models.py

@@ -13,6 +13,13 @@ class Population(BaseModel):
     updated_at: datetime
 
 
+class PopulationInfo(BaseModel):
+    population_id: uuid.UUID
+    name: str
+    description: Optional[str] = None
+    agent_count: int
+
+
 class DataItem(BaseModel):
     id: uuid.UUID
     agent_id: uuid.UUID

simile/resources.py

@@ -10,6 +10,7 @@ from .models import (
     AddContextRequest,
     AddContextResponse,
     SurveySessionDetailResponse,
+    SurveySessionCreateResponse,
 )
 
 if TYPE_CHECKING:
@@ -307,13 +308,13 @@ class SurveySession:
             "POST",
             endpoint,
             params=params,
-            response_model=None,  # Raw response
+            response_model=SurveySessionCreateResponse,
         )
         
         # Create a new SurveySession instance from the response
         return SurveySession(
-            id=uuid.UUID(response["id"]),
-            agent_id=uuid.UUID(response["agent_id"]),
-            status=response.get("status", "active"),
+            id=response.id,
+            agent_id=response.agent_id,
+            status=response.status,
             client=self._client,
         )

{simile-0.3.1.dist-info → simile-0.3.3.dist-info}/METADATA

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

simile-0.3.3.dist-info/RECORD

@@ -0,0 +1,11 @@
+simile/__init__.py,sha256=gElQY-o-_BSigva3w9zIX2KuCR49OM3MpL8sqeUpKzk,936
+simile/auth_client.py,sha256=ICImmaA5fZX9ADbIPIUh4RED3hBZvLf3XSiaqELDAME,7923
+simile/client.py,sha256=Y-u8LH6peWoBJSEP8fva9pq5DkhYf5DpETwqcdg0ogg,12679
+simile/exceptions.py,sha256=-rJ3KZcpvNRi9JXbDpxWDSL2lU1mEJX2piwYRZvhKmg,1406
+simile/models.py,sha256=tqTgNWQjoHdSDffsAhiWyzf3wKgbiU137CtWr3pcFpg,5441
+simile/resources.py,sha256=Juoa32SlN8LdvzCFaobq7uV2itmcwbk5dM_mlc4bCVk,10420
+simile-0.3.3.dist-info/licenses/LICENSE,sha256=tpxX3bpODfyOQVyEM6kCMvPHFCpkjFDj0AICRqKqOFA,1066
+simile-0.3.3.dist-info/METADATA,sha256=FldEaAGi3LwkMt9SBv3UIxtpfCCqJJaJnFef8OskCyU,1276
+simile-0.3.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+simile-0.3.3.dist-info/top_level.txt,sha256=41lJneubAG4-ZOAs5qn7iDtDb-MDxa6DdvgBKwNX84M,7
+simile-0.3.3.dist-info/RECORD,,

simile-0.3.1.dist-info/RECORD

@@ -1,11 +0,0 @@
-simile/__init__.py,sha256=gElQY-o-_BSigva3w9zIX2KuCR49OM3MpL8sqeUpKzk,936
-simile/auth_client.py,sha256=ICImmaA5fZX9ADbIPIUh4RED3hBZvLf3XSiaqELDAME,7923
-simile/client.py,sha256=2ZCNzT1j-wZoT0SanLcdO0FPQN6gToksLL5hkBCcc9E,12292
-simile/exceptions.py,sha256=-rJ3KZcpvNRi9JXbDpxWDSL2lU1mEJX2piwYRZvhKmg,1406
-simile/models.py,sha256=SOAqUpxkiZduDBuMooTaOjiLX_3bBolWn7ivj8A-MgY,5304
-simile/resources.py,sha256=vMt6UQ48OH6EXoCIeHSY2KaBTyJ5pUKToX7B90-GIXI,10425
-simile-0.3.1.dist-info/licenses/LICENSE,sha256=tpxX3bpODfyOQVyEM6kCMvPHFCpkjFDj0AICRqKqOFA,1066
-simile-0.3.1.dist-info/METADATA,sha256=a_8_7NIn0jT_QfQgnZb2U7zI9LyD-DzyNlqcDtPdpLI,1276
-simile-0.3.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-simile-0.3.1.dist-info/top_level.txt,sha256=41lJneubAG4-ZOAs5qn7iDtDb-MDxa6DdvgBKwNX84M,7
-simile-0.3.1.dist-info/RECORD,,