simile 0.3.9__py3-none-any.whl → 0.3.11__py3-none-any.whl

simile/__init__.py

@@ -16,10 +16,10 @@ from .models import (
     ClosedGenerationResponse,
 )
 from .exceptions import (
-    SimileAPIError, 
-    SimileAuthenticationError, 
-    SimileNotFoundError, 
-    SimileBadRequestError
+    SimileAPIError,
+    SimileAuthenticationError,
+    SimileNotFoundError,
+    SimileBadRequestError,
 )
 
 __all__ = [

simile/client.py

@@ -169,10 +169,14 @@ class Simile:
         )
         return response_data
 
-    async def get_population_info(self, population_id: Union[str, uuid.UUID]) -> PopulationInfo:
+    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
+            "GET",
+            f"populations/info/{str(population_id)}",
+            response_model=PopulationInfo,
         )
         return response_data
 
@@ -259,10 +263,16 @@ class Simile:
         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
+        self,
+        agent_id: Union[str, uuid.UUID],
+        data_type: str,
+        content: Any,
+        metadata: Optional[Dict[str, Any]] = None,
     ) -> DataItem:
         """Creates a new data item for a specific agent."""
-        payload = CreateDataItemPayload(data_type=data_type, content=content, metadata=metadata)
+        payload = CreateDataItemPayload(
+            data_type=data_type, content=content, metadata=metadata
+        )
         response_data = await self._request(
             "POST",
             f"data_item/create/{str(agent_id)}",
@@ -319,6 +329,7 @@ class Simile:
         data_types: Optional[List[str]] = None,
         exclude_data_types: Optional[List[str]] = None,
         images: Optional[Dict[str, str]] = None,
+        reasoning: bool = False,
     ) -> AsyncGenerator[str, None]:
         """Streams an open response from an agent."""
         endpoint = f"/generation/open-stream/{str(agent_id)}"
@@ -327,12 +338,15 @@ class Simile:
             data_types=data_types,
             exclude_data_types=exclude_data_types,
             images=images,
+            reasoning=reasoning,
         )
 
         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:
+            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
@@ -352,7 +366,7 @@ class Simile:
     ) -> AsyncGenerator[str, None]:
         """Streams a closed response from an agent."""
         endpoint = f"/generation/closed-stream/{str(agent_id)}"
-        
+
         request_payload = {
             "question": question,
             "options": options,
@@ -373,7 +387,6 @@ class Simile:
                         else:
                             yield line.strip()
 
-
     async def generate_open_response(
         self,
         agent_id: uuid.UUID,
@@ -381,6 +394,7 @@ class Simile:
         data_types: Optional[List[str]] = None,
         exclude_data_types: Optional[List[str]] = None,
         images: Optional[Dict[str, str]] = None,
+        reasoning: bool = False,
     ) -> OpenGenerationResponse:
         """Generates an open response from an agent based on a question."""
         endpoint = f"/generation/open/{str(agent_id)}"
@@ -389,6 +403,7 @@ class Simile:
             data_types=data_types,
             exclude_data_types=exclude_data_types,
             images=images,
+            reasoning=reasoning,
         )
         response_data = await self._request(
             "POST",
@@ -406,6 +421,7 @@ class Simile:
         data_types: Optional[List[str]] = None,
         exclude_data_types: Optional[List[str]] = None,
         images: Optional[Dict[str, str]] = None,
+        reasoning: bool = False,
     ) -> ClosedGenerationResponse:
         """Generates a closed response from an agent."""
         endpoint = f"generation/closed/{str(agent_id)}"
@@ -415,6 +431,7 @@ class Simile:
             data_types=data_types,
             exclude_data_types=exclude_data_types,
             images=images,
