simile 0.2.14__tar.gz → 0.2.15__tar.gz

{simile-0.2.14 → simile-0.2.15}/PKG-INFO

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

{simile-0.2.14 → simile-0.2.15}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "simile"
-version = "0.2.14"
+version = "0.2.15"
 authors = [
   { name="Simile AI", email="cqz@simile.ai" },
 ]

{simile-0.2.14 → simile-0.2.15}/simile/client.py

@@ -308,12 +308,18 @@ class Simile:
         agent_id: uuid.UUID,
         question: str,
         options: List[str],
-        image_url: Optional[str] = None,
+        data_types: Optional[List[str]] = None,
+        exclude_data_types: Optional[List[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,
+            data_types=data_types,
+            exclude_data_types=exclude_data_types,
+            images=images,
         )
         response_data = await self._request(
             "POST",

{simile-0.2.14 → simile-0.2.15}/simile/models.py

@@ -78,6 +78,8 @@ class QualGenerationResponse(BaseModel):
 class MCGenerationRequest(BaseModel):
     question: str
     options: List[str]
+    data_types: Optional[List[str]] = None
+    exclude_data_types: Optional[List[str]] = None
     images: Optional[Dict[str, str]] = None
 
 

{simile-0.2.14 → simile-0.2.15}/simile/resources.py

@@ -28,21 +28,36 @@ class Agent:
         return self._agent_id
 
     async def generate_qual_response(
-        self, question: str, images: Optional[Dict[str, str]] = None
+        self,
+        question: str,
+        data_types: Optional[List[str]] = None,
+        exclude_data_types: Optional[List[str]] = None,
+        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, images=images
+            agent_id=self._agent_id,
+            question=question,
+            data_types=data_types,
+            exclude_data_types=exclude_data_types,
+            images=images,
         )
 
     async def generate_mc_response(
-        self, question: str, options: List[str], images: Optional[Dict[str, str]] = None
+        self,
+        question: str,
+        options: List[str],
+        data_types: Optional[List[str]] = None,
+        exclude_data_types: Optional[List[str]] = None,
+        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,
+            data_types=data_types,
+            exclude_data_types=exclude_data_types,
             images=images,
         )
 
@@ -77,14 +92,16 @@ class SurveySession:
     async def generate_qual_response(
         self,
         question: str,
+        data_types: Optional[List[str]] = None,
+        exclude_data_types: Optional[List[str]] = None,
         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,
-            data_types=None,
-            exclude_data_types=None,
+            data_types=data_types,
+            exclude_data_types=exclude_data_types,
             images=images,
         )
         return await self._client._request(
@@ -95,11 +112,22 @@ class SurveySession:
         )
 
     async def generate_mc_response(
-        self, question: str, options: List[str], images: Optional[Dict[str, str]] = None
+        self,
+        question: str,
+        options: List[str],
+        data_types: Optional[List[str]] = None,
+        exclude_data_types: Optional[List[str]] = None,
+        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, images=images)
+        payload = MCGenerationRequest(
+            question=question,
+            options=options,
+            data_types=data_types,
+            exclude_data_types=exclude_data_types,
+            images=images,
+        )
         return await self._client._request(
             "POST",
             endpoint,

{simile-0.2.14 → simile-0.2.15}/simile.egg-info/PKG-INFO

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