simile 0.4.1__py3-none-any.whl → 0.4.3__py3-none-any.whl

simile/client.py

@@ -346,6 +346,8 @@ class Simile:
         exclude_data_types: Optional[List[str]] = None,
         images: Optional[Dict[str, str]] = None,
         reasoning: bool = False,
+        evidence: bool = False,
+        confidence: bool = False,
         memory_stream: Optional[MemoryStream] = None,
     ) -> AsyncGenerator[str, None]:
         """Streams an open response from an agent."""
@@ -356,6 +358,8 @@ class Simile:
             exclude_data_types=exclude_data_types,
             images=images,
             reasoning=reasoning,
+            evidence=evidence,
+            confidence=confidence,
         )
 
         url = self.base_url + endpoint  # assuming self.base_url is defined
@@ -412,6 +416,8 @@ class Simile:
         exclude_data_types: Optional[List[str]] = None,
         images: Optional[Dict[str, str]] = None,
         reasoning: bool = False,
+        evidence: bool = False,
+        confidence: bool = False,
         memory_stream: Optional[MemoryStream] = None,
         use_memory: Optional[Union[str, uuid.UUID]] = None,  # Session ID to load memory from
         exclude_memory_ids: Optional[List[str]] = None,  # Study/question IDs to exclude
@@ -431,15 +437,6 @@ class Simile:
             exclude_memory_ids: Study/question IDs to exclude from loaded memory
             save_memory: Session ID to automatically save response to memory
         """
-        # If use_memory is provided and no explicit memory_stream, load it
-        if use_memory and not memory_stream:
-            memory_stream = await self.get_memory(
-                session_id=use_memory,
-                agent_id=agent_id,
-                exclude_study_ids=exclude_memory_ids,
-                use_memory=True
-            )
-        
         endpoint = f"/generation/open/{str(agent_id)}"
         # Build request payload directly as dict to avoid serialization issues
         request_payload = {
@@ -448,8 +445,20 @@ class Simile:
             "exclude_data_types": exclude_data_types,
             "images": images,
             "reasoning": reasoning,
+            "evidence": evidence,
+            "confidence": confidence,
         }
         
+        # Pass memory parameters to API for server-side handling
+        if use_memory:
+            request_payload["use_memory"] = str(use_memory)
+            if exclude_memory_ids:
+                request_payload["exclude_memory_ids"] = exclude_memory_ids
+        
+        if save_memory:
+            request_payload["save_memory"] = str(save_memory)
+        
+        # Only include explicit memory_stream if provided directly
         if memory_stream:
             request_payload["memory_stream"] = memory_stream.to_dict()
         
@@ -460,26 +469,8 @@ class Simile:
             response_model=OpenGenerationResponse,
         )
         
-        # If save_memory is provided, save the response
-        if save_memory and response_data:
-            from .models import OpenQuestionMemoryTurn
-            
-            memory_turn = OpenQuestionMemoryTurn(
-                user_question=question,
-                user_images=images,
-                llm_response=response_data.answer,
-                llm_reasoning=response_data.reasoning if reasoning else None
-            )
-            
-            await self.save_memory(
-                agent_id=agent_id,
-                response=response_data.answer,
-                session_id=save_memory,
-                memory_turn=memory_turn.to_dict(),
-                memory_stream_used=memory_stream.to_dict() if memory_stream else None,
-                reasoning=response_data.reasoning if reasoning else None,
-                metadata={"question_type": "open"}
-            )
+        # Don't save memory here - API should handle it when save_memory is passed
+        # Memory saving is now handled server-side for better performance
         
         return response_data
 
@@ -492,6 +483,8 @@ class Simile:
         exclude_data_types: Optional[List[str]] = None,
         images: Optional[Dict[str, str]] = None,
         reasoning: bool = False,
+        evidence: bool = False,
+        confidence: bool = False,
         memory_stream: Optional[MemoryStream] = None,
         use_memory: Optional[Union[str, uuid.UUID]] = None,  # Session ID to load memory from
         exclude_memory_ids: Optional[List[str]] = None,  # Study/question IDs to exclude
@@ -512,15 +505,6 @@ class Simile:
             exclude_memory_ids: Study/question IDs to exclude from loaded memory
             save_memory: Session ID to automatically save response to memory
         """
-        # If use_memory is provided and no explicit memory_stream, load it
-        if use_memory and not memory_stream:
-            memory_stream = await self.get_memory(
-                session_id=use_memory,
-                agent_id=agent_id,
-                exclude_study_ids=exclude_memory_ids,
-                use_memory=True
-            )
-        
         endpoint = f"generation/closed/{str(agent_id)}"
         # Build request payload directly as dict to avoid serialization issues
         request_payload = {
@@ -530,8 +514,20 @@ class Simile:
             "exclude_data_types": exclude_data_types,
             "images": images,
             "reasoning": reasoning,
+            "evidence": evidence,
+            "confidence": confidence,
         }
         
+        # Pass memory parameters to API for server-side handling
+        if use_memory:
+            request_payload["use_memory"] = str(use_memory)
+            if exclude_memory_ids:
+                request_payload["exclude_memory_ids"] = exclude_memory_ids
+        
+        if save_memory:
+            request_payload["save_memory"] = str(save_memory)
+        
+        # Only include explicit memory_stream if provided directly
         if memory_stream:
             request_payload["memory_stream"] = memory_stream.to_dict()
         