+            reasoning=reasoning,
         )
         response_data = await self._request(
             "POST",

simile/exceptions.py

@@ -1,5 +1,6 @@
 class SimileAPIError(Exception):
     """Base exception for Simile API client errors."""
+
     def __init__(self, message: str, status_code: int = None, detail: str = None):
         super().__init__(message)
         self.status_code = status_code
@@ -15,17 +16,32 @@ class SimileAPIError(Exception):
 
 class SimileAuthenticationError(SimileAPIError):
     """Exception for authentication errors (e.g., invalid API key)."""
-    def __init__(self, message: str = "Authentication failed. Ensure API key is valid.", status_code: int = 401, detail: str = None):
+
+    def __init__(
+        self,
+        message: str = "Authentication failed. Ensure API key is valid.",
+        status_code: int = 401,
+        detail: str = None,
+    ):
         super().__init__(message, status_code, detail)
 
 
 class SimileNotFoundError(SimileAPIError):
     """Exception for resource not found errors (404)."""
-    def __init__(self, message: str = "Resource not found.", status_code: int = 404, detail: str = None):
+
+    def __init__(
+        self,
+        message: str = "Resource not found.",
+        status_code: int = 404,
+        detail: str = None,
+    ):
         super().__init__(message, status_code, detail)
 
 
 class SimileBadRequestError(SimileAPIError):
     """Exception for bad request errors (400)."""
-    def __init__(self, message: str = "Bad request.", status_code: int = 400, detail: str = None):
+
+    def __init__(
+        self, message: str = "Bad request.", status_code: int = 400, detail: str = None
+    ):
         super().__init__(message, status_code, detail)

simile/models.py

@@ -78,11 +78,13 @@ class OpenGenerationRequest(BaseModel):
     images: Optional[Dict[str, str]] = (
         None  # Dict of {description: url} for multiple images
     )
+    reasoning: bool = False
 
 
 class OpenGenerationResponse(BaseModel):
     question: str
     answer: str
+    reasoning: Optional[str] = ""
 
 
 class ClosedGenerationRequest(BaseModel):
@@ -91,12 +93,14 @@ class ClosedGenerationRequest(BaseModel):
     data_types: Optional[List[str]] = None
     exclude_data_types: Optional[List[str]] = None
     images: Optional[Dict[str, str]] = None
+    reasoning: bool = False
 
 
 class ClosedGenerationResponse(BaseModel):
     question: str
     options: List[str]
     response: str
+    reasoning: Optional[str] = ""
 
 
 class AddContextRequest(BaseModel):
@@ -166,9 +170,7 @@ class ClosedQuestionTurn(BaseTurn):
         if not v:
             raise ValueError("Closed questions must have at least one option")
         if len(v) < 2:
-            raise ValueError(
-                "Closed questions should have at least two options"
-            )
+            raise ValueError("Closed questions should have at least two options")
         return v
 
     @validator("llm_response")
@@ -183,9 +185,7 @@ class ClosedQuestionTurn(BaseTurn):
 
 
 # Union type for all possible turn types
-SurveySessionTurn = Union[
-    ContextTurn, ImageTurn, OpenQuestionTurn, ClosedQuestionTurn
-]
+SurveySessionTurn = Union[ContextTurn, ImageTurn, OpenQuestionTurn, ClosedQuestionTurn]
 
 
 class SurveySessionCreateResponse(BaseModel):

simile/resources.py

@@ -89,7 +89,7 @@ class SurveySession:
     async def get_details(self) -> SurveySessionDetailResponse:
         """Retrieves detailed information about this survey session including typed conversation history."""
         return await self._client.get_survey_session_details(self._id)
-    
+
     async def view(self) -> SurveySessionDetailResponse:
         """Alias for get_details() - retrieves all turns in this session."""
         return await self.get_details()
@@ -100,6 +100,7 @@ class SurveySession:
         data_types: Optional[List[str]] = None,
         exclude_data_types: Optional[List[str]] = None,
         images: Optional[Dict[str, str]] = None,
+        reasoning: bool = False,
     ) -> OpenGenerationResponse:
         """Generates an open response within this survey session."""
         endpoint = f"sessions/{str(self._id)}/open"
@@ -108,6 +109,7 @@ class SurveySession:
             data_types=data_types,
             exclude_data_types=exclude_data_types,
             images=images,
