simile 0.2.3__tar.gz → 0.2.5__tar.gz

{simile-0.2.3 → simile-0.2.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: simile
-Version: 0.2.3
+Version: 0.2.5
 Summary: Package for interfacing with Simile AI agents for simulation
 Author-email: Simile AI <cqz@simile.ai>
 License: MIT
@@ -21,6 +21,7 @@ Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: httpx>=0.20.0
 Requires-Dist: pydantic>=2.0.0
+Requires-Dist: genagent>=0.2.3
 Dynamic: license-file
 
 # Simile API Python Client

{simile-0.2.3 → simile-0.2.5}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "simile"
-version = "0.2.3"
+version = "0.2.5"
 authors = [
   { name="Simile AI", email="cqz@simile.ai" },
 ]
@@ -29,6 +29,7 @@ classifiers = [
 dependencies = [
     "httpx>=0.20.0",
     "pydantic>=2.0.0",
+    "genagent>=0.2.3",
 ]
 
 [project.urls]

{simile-0.2.3 → simile-0.2.5}/simile/client.py

@@ -64,8 +64,22 @@ class Simile:
                 raise SimileBadRequestError(detail=detail)
             else:
                 raise SimileAPIError(f"API request failed: {e}", status_code=status_code, detail=detail)
+        except httpx.ConnectTimeout:
+            raise SimileAPIError("Connection timed out while trying to connect.")
+        except httpx.ReadTimeout:
+            raise SimileAPIError("Timed out waiting for data from the server.")
+        except httpx.WriteTimeout:
+            raise SimileAPIError("Timed out while sending data to the server.")
+        except httpx.PoolTimeout:
+            raise SimileAPIError("Timed out waiting for a connection from the pool.")
+        except httpx.ConnectError:
+            raise SimileAPIError("Failed to connect to the server.")
+        except httpx.ProtocolError:
+            raise SimileAPIError("A protocol error occurred.")
+        except httpx.DecodingError:
+            raise SimileAPIError("Failed to decode the response.")
         except httpx.RequestError as e:
-            raise SimileAPIError(f"Request error: {e}")
+            raise SimileAPIError(f"An unknown request error occurred: {type(e).__name__}: {e}")
 
     def agent(self, agent_id: uuid.UUID) -> Agent:
         """Returns an Agent object to interact with a specific agent."""
@@ -105,7 +119,7 @@ class Simile:
         """Retrieves all agents belonging to a specific population."""
         endpoint = f"populations/{str(population_id)}/agents"
         raw_response = await self._request("GET", endpoint)
-        agents_data_list = raw_response.model_dump_json()
+        agents_data_list = raw_response.json()
         return [AgentModel(**data) for data in agents_data_list]
 
     async def create_agent(self, name: str, population_id: Optional[Union[str, uuid.UUID]] = None, agent_data: Optional[List[Dict[str, Any]]] = None) -> AgentModel:
@@ -142,7 +156,7 @@ class Simile:
             params["data_type"] = data_type
         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.model_dump_json()] 
+        return [DataItem(**item) for item in raw_response.json()] 
 
     async def update_data_item(self, data_item_id: Union[str, uuid.UUID], content: Any) -> DataItem:
         """Updates an existing data item."""
@@ -154,10 +168,10 @@ class Simile:
         response_data = await self._request("DELETE", f"data_item/delete/{str(data_item_id)}", response_model=DeletionResponse)
         return response_data
 
-    async def generate_qual_response(self, agent_id: uuid.UUID, question: str) -> QualGenerationResponse:
+    async def generate_qual_response(self, agent_id: uuid.UUID, question: str, image_url: Optional[str] = None) -> QualGenerationResponse:
         """Generates a qualitative response from an agent based on a question."""
         endpoint = f"/generation/qual/{str(agent_id)}"
-        request_payload = QualGenerationRequest(question=question)
+        request_payload = QualGenerationRequest(question=question, image_url=image_url)
         response_data = await self._request(
             "POST", 
             endpoint, 
@@ -166,10 +180,10 @@ class Simile:
         )
         return response_data
 
-    async def generate_mc_response(self, agent_id: uuid.UUID, question: str, options: List[str]) -> MCGenerationResponse:
+    async def generate_mc_response(self, agent_id: uuid.UUID, question: str, options: List[str], image_url: Optional[str] = None) -> MCGenerationResponse:
         """Generates a multiple-choice response from an agent."""
         endpoint = f"generation/mc/{str(agent_id)}"
-        request_payload = MCGenerationRequest(question=question, options=options)
+        request_payload = MCGenerationRequest(question=question, options=options, image_url=image_url)
         response_data = await self._request(
             "POST", 
             endpoint, 

{simile-0.2.3 → simile-0.2.5}/simile/models.py

@@ -61,6 +61,7 @@ class DeletionResponse(BaseModel):
 # --- Generation Operation Models ---
 class QualGenerationRequest(BaseModel):
     question: str
+    image_url: Optional[str] = None
 
 class QualGenerationResponse(BaseModel):
     question: str
@@ -69,6 +70,7 @@ class QualGenerationResponse(BaseModel):
 class MCGenerationRequest(BaseModel):
     question: str
     options: List[str]
+    image_url: Optional[str] = None
 
 class MCGenerationResponse(BaseModel):
     question: str

{simile-0.2.3 → simile-0.2.5}/simile/resources.py

@@ -1,5 +1,5 @@
 import uuid
-from typing import TYPE_CHECKING, List
+from typing import TYPE_CHECKING, List, Optional
 
 from .models import (
     QualGenerationRequest,
@@ -22,13 +22,13 @@ class Agent:
     def id(self) -> uuid.UUID:
         return self._agent_id
 
-    async def generate_qual_response(self, question: str) -> QualGenerationResponse:
+    async def generate_qual_response(self, question: str, image_url: Optional[str] = None) -> QualGenerationResponse:
         """Generates a qualitative response from this agent based on a question."""
-        return await self._client.generate_qual_response(agent_id=self._agent_id, question=question)
+        return await self._client.generate_qual_response(agent_id=self._agent_id, question=question, image_url=image_url)
 
-    async def generate_mc_response(self, question: str, options: List[str]) -> MCGenerationResponse:
+    async def generate_mc_response(self, question: str, options: List[str], image_url: Optional[str] = None) -> MCGenerationResponse:
         """Generates a multiple-choice response from this agent."""
-        return await self._client.generate_mc_response(agent_id=self._agent_id, question=question, options=options)
+        return await self._client.generate_mc_response(agent_id=self._agent_id, question=question, options=options, image_url=image_url)
 
 
 class SurveySession:
@@ -51,10 +51,10 @@ class SurveySession:
     def status(self) -> str:
         return self._status
 
-    async def generate_qual_response(self, question: str) -> QualGenerationResponse:
+    async def generate_qual_response(self, question: str, image_url: Optional[str] = None) -> QualGenerationResponse:
         """Generates a qualitative response within this survey session."""
         endpoint = f"sessions/{str(self._id)}/qual"
-        payload = QualGenerationRequest(question=question) 
+        payload = QualGenerationRequest(question=question, image_url=image_url) 
         return await self._client._request(
             "POST",
             endpoint,
@@ -62,10 +62,10 @@ class SurveySession:
             response_model=QualGenerationResponse
         )
 
-    async def generate_mc_response(self, question: str, options: List[str]) -> MCGenerationResponse:
+    async def generate_mc_response(self, question: str, options: List[str], image_url: Optional[str] = None) -> MCGenerationResponse:
         """Generates a multiple-choice response within this survey session."""
         endpoint = f"sessions/{str(self._id)}/mc"
-        payload = MCGenerationRequest(question=question, options=options) 
+        payload = MCGenerationRequest(question=question, options=options, image_url=image_url) 
         return await self._client._request(
             "POST",
             endpoint,

{simile-0.2.3 → simile-0.2.5}/simile.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: simile
-Version: 0.2.3
+Version: 0.2.5
 Summary: Package for interfacing with Simile AI agents for simulation
 Author-email: Simile AI <cqz@simile.ai>
 License: MIT
@@ -21,6 +21,7 @@ Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: httpx>=0.20.0
 Requires-Dist: pydantic>=2.0.0
+Requires-Dist: genagent>=0.2.3
 Dynamic: license-file
 
 # Simile API Python Client

{simile-0.2.3 → simile-0.2.5}/simile.egg-info/requires.txt

@@ -1,2 +1,3 @@
 httpx>=0.20.0
 pydantic>=2.0.0
+genagent>=0.2.3