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

simile/client.py

@@ -7,6 +7,7 @@ from pydantic import BaseModel
 from .models import (
     Population,
     PopulationInfo,
+    UpdatePopulationMetadataPayload,
     Agent as AgentModel,
     DataItem,
     DeletionResponse,
@@ -164,6 +165,32 @@ class Simile:
         )
         return response_data
 
+    async def update_population_metadata(
+        self,
+        population_id: Union[str, uuid.UUID],
+        metadata: Dict[str, Any],
+        mode: str = "merge",
+    ) -> Population:
+        """
+        Update a population's metadata (jsonb).
+
+        Args:
+            population_id: The ID of the population
+            metadata: A dictionary of metadata to merge or replace
+            mode: Either "merge" (default) or "replace"
+
+        Returns:
+            Updated Population object
+        """
+        payload = UpdatePopulationMetadataPayload(metadata=metadata, mode=mode)
+        response_data = await self._request(
+            "PATCH",
+            f"populations/{str(population_id)}/metadata",
+            json=payload,
+            response_model=Population,
+        )
+        return response_data
+
     async def get_population(self, population_id: Union[str, uuid.UUID]) -> Population:
         response_data = await self._request(
             "GET", f"populations/get/{str(population_id)}", response_model=Population
@@ -203,6 +230,8 @@ class Simile:
     async def create_agent(
         self,
         name: str,
+        source: Optional[str] = None,
+        source_id: Optional[str] = None,
         population_id: Optional[Union[str, uuid.UUID]] = None,
         agent_data: Optional[List[Dict[str, Any]]] = None,
     ) -> AgentModel:
@@ -255,16 +284,16 @@ class Simile:
             "DELETE", f"agents/{str(agent_id)}/populations/{str(population_id)}"
         )
         return raw_response.json()
-    
+
     async def batch_add_agents_to_population(
-        self, agent_ids: List[Union[str, uuid.UUID]], population_id: Union[str, uuid.UUID]
+        self,
+        agent_ids: List[Union[str, uuid.UUID]],
+        population_id: Union[str, uuid.UUID],
     ) -> Dict[str, Any]:
         """Add multiple agents to a population in a single batch operation."""
         agent_id_strs = [str(aid) for aid in agent_ids]
         raw_response = await self._request(
-            "POST", 
-            f"populations/{str(population_id)}/agents/batch",
-            json=agent_id_strs
+            "POST", f"populations/{str(population_id)}/agents/batch", json=agent_id_strs
         )
         return raw_response.json()
 
@@ -419,12 +448,16 @@ class Simile:
         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
+        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
-        save_memory: Optional[Union[str, uuid.UUID]] = None,  # Session ID to save memory to
+        save_memory: Optional[
+            Union[str, uuid.UUID]
+        ] = None,  # Session ID to save memory to
     ) -> OpenGenerationResponse:
         """Generates an open response from an agent based on a question.
-        
+
         Args:
             agent_id: The agent to query
             question: The question to ask
@@ -448,30 +481,30 @@ class Simile:
             "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()
-        
+
         response_data = await self._request(
             "POST",
             endpoint,
             json=request_payload,
             response_model=OpenGenerationResponse,
         )
-        
+
         # 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
 
     async def generate_closed_response(
@@ -486,12 +519,16 @@ class Simile:
         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
+        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
-        save_memory: Optional[Union[str, uuid.UUID]] = None,  # Session ID to save memory to
+        save_memory: Optional[
+            Union[str, uuid.UUID]
+        ] = None,  # Session ID to save memory to
     ) -> ClosedGenerationResponse:
         """Generates a closed response from an agent.
-        
+
         Args:
             agent_id: The agent to query
             question: The question to ask
@@ -517,34 +554,34 @@ class Simile:
             "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()
-        
+
         response_data = await self._request(
             "POST",
             endpoint,
             json=request_payload,
             response_model=ClosedGenerationResponse,
         )
-        
+
         # 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
 
     # Memory Management Methods
-    
+
     async def save_memory(
         self,
         agent_id: Union[str, uuid.UUID],
@@ -559,7 +596,7 @@ class Simile:
     ) -> str:
         """
         Save a response with associated memory information.
-        
+
         Args:
             agent_id: The agent ID
             response: The agent's response text
@@ -570,7 +607,7 @@ class Simile:
             memory_stream_used: The memory stream that was used
             reasoning: Optional reasoning
             metadata: Additional metadata
-        
+
         Returns:
             Response ID if saved successfully
         """
