simile 0.2.9__py3-none-any.whl → 0.2.10__py3-none-any.whl

simile/client.py

@@ -137,7 +137,7 @@ class Simile:
         self, session_id: Union[str, uuid.UUID]
     ) -> SurveySession:
         """Resume an existing survey session by ID and return a SurveySession object."""
-        session_details = await self.get_survey_session_details(session_id)
+        session_details = await self.get_survey_session(session_id)
 
         if session_details.status == "closed":
             raise ValueError(f"Session {session_id} is already closed")
@@ -280,11 +280,14 @@ class Simile:
         return response_data
 
     async def generate_qual_response(
-        self, agent_id: uuid.UUID, question: str, image_url: Optional[str] = None
+        self,
+        agent_id: uuid.UUID,
+        question: str,
+        images: Optional[Dict[str, 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, image_url=image_url)
+        request_payload = QualGenerationRequest(question=question, images=images)
         response_data = await self._request(
             "POST",
             endpoint,
@@ -298,12 +301,12 @@ class Simile:
         agent_id: uuid.UUID,
         question: str,
         options: List[str],
-        image_url: Optional[str] = None,
+        images: Optional[Dict[str, 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, image_url=image_url
+            question=question, options=options, images=images
         )
         response_data = await self._request(
             "POST",

simile/models.py

@@ -30,6 +30,7 @@ class Agent(BaseModel):
     updated_at: datetime
     data_items: List[DataItem] = Field(default_factory=list)
 
+
 class CreatePopulationPayload(BaseModel):
     name: str
     description: Optional[str] = None
@@ -62,26 +63,32 @@ class DeletionResponse(BaseModel):
 # --- Generation Operation Models ---
 class QualGenerationRequest(BaseModel):
     question: str
-    image_url: Optional[str] = None  # For backward compatibility
-    images: Optional[Dict[str, str]] = None  # Dict of {description: url} for multiple images
+    images: Optional[Dict[str, str]] = (
+        None  # Dict of {description: url} for multiple images
+    )
+
 
 class QualGenerationResponse(BaseModel):
     question: str
     answer: str
 
+
 class MCGenerationRequest(BaseModel):
     question: str
     options: List[str]
-    image_url: Optional[str] = None
+    images: Optional[Dict[str, str]] = None
+
 
 class MCGenerationResponse(BaseModel):
     question: str
     options: List[str]
     chosen_option: str
 
+
 class AddContextRequest(BaseModel):
     context: str
 
+
 class AddContextResponse(BaseModel):
     message: str
     session_id: uuid.UUID
@@ -90,6 +97,7 @@ class AddContextResponse(BaseModel):
 # --- Survey Session Models ---
 class TurnType(str, Enum):
     """Enum for different types of conversation turns."""
+
     CONTEXT = "context"
     IMAGE = "image"
     QUALITATIVE_QUESTION = "qualitative_question"
@@ -98,101 +106,111 @@ class TurnType(str, Enum):
 
 class BaseTurn(BaseModel):
     """Base model for all conversation turns."""
+
     timestamp: datetime = Field(default_factory=lambda: datetime.now())
     type: TurnType
-    
+
     class Config:
         use_enum_values = True
 
 
 class ContextTurn(BaseTurn):
     """A context turn that provides background information."""
+
     type: Literal[TurnType.CONTEXT] = TurnType.CONTEXT
     user_context: str
 
 
 class ImageTurn(BaseTurn):
     """A standalone image turn (e.g., for context or reference)."""
+
     type: Literal[TurnType.IMAGE] = TurnType.IMAGE
-    image_url: str
+    images: Dict[str, str]
     caption: Optional[str] = None
 
 
 class QualitativeQuestionTurn(BaseTurn):
     """A qualitative question-answer turn."""
+
     type: Literal[TurnType.QUALITATIVE_QUESTION] = TurnType.QUALITATIVE_QUESTION
     user_question: str
-    user_image_url: Optional[str] = None  # For backward compatibility
-    user_images: Optional[Dict[str, str]] = None  # Dict of {description: url} for multiple images
+    user_images: Optional[Dict[str, str]] = None
     llm_response: Optional[str] = None
 
 
 class MultipleChoiceQuestionTurn(BaseTurn):
     """A multiple choice question-answer turn."""
+
     type: Literal[TurnType.MULTIPLE_CHOICE_QUESTION] = TurnType.MULTIPLE_CHOICE_QUESTION
     user_question: str
     user_options: List[str]
-    user_image_url: Optional[str] = None
+    user_images: Optional[Dict[str, str]] = None
     llm_chosen_option: Optional[str] = None
-    
-    @validator('user_options')
+
+    @validator("user_options")
     def validate_options(cls, v):
         if not v:
             raise ValueError("Multiple choice questions must have at least one option")
         if len(v) < 2:
-            raise ValueError("Multiple choice questions should have at least two options")
+            raise ValueError(
+                "Multiple choice questions should have at least two options"
+            )
         return v
-    
-    @validator('llm_chosen_option')
+
+    @validator("llm_chosen_option")
     def validate_chosen_option(cls, v, values):
-        if v is not None and 'user_options' in values and v not in values['user_options']:
+        if (
+            v is not None
+            and "user_options" in values
+            and v not in values["user_options"]
+        ):
             raise ValueError(f"Chosen option '{v}' must be one of the provided options")
         return v
 
 
 # Union type for all possible turn types
-SurveySessionTurn = Union[ContextTurn, ImageTurn, QualitativeQuestionTurn, MultipleChoiceQuestionTurn]
+SurveySessionTurn = Union[
+    ContextTurn, ImageTurn, QualitativeQuestionTurn, MultipleChoiceQuestionTurn
+]
 
 
 class SurveySessionCreateResponse(BaseModel):
-    id: uuid.UUID # Session ID
+    id: uuid.UUID  # Session ID
     agent_id: uuid.UUID
     created_at: datetime
     status: str
-    
+
 
 class SurveySessionDetailResponse(BaseModel):
     """Detailed survey session response with typed conversation turns."""
+
     id: uuid.UUID
     agent_id: uuid.UUID
     created_at: datetime
     updated_at: datetime
     status: str
     conversation_history: List[SurveySessionTurn] = Field(default_factory=list)
-    
+
     class Config:
-        json_encoders = {
-            datetime: lambda v: v.isoformat()
-        }
+        json_encoders = {datetime: lambda v: v.isoformat()}
 
 
 class SurveySessionListItemResponse(BaseModel):
     """Summary response for listing survey sessions."""
+
     id: uuid.UUID
     agent_id: uuid.UUID
     created_at: datetime
     updated_at: datetime
     status: str
     turn_count: int = Field(description="Number of turns in conversation history")
-    
+
     class Config:
-        json_encoders = {
-            datetime: lambda v: v.isoformat()
-        }
+        json_encoders = {datetime: lambda v: v.isoformat()}
 
 
 class SurveySessionCloseResponse(BaseModel):
-    id: uuid.UUID # Session ID
-    status: str 
-    updated_at: datetime 
-    message: Optional[str] = None
+    id: uuid.UUID  # Session ID
+    status: str
+    updated_at: datetime
+    message: Optional[str] = None

simile/resources.py

@@ -9,15 +9,17 @@ from .models import (
     SurveySessionCloseResponse,
     AddContextRequest,
     AddContextResponse,
-    SurveySessionDetailResponse
+    SurveySessionDetailResponse,
 )
 
 if TYPE_CHECKING:
     from .client import Simile
 
+
 class Agent:
     """Represents an agent and provides methods for interacting with it directly."""
-    def __init__(self, agent_id: uuid.UUID, client: 'Simile'):
+
+    def __init__(self, agent_id: uuid.UUID, client: "Simile"):
         self._agent_id = agent_id
         self._client = client
 
@@ -25,18 +27,32 @@ class Agent:
     def id(self) -> uuid.UUID:
         return self._agent_id
 
-    async def generate_qual_response(self, question: str, image_url: Optional[str] = None) -> QualGenerationResponse:
+    async def generate_qual_response(
+        self, question: str, images: Optional[Dict[str, 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, image_url=image_url)
+        return await self._client.generate_qual_response(
+            agent_id=self._agent_id, question=question, images=images
+        )
 
-    async def generate_mc_response(self, question: str, options: List[str], image_url: Optional[str] = None) -> MCGenerationResponse:
+    async def generate_mc_response(
+        self, question: str, options: List[str], images: Optional[Dict[str, 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, image_url=image_url)
+        return await self._client.generate_mc_response(
+            agent_id=self._agent_id,
+            question=question,
+            options=options,
+            images=images,
+        )
 
 
 class SurveySession:
     """Represents an active survey session with an agent, allowing for contextual multi-turn generation."""
-    def __init__(self, id: uuid.UUID, agent_id: uuid.UUID, status: str, client: 'Simile'):
+
+    def __init__(
+        self, id: uuid.UUID, agent_id: uuid.UUID, status: str, client: "Simile"
+    ):
         self._id = id
         self._agent_id = agent_id
         self._status = status
@@ -56,28 +72,34 @@ 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(self._id)
+        return await self._client.get_survey_session_details(self._id)
 
-    async def generate_qual_response(self, question: str, image_url: Optional[str] = None, images: Optional[Dict[str, str]] = None) -> QualGenerationResponse:
+    async def generate_qual_response(
+        self,
+        question: str,
+        images: Optional[Dict[str, str]] = None,
+    ) -> QualGenerationResponse:
         """Generates a qualitative response within this survey session."""
         endpoint = f"sessions/{str(self._id)}/qual"
-        payload = QualGenerationRequest(question=question, image_url=image_url, images=images) 
+        payload = QualGenerationRequest(question=question, images=images)
         return await self._client._request(
             "POST",
             endpoint,
-            json=payload.model_dump(), 
-            response_model=QualGenerationResponse
+            json=payload.model_dump(),
+            response_model=QualGenerationResponse,
         )
 
-    async def generate_mc_response(self, question: str, options: List[str], image_url: Optional[str] = None) -> MCGenerationResponse:
+    async def generate_mc_response(
+        self, question: str, options: List[str], images: Optional[Dict[str, 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, image_url=image_url) 
+        payload = MCGenerationRequest(question=question, options=options, images=images)
         return await self._client._request(
             "POST",
             endpoint,
-            json=payload.model_dump(), 
-            response_model=MCGenerationResponse
+            json=payload.model_dump(),
+            response_model=MCGenerationResponse,
         )
 
     async def add_context(self, ctx: str) -> AddContextResponse:
@@ -88,14 +110,12 @@ class SurveySession:
             "POST",
             endpoint,
             json=payload.model_dump(),
-            response_model=AddContextResponse
+            response_model=AddContextResponse,
         )
 
     async def close(self) -> SurveySessionCloseResponse:
         """Closes this survey session on the server."""
         endpoint = f"sessions/{str(self._id)}/close"
         return await self._client._request(
-            "POST",
-            endpoint,
-            response_model=SurveySessionCloseResponse
+            "POST", endpoint, response_model=SurveySessionCloseResponse
         )

{simile-0.2.9.dist-info → simile-0.2.10.dist-info}/METADATA

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

simile-0.2.10.dist-info/RECORD

@@ -0,0 +1,10 @@
+simile/__init__.py,sha256=2OZ1LQIkAEtSs0EI5Fzlg7QGKduCgCe_nTh9FfIuHlQ,865
+simile/client.py,sha256=C7ZA9UAYjOKT5jc1nzO3JOgPwq0GK8gp12xE3ToN-0A,11855
+simile/exceptions.py,sha256=-rJ3KZcpvNRi9JXbDpxWDSL2lU1mEJX2piwYRZvhKmg,1406
+simile/models.py,sha256=1o28eOo53dZCrSSWT2myM4WOghyBhStBm2KRHBxX_Ho,5260
+simile/resources.py,sha256=Mt0TvipiseVtkiB7kLssAa1m4JVIrtmXG6rFd7HXHkM,4089
+simile-0.2.10.dist-info/licenses/LICENSE,sha256=tpxX3bpODfyOQVyEM6kCMvPHFCpkjFDj0AICRqKqOFA,1066
+simile-0.2.10.dist-info/METADATA,sha256=pGJJK0dcFMikY6jat6hGYsO_3SkBUp8x1dqNcfeRiio,1277
+simile-0.2.10.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+simile-0.2.10.dist-info/top_level.txt,sha256=41lJneubAG4-ZOAs5qn7iDtDb-MDxa6DdvgBKwNX84M,7
+simile-0.2.10.dist-info/RECORD,,

simile-0.2.9.dist-info/RECORD

@@ -1,10 +0,0 @@
-simile/__init__.py,sha256=2OZ1LQIkAEtSs0EI5Fzlg7QGKduCgCe_nTh9FfIuHlQ,865
-simile/client.py,sha256=UvheMza-03UJf-hTOMDEDzcuexq2kWd3nD8mqw2V0tY,11834
-simile/exceptions.py,sha256=-rJ3KZcpvNRi9JXbDpxWDSL2lU1mEJX2piwYRZvhKmg,1406
-simile/models.py,sha256=gM6QqVySJe6sG-X7deDcpQYo0JMhfxtfZtBHFP-ZqSI,5376
-simile/resources.py,sha256=KPSmcX6cgVfOCPvL1eD5zc9fvOw5n4oe6pOieuzp8zg,3982
-simile-0.2.9.dist-info/licenses/LICENSE,sha256=tpxX3bpODfyOQVyEM6kCMvPHFCpkjFDj0AICRqKqOFA,1066
-simile-0.2.9.dist-info/METADATA,sha256=DCAjpe8YeVui8WbavtdtZFRuawtljXv3_xV4o3RR51Y,1276
-simile-0.2.9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-simile-0.2.9.dist-info/top_level.txt,sha256=41lJneubAG4-ZOAs5qn7iDtDb-MDxa6DdvgBKwNX84M,7
-simile-0.2.9.dist-info/RECORD,,