simile 0.3.4__py3-none-any.whl → 0.3.6__py3-none-any.whl

simile/client.py

@@ -233,11 +233,36 @@ class Simile:
         )
         return response_data
 
+    async def add_agent_to_population(
+        self, agent_id: Union[str, uuid.UUID], population_id: Union[str, uuid.UUID]
+    ) -> Dict[str, str]:
+        """Add an agent to an additional population."""
+        raw_response = await self._request(
+            "POST", f"agents/{str(agent_id)}/populations/{str(population_id)}"
+        )
+        return raw_response.json()
+
+    async def remove_agent_from_population(
+        self, agent_id: Union[str, uuid.UUID], population_id: Union[str, uuid.UUID]
+    ) -> Dict[str, str]:
+        """Remove an agent from a population."""
+        raw_response = await self._request(
+            "DELETE", f"agents/{str(agent_id)}/populations/{str(population_id)}"
+        )
+        return raw_response.json()
+
+    async def get_populations_for_agent(
+        self, agent_id: Union[str, uuid.UUID]
+    ) -> Dict[str, Any]:
+        """Get all populations an agent belongs to."""
+        raw_response = await self._request("GET", f"agents/{str(agent_id)}/populations")
+        return raw_response.json()
+
     async def create_data_item(
-        self, agent_id: Union[str, uuid.UUID], data_type: str, content: Any
+        self, agent_id: Union[str, uuid.UUID], data_type: str, content: Any, metadata: Optional[Dict[str, Any]] = None
     ) -> DataItem:
         """Creates a new data item for a specific agent."""
-        payload = CreateDataItemPayload(data_type=data_type, content=content)
+        payload = CreateDataItemPayload(data_type=data_type, content=content, metadata=metadata)
         response_data = await self._request(
             "POST",
             f"data_item/create/{str(agent_id)}",

simile/models.py

@@ -57,6 +57,7 @@ class CreateAgentPayload(BaseModel):
 class CreateDataItemPayload(BaseModel):
     data_type: str
     content: Any
+    metadata: Optional[Dict[str, Any]] = None
 
 
 class UpdateDataItemPayload(BaseModel):

{simile-0.3.4.dist-info → simile-0.3.6.dist-info}/METADATA

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

simile-0.3.6.dist-info/RECORD

@@ -0,0 +1,11 @@
+simile/__init__.py,sha256=GgbhCv5G5F8X_DIYw0H72D8fRuVHW7SLezGJykALpd0,1036
+simile/auth_client.py,sha256=ICImmaA5fZX9ADbIPIUh4RED3hBZvLf3XSiaqELDAME,7923
+simile/client.py,sha256=zJcB2l7pqrgCK7FlY_9eG3ATe39eMRpawEASwTwKXn0,13779
+simile/exceptions.py,sha256=-rJ3KZcpvNRi9JXbDpxWDSL2lU1mEJX2piwYRZvhKmg,1406
+simile/models.py,sha256=vbWrK5VKDPsiI9K4n2dCC41Y9who7cgTHSL9ARs25K0,5487
+simile/resources.py,sha256=Juoa32SlN8LdvzCFaobq7uV2itmcwbk5dM_mlc4bCVk,10420
+simile-0.3.6.dist-info/licenses/LICENSE,sha256=tpxX3bpODfyOQVyEM6kCMvPHFCpkjFDj0AICRqKqOFA,1066
+simile-0.3.6.dist-info/METADATA,sha256=-LjtcX15bPSTz4t8TjEgsZ6XOnfOdmkhLff75hJIBPY,1276
+simile-0.3.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+simile-0.3.6.dist-info/top_level.txt,sha256=41lJneubAG4-ZOAs5qn7iDtDb-MDxa6DdvgBKwNX84M,7
+simile-0.3.6.dist-info/RECORD,,

simile-0.3.4.dist-info/RECORD

@@ -1,11 +0,0 @@
-simile/__init__.py,sha256=GgbhCv5G5F8X_DIYw0H72D8fRuVHW7SLezGJykALpd0,1036
-simile/auth_client.py,sha256=ICImmaA5fZX9ADbIPIUh4RED3hBZvLf3XSiaqELDAME,7923
-simile/client.py,sha256=Y-u8LH6peWoBJSEP8fva9pq5DkhYf5DpETwqcdg0ogg,12679
-simile/exceptions.py,sha256=-rJ3KZcpvNRi9JXbDpxWDSL2lU1mEJX2piwYRZvhKmg,1406
-simile/models.py,sha256=tqTgNWQjoHdSDffsAhiWyzf3wKgbiU137CtWr3pcFpg,5441
-simile/resources.py,sha256=Juoa32SlN8LdvzCFaobq7uV2itmcwbk5dM_mlc4bCVk,10420
-simile-0.3.4.dist-info/licenses/LICENSE,sha256=tpxX3bpODfyOQVyEM6kCMvPHFCpkjFDj0AICRqKqOFA,1066
-simile-0.3.4.dist-info/METADATA,sha256=udSrz6PJqBo8ewtNnaGTzYWHcieaV6iB6VvH9FY102c,1276
-simile-0.3.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-simile-0.3.4.dist-info/top_level.txt,sha256=41lJneubAG4-ZOAs5qn7iDtDb-MDxa6DdvgBKwNX84M,7
-simile-0.3.4.dist-info/RECORD,,