@@ -578,7 +615,7 @@ class Simile:
             "agent_id": str(agent_id),
             "response": response,
         }
-        
+
         if session_id:
             payload["session_id"] = str(session_id)
         if question_id:
@@ -593,13 +630,13 @@ class Simile:
             payload["reasoning"] = reasoning
         if metadata:
             payload["metadata"] = metadata
-        
+
         response = await self._request("POST", "memory/save", json=payload)
         data = response.json()
         if data.get("success"):
             return data.get("response_id")
         raise SimileAPIError("Failed to save memory")
-    
+
     async def get_memory(
         self,
         session_id: Union[str, uuid.UUID],
@@ -611,7 +648,7 @@ class Simile:
     ) -> Optional[MemoryStream]:
         """
         Retrieve the memory stream for an agent in a session.
-        
+
         Args:
             session_id: Session ID to filter by
             agent_id: The agent ID
@@ -619,7 +656,7 @@ class Simile:
             exclude_question_ids: List of question IDs to exclude
             limit: Maximum number of turns to include
             use_memory: Whether to use memory at all
-        
+
         Returns:
             MemoryStream object or None
         """
@@ -628,31 +665,31 @@ class Simile:
             "agent_id": str(agent_id),
             "use_memory": use_memory,
         }
-        
+
         if exclude_study_ids:
             payload["exclude_study_ids"] = [str(id) for id in exclude_study_ids]
         if exclude_question_ids:
             payload["exclude_question_ids"] = [str(id) for id in exclude_question_ids]
         if limit:
             payload["limit"] = limit
-        
+
         response = await self._request("POST", "memory/get", json=payload)
         data = response.json()
-        
+
         if data.get("success") and data.get("memory_stream"):
             return MemoryStream.from_dict(data["memory_stream"])
         return None
-    
+
     async def get_memory_summary(
         self,
         session_id: Union[str, uuid.UUID],
     ) -> Dict[str, Any]:
         """
         Get a summary of memory usage for a session.
-        
+
         Args:
             session_id: Session ID to analyze
-        
+
         Returns:
             Dictionary with memory statistics
         """
@@ -661,7 +698,7 @@ class Simile:
         if data.get("success"):
             return data.get("summary", {})
         return {}
-    
+
     async def clear_memory(
         self,
         session_id: Union[str, uuid.UUID],
@@ -670,28 +707,28 @@ class Simile:
     ) -> bool:
         """
         Clear memory for a session, optionally filtered by agent or study.
-        
+
         Args:
             session_id: Session ID to clear memory for
             agent_id: Optional agent ID to filter by
             study_id: Optional study ID to filter by
-        
+
         Returns:
             True if cleared successfully, False otherwise
         """
         payload = {
             "session_id": str(session_id),
         }
-        
+
         if agent_id:
             payload["agent_id"] = str(agent_id)
         if study_id:
             payload["study_id"] = str(study_id)
-        
+
         response = await self._request("POST", "memory/clear", json=payload)
         data = response.json()
         return data.get("success", False)
-    
+
     async def copy_memory(
         self,
         from_session_id: Union[str, uuid.UUID],
@@ -700,12 +737,12 @@ class Simile:
     ) -> int:
         """
         Copy memory from one session to another.
-        
+
         Args:
             from_session_id: Source session ID
             to_session_id: Destination session ID
             agent_id: Optional agent ID to filter by
-        
+
         Returns:
             Number of memory turns copied
         """
@@ -713,10 +750,10 @@ class Simile:
             "from_session_id": str(from_session_id),
             "to_session_id": str(to_session_id),
         }
-        
+
         if agent_id:
             payload["agent_id"] = str(agent_id)
-        
+
         response = await self._request("POST", "memory/copy", json=payload)
         data = response.json()
         if data.get("success"):

simile/models.py

@@ -11,6 +11,7 @@ class Population(BaseModel):
     description: Optional[str] = None
     created_at: datetime
     updated_at: datetime
+    metadata: Optional[Dict[str, Any]] = None
 
 
 class PopulationInfo(BaseModel):