+            reasoning=reasoning,
         )
         return await self._client._request(
             "POST",
@@ -123,6 +125,7 @@ class SurveySession:
         data_types: Optional[List[str]] = None,
         exclude_data_types: Optional[List[str]] = None,
         images: Optional[Dict[str, str]] = None,
+        reasoning: bool = False,
     ) -> ClosedGenerationResponse:
         """Generates a closed response within this survey session."""
         endpoint = f"sessions/{str(self._id)}/closed"
@@ -132,6 +135,7 @@ class SurveySession:
             data_types=data_types,
             exclude_data_types=exclude_data_types,
             images=images,
+            reasoning=reasoning,
         )
         return await self._client._request(
             "POST",
@@ -157,14 +161,14 @@ class SurveySession:
         timestamp: str,
     ) -> Dict:
         """Adds context to this session with a specific timestamp.
-        
+
         This is a lower-level method that allows specifying when the context was added.
         For normal use, prefer the add_context() method.
-        
+
         Args:
             context_text: The context text to add
             timestamp: ISO timestamp of when this interaction occurred
-            
+
         Returns:
             Dictionary with success status and the added turn details
         """
@@ -174,7 +178,7 @@ class SurveySession:
             "context_text": context_text,
             "timestamp": timestamp,
         }
-            
+
         return await self._client._request(
             "POST",
             endpoint,
@@ -188,11 +192,11 @@ class SurveySession:
         timestamp: Optional[str] = None,
     ) -> Dict:
         """Adds images to the session's conversation history.
-        
+
         Args:
             images: Dictionary mapping image descriptions to URLs
             timestamp: Optional ISO timestamp of when this interaction occurred
-            
+
         Returns:
             Dictionary with success status and the added turn details
         """
@@ -203,7 +207,7 @@ class SurveySession:
         }
         if timestamp:
             payload["timestamp"] = timestamp
-            
+
         return await self._client._request(
             "POST",
             endpoint,
@@ -218,12 +222,12 @@ class SurveySession:
         timestamp: Optional[str] = None,
     ) -> Dict:
         """Adds an open question-answer pair to the session's history.
-        
+
         Args:
             question: The open question text
             response: The response that was given
             timestamp: Optional ISO timestamp of when this interaction occurred
-            
+
         Returns:
             Dictionary with success status and the added turn details
         """
@@ -235,7 +239,7 @@ class SurveySession:
         }
         if timestamp:
             payload["timestamp"] = timestamp
-            
+
         return await self._client._request(
             "POST",
             endpoint,
@@ -258,13 +262,13 @@ class SurveySession:
         timestamp: Optional[str] = None,
     ) -> Dict:
         """Adds a closed question-answer pair to the session's history.
-        
+
         Args:
             question: The closed question text
             options: List of answer options
             response: The option that was selected
             timestamp: Optional ISO timestamp of when this interaction occurred
-            
+
         Returns:
             Dictionary with success status and the added turn details
         """
@@ -277,40 +281,39 @@ class SurveySession:
         }
         if timestamp:
             payload["timestamp"] = timestamp
-            
+
         return await self._client._request(
             "POST",
             endpoint,
             json=payload,
             response_model=None,  # Return raw dict since we don't have a specific model
         )
-    
-    
+
     async def fork(self, turn_index: int) -> "SurveySession":
         """Fork this session at a specific turn.
-        
+
         Creates a new session with the same agent and copies turns from this session
         up to and including the specified turn index.
-        
+
         Args:
             turn_index: The 0-based index of the last turn to include in the fork
-            
+
         Returns:
             A new SurveySession object representing the forked session
-            
+
         Raises:
             Simile.APIError: If the API request fails
         """
         endpoint = f"sessions/{str(self._id)}/fork"
         params = {"turn_index": turn_index}
-        
+
         response = await self._client._request(
             "POST",
             endpoint,
             params=params,
             response_model=SurveySessionCreateResponse,
         )
