simile 0.2.0__py3-none-any.whl → 0.2.1__py3-none-any.whl

simile/client.py

@@ -17,6 +17,12 @@ DEFAULT_BASE_URL = "https://simile-api-3a83be7adae0.herokuapp.com/api/v1"
 TIMEOUT_CONFIG = httpx.Timeout(5.0, read=30.0, write=30.0, pool=30.0)
 
 class Simile:
+    # Make exceptions available as class attributes for convenience
+    APIError = SimileAPIError
+    AuthenticationError = SimileAuthenticationError
+    NotFoundError = SimileNotFoundError
+    BadRequestError = SimileBadRequestError
+
     def __init__(self, api_key: str, base_url: str = DEFAULT_BASE_URL):
         if not api_key:
             raise ValueError("API key is required.")
@@ -59,7 +65,9 @@ class Simile:
             raise SimileAPIError(f"Request error: {e}")
 
     # --- Population Endpoints ---
-    async def create_population(self, payload: CreatePopulationPayload) -> Population:
+    async def create_population(self, name: str, description: Optional[str] = None) -> Population:
+        """Creates a new population."""
+        payload = CreatePopulationPayload(name=name, description=description)
         response_data = await self._request("POST", "populations/create", json=payload.model_dump(mode='json', exclude_none=True), response_model=Population)
         return response_data
 
@@ -72,7 +80,14 @@ class Simile:
         return response_data
 
     # --- Agent Endpoints ---
-    async def create_agent(self, payload: CreateAgentPayload) -> Agent:
+    async def create_agent(self, name: str, population_id: Optional[Union[str, uuid.UUID]] = None, agent_data: Optional[List[Dict[str, Any]]] = None) -> Agent:
+        """Creates a new agent, optionally within a population and with initial data items."""
+        # Ensure population_id is uuid if provided
+        pop_id_uuid: Optional[uuid.UUID] = None
+        if population_id:
+            pop_id_uuid = uuid.UUID(str(population_id)) if not isinstance(population_id, uuid.UUID) else population_id
+        
+        payload = CreateAgentPayload(name=name, population_id=pop_id_uuid, agent_data=agent_data)
         response_data = await self._request("POST", "agents/create", json=payload.model_dump(mode='json', exclude_none=True), response_model=Agent)
         return response_data
 
@@ -85,7 +100,9 @@ class Simile:
         return response_data
 
     # --- Data Item Endpoints ---
-    async def create_data_item(self, agent_id: Union[str, uuid.UUID], payload: CreateDataItemPayload) -> DataItem:
+    async def create_data_item(self, agent_id: Union[str, uuid.UUID], data_type: str, content: Any) -> DataItem:
+        """Creates a new data item for a specific agent."""
+        payload = CreateDataItemPayload(data_type=data_type, content=content)
         response_data = await self._request("POST", f"data_item/create/{str(agent_id)}", json=payload.model_dump(mode='json'), response_model=DataItem)
         return response_data
 
@@ -97,11 +114,15 @@ class Simile:
         params = {}
         if data_type:
             params["data_type"] = data_type
-        raw_response = await self._request("GET", f"data_item/list/{str(agent_id)}", params=params) 
+        # Ensure agent_id is string for URL
+        agent_id_str = str(agent_id)
+        raw_response = await self._request("GET", f"data_item/list/{agent_id_str}", params=params) 
         return [DataItem(**item) for item in raw_response.json()] 
 
-    async def update_data_item(self, data_item_id: Union[str, uuid.UUID], payload: UpdateDataItemPayload) -> DataItem:
-        response_data = await self._request("POST", f"data_item/update/{data_item_id}", json=payload.model_dump(), response_model=DataItem)
+    async def update_data_item(self, data_item_id: Union[str, uuid.UUID], content: Any) -> DataItem:
+        """Updates an existing data item."""
+        payload = UpdateDataItemPayload(content=content)
+        response_data = await self._request("POST", f"data_item/update/{str(data_item_id)}", json=payload.model_dump(), response_model=DataItem)
         return response_data
 
     async def delete_data_item(self, data_item_id: Union[str, uuid.UUID]) -> DeletionResponse:
@@ -109,8 +130,10 @@ class Simile:
         return response_data
 
     # --- LLM Generation Methods ---
-    async def generate_qual_response(self, agent_id: uuid.UUID, request_payload: QualGenerationRequest) -> QualGenerationResponse:
+    async def generate_qual_response(self, agent_id: uuid.UUID, question: str) -> QualGenerationResponse:
+        """Generates a qualitative response from an agent based on a question."""
         endpoint = f"/generation/qual/{str(agent_id)}"