@@ -18,6 +19,7 @@ class PopulationInfo(BaseModel):
     name: str
     description: Optional[str] = None
     agent_count: int
+    metadata: Optional[Dict[str, Any]] = None
 
 
 class DataItem(BaseModel):
@@ -37,6 +39,8 @@ class Agent(BaseModel):
     created_at: datetime
     updated_at: datetime
     data_items: List[DataItem] = Field(default_factory=list)
+    source: Optional[str] = None
+    source_id: Optional[str] = None
 
 
 class CreatePopulationPayload(BaseModel):
@@ -44,6 +48,11 @@ class CreatePopulationPayload(BaseModel):
     description: Optional[str] = None
 
 
+class UpdatePopulationMetadataPayload(BaseModel):
+    metadata: Dict[str, Any]
+    mode: Optional[Literal["merge", "replace"]] = "merge"
+
+
 class InitialDataItemPayload(BaseModel):
     data_type: str
     content: Any
@@ -52,6 +61,8 @@ class InitialDataItemPayload(BaseModel):
 
 class CreateAgentPayload(BaseModel):
     name: str
+    source: Optional[str] = None
+    source_id: Optional[str] = None
     population_id: Optional[uuid.UUID] = None
     agent_data: Optional[List[InitialDataItemPayload]] = None
 
@@ -259,7 +270,7 @@ class BaseMemoryTurn(BaseModel):
 
     class Config:
         use_enum_values = True
-    
+
     def to_dict(self) -> Dict[str, Any]:
         """Convert to dictionary for serialization."""
         data = self.model_dump()
@@ -354,9 +365,7 @@ class MemoryStream(BaseModel):
 
     def to_dict(self) -> Dict[str, Any]:
         """Convert memory stream to a dictionary for serialization."""
-        return {
-            "turns": [turn.to_dict() for turn in self.turns]
-        }
+        return {"turns": [turn.to_dict() for turn in self.turns]}
 
     @classmethod
     def from_dict(cls, data: Dict[str, Any]) -> "MemoryStream":
@@ -377,7 +386,9 @@ class MemoryStream(BaseModel):
     def fork(self, up_to_index: Optional[int] = None) -> "MemoryStream":
         """Create a copy of this memory stream, optionally up to a specific index."""
         new_memory = MemoryStream()
-        turns_to_copy = self.turns[:up_to_index] if up_to_index is not None else self.turns
+        turns_to_copy = (
+            self.turns[:up_to_index] if up_to_index is not None else self.turns
+        )
         for turn in turns_to_copy:
             new_memory.add_turn(turn.model_copy())
         return new_memory

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

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

simile-0.4.5.dist-info/RECORD

@@ -0,0 +1,11 @@
+simile/__init__.py,sha256=JAYtieyGg6YYiCackJ6YNlGJkFWmhryzbwwVt4K67uI,1360
+simile/auth_client.py,sha256=ICImmaA5fZX9ADbIPIUh4RED3hBZvLf3XSiaqELDAME,7923
+simile/client.py,sha256=JkXw4jbn6vr_0g2kXCPc8uR5i8ioGpBLOzokfxCiZvo,28049
+simile/exceptions.py,sha256=Q1lbfwR7mEn_LYmwjAnsMc8BW79JNPvmCmVoPibYisU,1502
+simile/models.py,sha256=qYXQ7KCKLsWf8TkbBlJUDU2kLZPZ69WIjX8DlavB_Bk,12316
+simile/resources.py,sha256=LSYZSzx1YO69xvShGxxLFVPjQHw1WukHXhdYc1EyuOs,10555
+simile-0.4.5.dist-info/licenses/LICENSE,sha256=tpxX3bpODfyOQVyEM6kCMvPHFCpkjFDj0AICRqKqOFA,1066
+simile-0.4.5.dist-info/METADATA,sha256=h9_Tzwu5be6vtk-mL-kz0C2pSXiYrlaVUdgoZqpY8ls,1598
+simile-0.4.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+simile-0.4.5.dist-info/top_level.txt,sha256=41lJneubAG4-ZOAs5qn7iDtDb-MDxa6DdvgBKwNX84M,7
+simile-0.4.5.dist-info/RECORD,,

simile-0.4.3.dist-info/RECORD

@@ -1,11 +0,0 @@
-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,,