simile 0.4.1__tar.gz → 0.4.3__tar.gz

{simile-0.4.1 → simile-0.4.3}/PKG-INFO

@@ -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.1 → simile-0.4.3}/pyproject.toml

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

{simile-0.4.1 → simile-0.4.3}/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-0.4.1 → simile-0.4.3}/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 → simile-0.4.3}/simile.egg-info/PKG-INFO

@@ -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