+        request_payload = QualGenerationRequest(question=question) # Create payload internally
         response_data = await self._request(
             "POST", 
             endpoint, 
@@ -119,8 +142,10 @@ class Simile:
         )
         return response_data
 
-    async def generate_mc_response(self, agent_id: uuid.UUID, request_payload: MCGenerationRequest) -> MCGenerationResponse:
-        endpoint = f"generation/mc/{agent_id}" 
+    async def generate_mc_response(self, agent_id: uuid.UUID, question: str, options: List[str]) -> MCGenerationResponse:
+        """Generates a multiple-choice response from an agent."""
+        endpoint = f"generation/mc/{str(agent_id)}" # Ensured str(agent_id)
+        request_payload = MCGenerationRequest(question=question, options=options) # Create payload internally
         response_data = await self._request("POST", endpoint, json=request_payload.model_dump(), response_model=MCGenerationResponse)
         return response_data
 

simile/models.py

@@ -34,10 +34,15 @@ class CreatePopulationPayload(BaseModel):
     description: Optional[str] = None
 
 
+class InitialDataItemPayload(BaseModel):
+    data_type: str
+    content: Any
+
+
 class CreateAgentPayload(BaseModel):
     name: str
     population_id: Optional[uuid.UUID] = None
-    agent_data: Optional[List[Dict[str, Any]]] = None # For initial data items
+    agent_data: Optional[List[InitialDataItemPayload]] = None # For initial data items
 
 
 class CreateDataItemPayload(BaseModel):

{simile-0.2.0.dist-info → simile-0.2.1.dist-info}/METADATA

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

simile-0.2.1.dist-info/RECORD

@@ -0,0 +1,9 @@
+simile/__init__.py,sha256=2OZ1LQIkAEtSs0EI5Fzlg7QGKduCgCe_nTh9FfIuHlQ,865
+simile/client.py,sha256=HjIwjUlkH2qVo67gcC1lwhqKQlzzTMNeLSZIKpLy36Y,7934
+simile/exceptions.py,sha256=-rJ3KZcpvNRi9JXbDpxWDSL2lU1mEJX2piwYRZvhKmg,1406
+simile/models.py,sha256=BV6n1zgXT1CyNSdwWd1ASm2llknC401aybrw6EtGXqM,1617
+simile-0.2.1.dist-info/licenses/LICENSE,sha256=tpxX3bpODfyOQVyEM6kCMvPHFCpkjFDj0AICRqKqOFA,1066
+simile-0.2.1.dist-info/METADATA,sha256=ee9w-3KBLIaH3fukxpvY65Tayoboj4kOeRxXwoO9zzI,1245
+simile-0.2.1.dist-info/WHEEL,sha256=DnLRTWE75wApRYVsjgc6wsVswC54sMSJhAEd4xhDpBk,91
+simile-0.2.1.dist-info/top_level.txt,sha256=41lJneubAG4-ZOAs5qn7iDtDb-MDxa6DdvgBKwNX84M,7
+simile-0.2.1.dist-info/RECORD,,

simile-0.2.0.dist-info/RECORD

@@ -1,9 +0,0 @@
-simile/__init__.py,sha256=2OZ1LQIkAEtSs0EI5Fzlg7QGKduCgCe_nTh9FfIuHlQ,865
-simile/client.py,sha256=8FMhdyi35CQPJFh7pMLf6c9pczN7IlcCvnBh8kocOiI,6377
-simile/exceptions.py,sha256=-rJ3KZcpvNRi9JXbDpxWDSL2lU1mEJX2piwYRZvhKmg,1406
-simile/models.py,sha256=3T1UYCPp3KWMBD-u7lZEMahm4fS4ltm9h0oPUWs_oyI,1530
-simile-0.2.0.dist-info/licenses/LICENSE,sha256=tpxX3bpODfyOQVyEM6kCMvPHFCpkjFDj0AICRqKqOFA,1066
-simile-0.2.0.dist-info/METADATA,sha256=CTeN0Kl12ehJhMPbvra6o7ruVuojOyfpPB7uvUHuHjw,1245
-simile-0.2.0.dist-info/WHEEL,sha256=DnLRTWE75wApRYVsjgc6wsVswC54sMSJhAEd4xhDpBk,91
-simile-0.2.0.dist-info/top_level.txt,sha256=41lJneubAG4-ZOAs5qn7iDtDb-MDxa6DdvgBKwNX84M,7
-simile-0.2.0.dist-info/RECORD,,