-        
+
         # Create a new SurveySession instance from the response
         return SurveySession(
             id=response.id,

{simile-0.3.9.dist-info → simile-0.3.11.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: simile
-Version: 0.3.9
+Version: 0.3.11
 Summary: Package for interfacing with Simile AI agents for simulation
 Author-email: Simile AI <cqz@simile.ai>
 License: MIT
@@ -46,3 +46,18 @@ from simile import Simile
 
 client = Simile(api_key="your_api_key")
 ```
+
+## Publishing
+
+First, bump the version in `pyproject.toml`. Then, create the distribution files:
+```bash
+python3 -m build
+```
+
+Afterwards, use [Twine](https://pypi.org/project/twine/) to upload the package:
+```bash
+pip install twine
+twine upload dist/*
+```
+
+If you need the PyPI credentials, please ask Carolyn or Chris.

simile-0.3.11.dist-info/RECORD

@@ -0,0 +1,11 @@
+simile/__init__.py,sha256=D88zktKWLzF2EgbNm743Ype_p4s7xXWvNEQBr6mRsCI,1034
+simile/auth_client.py,sha256=ICImmaA5fZX9ADbIPIUh4RED3hBZvLf3XSiaqELDAME,7923
+simile/client.py,sha256=zOg38vzTmjz6tY4j5wWaU60823u8sUZBMNtwkun4stM,16645
+simile/exceptions.py,sha256=Q1lbfwR7mEn_LYmwjAnsMc8BW79JNPvmCmVoPibYisU,1502
+simile/models.py,sha256=NvRWSs7jmN3CelzVQeNBD2ELH0z5LOE3Pm-45nGc7fU,5667
+simile/resources.py,sha256=D6K3crR4tb3CAFH7mF-y536pJwJ3rmNDCtm7Y2vMtsc,10347
+simile-0.3.11.dist-info/licenses/LICENSE,sha256=tpxX3bpODfyOQVyEM6kCMvPHFCpkjFDj0AICRqKqOFA,1066
+simile-0.3.11.dist-info/METADATA,sha256=sOqNFOQvJkqh7uTo7nNpHxIyLW3wbEIjqWQxLAmLwZA,1599
+simile-0.3.11.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+simile-0.3.11.dist-info/top_level.txt,sha256=41lJneubAG4-ZOAs5qn7iDtDb-MDxa6DdvgBKwNX84M,7
+simile-0.3.11.dist-info/RECORD,,

simile-0.3.9.dist-info/RECORD

@@ -1,11 +0,0 @@
-simile/__init__.py,sha256=GgbhCv5G5F8X_DIYw0H72D8fRuVHW7SLezGJykALpd0,1036
-simile/auth_client.py,sha256=ICImmaA5fZX9ADbIPIUh4RED3hBZvLf3XSiaqELDAME,7923
-simile/client.py,sha256=bV2ih_z6c7ztZooUJetPTTTlbIFOmgx5F6TUruIKA1M,16332
-simile/exceptions.py,sha256=-rJ3KZcpvNRi9JXbDpxWDSL2lU1mEJX2piwYRZvhKmg,1406
-simile/models.py,sha256=BrrKfGbvjGiKXt3FvaIKLdrqCbNK_7mJhMRBPxsNQLs,5579
-simile/resources.py,sha256=Juoa32SlN8LdvzCFaobq7uV2itmcwbk5dM_mlc4bCVk,10420
-simile-0.3.9.dist-info/licenses/LICENSE,sha256=tpxX3bpODfyOQVyEM6kCMvPHFCpkjFDj0AICRqKqOFA,1066
-simile-0.3.9.dist-info/METADATA,sha256=KcED9Djr77xkU53CBRq0hpr2xtsudoi0SS62bCqhfSk,1276
-simile-0.3.9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-simile-0.3.9.dist-info/top_level.txt,sha256=41lJneubAG4-ZOAs5qn7iDtDb-MDxa6DdvgBKwNX84M,7
-simile-0.3.9.dist-info/RECORD,,