@@ -542,27 +538,8 @@ class Simile:
             response_model=ClosedGenerationResponse,
         )
         
-        # If save_memory is provided, save the response
-        if save_memory and response_data:
-            from .models import ClosedQuestionMemoryTurn
-            
-            memory_turn = ClosedQuestionMemoryTurn(
-                user_question=question,
-                user_options=options,
-                user_images=images,
-                llm_response=response_data.response,
-                llm_reasoning=response_data.reasoning if reasoning else None
-            )
-            
-            await self.save_memory(
-                agent_id=agent_id,
-                response=response_data.response,
-                session_id=save_memory,
-                memory_turn=memory_turn.to_dict(),
-                memory_stream_used=memory_stream.to_dict() if memory_stream else None,
-                reasoning=response_data.reasoning if reasoning else None,
-                metadata={"question_type": "closed", "options": options}
-            )
+        # Don't save memory here - API should handle it when save_memory is passed
+        # Memory saving is now handled server-side for better performance
         
         return response_data
 

simile/models.py

@@ -80,6 +80,8 @@ class OpenGenerationRequest(BaseModel):
         None  # Dict of {description: url} for multiple images
     )
     reasoning: bool = False
+    evidence: bool = False
+    confidence: bool = False
     memory_stream: Optional["MemoryStream"] = None
 
 
@@ -87,6 +89,8 @@ class OpenGenerationResponse(BaseModel):
     question: str
     answer: str
     reasoning: Optional[str] = ""
+    evidence: Optional[str] = ""
+    confidence: Optional[float] = None
 
 
 class ClosedGenerationRequest(BaseModel):
@@ -96,6 +100,8 @@ class ClosedGenerationRequest(BaseModel):
     exclude_data_types: Optional[List[str]] = None
     images: Optional[Dict[str, str]] = None
     reasoning: bool = False
+    evidence: bool = False
+    confidence: bool = False
     memory_stream: Optional["MemoryStream"] = None
 
 
@@ -104,6 +110,8 @@ class ClosedGenerationResponse(BaseModel):
     options: List[str]
     response: str
     reasoning: Optional[str] = ""
+    evidence: Optional[str] = ""
+    confidence: Optional[float] = None
 
 
 class AddContextRequest(BaseModel):

{simile-0.4.1.dist-info → simile-0.4.3.dist-info}/METADATA

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

simile-0.4.3.dist-info/RECORD

@@ -0,0 +1,11 @@
+simile/__init__.py,sha256=JAYtieyGg6YYiCackJ6YNlGJkFWmhryzbwwVt4K67uI,1360
+simile/auth_client.py,sha256=ICImmaA5fZX9ADbIPIUh4RED3hBZvLf3XSiaqELDAME,7923
+simile/client.py,sha256=hktN0Rgfsz9mPseQGQq5vBSZJ-RBhuAf68_mj5sAiJ0,27328
+simile/exceptions.py,sha256=Q1lbfwR7mEn_LYmwjAnsMc8BW79JNPvmCmVoPibYisU,1502
+simile/models.py,sha256=D-yoiEtnx0twfN5SmNL-j4MgjIYPtlcrgWfOQySHamo,11949
+simile/resources.py,sha256=LSYZSzx1YO69xvShGxxLFVPjQHw1WukHXhdYc1EyuOs,10555
+simile-0.4.3.dist-info/licenses/LICENSE,sha256=tpxX3bpODfyOQVyEM6kCMvPHFCpkjFDj0AICRqKqOFA,1066
+simile-0.4.3.dist-info/METADATA,sha256=zoikEVbRXmnWfYObo0MAoDXmMk9razQX--lpj-tWimw,1598
+simile-0.4.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+simile-0.4.3.dist-info/top_level.txt,sha256=41lJneubAG4-ZOAs5qn7iDtDb-MDxa6DdvgBKwNX84M,7
+simile-0.4.3.dist-info/RECORD,,

simile-0.4.1.dist-info/RECORD

@@ -1,11 +0,0 @@
-simile/__init__.py,sha256=JAYtieyGg6YYiCackJ6YNlGJkFWmhryzbwwVt4K67uI,1360
-simile/auth_client.py,sha256=ICImmaA5fZX9ADbIPIUh4RED3hBZvLf3XSiaqELDAME,7923
-simile/client.py,sha256=bou2NijZT8XF5cBzy5zrqnwE2bmLOYqagmpXHgZYwNk,28288
-simile/exceptions.py,sha256=Q1lbfwR7mEn_LYmwjAnsMc8BW79JNPvmCmVoPibYisU,1502
-simile/models.py,sha256=7-meixtd7oyMAc_k24SbKPjMD05tTI0npHVKAeGFcDo,11693
-simile/resources.py,sha256=LSYZSzx1YO69xvShGxxLFVPjQHw1WukHXhdYc1EyuOs,10555
-simile-0.4.1.dist-info/licenses/LICENSE,sha256=tpxX3bpODfyOQVyEM6kCMvPHFCpkjFDj0AICRqKqOFA,1066
-simile-0.4.1.dist-info/METADATA,sha256=YGu7eL55DR7TVW5Uon8MQfJq-sxOr1blFjG0vmQDuqY,1598
-simile-0.4.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-simile-0.4.1.dist-info/top_level.txt,sha256=41lJneubAG4-ZOAs5qn7iDtDb-MDxa6DdvgBKwNX84M,7
-simile-0.4.1.dist-info/RECORD,,