simile 0.2.14__py3-none-any.whl → 0.2.15__py3-none-any.whl

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/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/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.dist-info → simile-0.2.15.dist-info}/METADATA

@@ -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.15.dist-info/RECORD

@@ -0,0 +1,11 @@
+simile/__init__.py,sha256=qtWFBsMK2F1BNZ0V07fEb56fcuFOm48QcA4ne_s9t44,920
+simile/auth_client.py,sha256=ICImmaA5fZX9ADbIPIUh4RED3hBZvLf3XSiaqELDAME,7923
+simile/client.py,sha256=KumgPioNjR6UYNIT1MJipw3bMArPPNFnrluvHkw1eSw,12279
+simile/exceptions.py,sha256=-rJ3KZcpvNRi9JXbDpxWDSL2lU1mEJX2piwYRZvhKmg,1406
+simile/models.py,sha256=bdtXGM4K4x_bUTXRMf6DzGEtJRMgnofE9yy3OlA0qEQ,5448
+simile/resources.py,sha256=HWi2Fp32UBMqz7TQ9aA66Xs6mex7Olyz44TmfDOj_hc,6519
+simile-0.2.15.dist-info/licenses/LICENSE,sha256=tpxX3bpODfyOQVyEM6kCMvPHFCpkjFDj0AICRqKqOFA,1066
+simile-0.2.15.dist-info/METADATA,sha256=bZ6ThCzUNab_wKM7TJJ2pkoNtwe3_dqYJyIOLAS4fMg,1277
+simile-0.2.15.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+simile-0.2.15.dist-info/top_level.txt,sha256=41lJneubAG4-ZOAs5qn7iDtDb-MDxa6DdvgBKwNX84M,7
+simile-0.2.15.dist-info/RECORD,,

simile-0.2.14.dist-info/RECORD

@@ -1,11 +0,0 @@
-simile/__init__.py,sha256=qtWFBsMK2F1BNZ0V07fEb56fcuFOm48QcA4ne_s9t44,920
-simile/auth_client.py,sha256=ICImmaA5fZX9ADbIPIUh4RED3hBZvLf3XSiaqELDAME,7923
-simile/client.py,sha256=S1YhKNDhDS4KqrP95jNvRd-WhdLyG-ObhQtKfibA4rU,12062
-simile/exceptions.py,sha256=-rJ3KZcpvNRi9JXbDpxWDSL2lU1mEJX2piwYRZvhKmg,1406
-simile/models.py,sha256=1cVmaOdK90cyDuIiP6Ox27NJEKMTtx0LyNos_CzJ-UA,5354
-simile/resources.py,sha256=Tcj1o8K9lLtUVFvWSe3jig9OcTsOSVGtmK1BHB-WI9w,5686
-simile-0.2.14.dist-info/licenses/LICENSE,sha256=tpxX3bpODfyOQVyEM6kCMvPHFCpkjFDj0AICRqKqOFA,1066
-simile-0.2.14.dist-info/METADATA,sha256=ffR2H3McJKssQex2URy0nFlXH7fDLIw_lLqz9W7WTWI,1277
-simile-0.2.14.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-simile-0.2.14.dist-info/top_level.txt,sha256=41lJneubAG4-ZOAs5qn7iDtDb-MDxa6DdvgBKwNX84M,7
-simile-0.2.14.dist-info/RECORD,,