simile 0.4.7__tar.gz → 0.4.9__tar.gz

{simile-0.4.7 → simile-0.4.9}/PKG-INFO

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

{simile-0.4.7 → simile-0.4.9}/pyproject.toml

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

{simile-0.4.7 → simile-0.4.9}/simile/client.py

@@ -8,6 +8,7 @@ from .models import (
     Population,
     PopulationInfo,
     UpdatePopulationMetadataPayload,
+    UpdatePopulationInfoPayload,
     Agent as AgentModel,
     DataItem,
     DeletionResponse,
@@ -197,6 +198,29 @@ class Simile:
         )
         return response_data
 
+    async def update_population_info(
+        self,
+        population_id: Union[str, uuid.UUID],
+        name: Optional[str] = None,
+        description: Optional[str] = None,
+        metadata: Optional[Dict[str, Any]] = None,
+    ) -> Population:
+        """
+        Updates population information (name, description, metadata).
+        At least one of name, description, or metadata must be provided.
+        Requires write access to the population.
+        """
+        payload = UpdatePopulationInfoPayload(
+            name=name, description=description, metadata=metadata
+        )
+        response_data = await self._request(
+            "PUT",
+            f"populations/update/{str(population_id)}",
+            json=payload.model_dump(mode="json", exclude_none=True),
+            response_model=Population,
+        )
+        return response_data
+
     async def get_population_info(
         self, population_id: Union[str, uuid.UUID]
     ) -> PopulationInfo:
@@ -445,6 +469,8 @@ class Simile:
         self,
         agent_id: uuid.UUID,
         question: str,
+        question_id: Optional[str] = None,
+        study_id: Optional[str] = None,
         data_types: Optional[List[str]] = None,
         exclude_data_types: Optional[List[str]] = None,
         images: Optional[Dict[str, str]] = None,
@@ -465,6 +491,8 @@ class Simile:
         Args:
             agent_id: The agent to query
             question: The question to ask
+            question_id: Optional question ID for tracking
+            study_id: Optional study ID for tracking
             data_types: Optional data types to include
             exclude_data_types: Optional data types to exclude
             images: Optional images dict
@@ -478,6 +506,8 @@ class Simile:
         # Build request payload directly as dict to avoid serialization issues
         request_payload = {
             "question": question,
+            "question_id": question_id,
+            "study_id": study_id,
             "data_types": data_types,
             "exclude_data_types": exclude_data_types,
             "images": images,
@@ -516,6 +546,8 @@ class Simile:
         agent_id: uuid.UUID,
         question: str,
         options: List[str],
+        question_id: Optional[str] = None,
+        study_id: Optional[str] = None,
         data_types: Optional[List[str]] = None,
         exclude_data_types: Optional[List[str]] = None,
         images: Optional[Dict[str, str]] = None,
@@ -537,6 +569,8 @@ class Simile:
             agent_id: The agent to query
             question: The question to ask
             options: The options to choose from
+            question_id: Optional question ID for tracking
+            study_id: Optional study ID for tracking
             data_types: Optional data types to include
             exclude_data_types: Optional data types to exclude
             images: Optional images dict
@@ -551,6 +585,8 @@ class Simile:
         request_payload = {
             "question": question,
             "options": options,
+            "question_id": question_id,
+            "study_id": study_id,
             "data_types": data_types,
             "exclude_data_types": exclude_data_types,
             "images": images,

{simile-0.4.7 → simile-0.4.9}/simile/models.py

@@ -53,6 +53,21 @@ class UpdatePopulationMetadataPayload(BaseModel):
     mode: Optional[Literal["merge", "replace"]] = "merge"
 
 
+class UpdatePopulationInfoPayload(BaseModel):
+    name: Optional[str] = None
+    description: Optional[str] = None
+    metadata: Optional[Dict[str, Any]] = None
+
+    @validator("name", "description", "metadata")
+    def validate_at_least_one_field(cls, v, values):
+        # Check if at least one field is provided
+        if not any(values.values()):
+            raise ValueError(
+                "At least one of 'name', 'description', or 'metadata' must be provided"
+            )
+        return v
+
+
 class InitialDataItemPayload(BaseModel):
     data_type: str
     content: Any

{simile-0.4.7 → simile-0.4.9}/simile.egg-info/PKG-INFO

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