simile 0.4.8__tar.gz → 0.4.9__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: simile
-Version: 0.4.8
+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.8 → simile-0.4.9}/pyproject.toml

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

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

@@ -8,6 +8,7 @@ from .models import (
     Population,
     PopulationInfo,
     UpdatePopulationMetadataPayload,
+    UpdatePopulationInfoPayload,
     Agent as AgentModel,
     DataItem,
     DeletionResponse,
@@ -186,7 +187,7 @@ class Simile:
         response_data = await self._request(
             "PATCH",
             f"populations/{str(population_id)}/metadata",
-            json=payload,
+            json=payload.model_dump(mode="json", exclude_none=True),
             response_model=Population,
         )
         return response_data
@@ -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:

{simile-0.4.8 → 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.8 → simile-0.4.9}/simile.egg-info/PKG-INFO

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