cheshirecat-python-sdk 1.8.5__py3-none-any.whl → 1.8.6__py3-none-any.whl

cheshirecat_python_sdk/endpoints/users.py

@@ -59,10 +59,7 @@ class UsersEndpoint(AbstractEndpoint):
         response = self.get_http_client(agent_id).get(self.prefix)
         response.raise_for_status()
 
-        result = []
-        for item in response.json():
-            result.append(deserialize(item, UserOutput))
-        return result
+        return [deserialize(item, UserOutput) for item in response.json()]
 
     def get_user(self, user_id: str, agent_id: str) -> UserOutput:
         """

cheshirecat_python_sdk/endpoints/utils.py

@@ -1,7 +1,13 @@
-from typing import List
+from typing import List, Dict
 
 from cheshirecat_python_sdk.endpoints.base import AbstractEndpoint
-from cheshirecat_python_sdk.models.api.admins import ResetOutput, ClonedOutput, CreatedOutput
+from cheshirecat_python_sdk.models.api.admins import (
+    ResetOutput,
+    AgentClonedOutput,
+    AgentCreatedOutput,
+    AgentOutput,
+    AgentUpdatedOutput,
+)
 from cheshirecat_python_sdk.utils import deserialize
 
 
@@ -17,28 +23,33 @@ class UtilsEndpoint(AbstractEndpoint):
         """
         return self.post_json(self.format_url("/factory/reset/"), self.system_id, output_class=ResetOutput)
 
-    def get_agents(self) -> List[str]:
+    def get_agents(self) -> List[AgentOutput]:
         """
         Get a list of all agents.
-        :return: List[str], the IDs of the agents.
+        :return: List[AgentOutput], the ID and the metadata of the agents.
         """
-        return self.get(self.format_url("/agents/"), self.system_id)
+        response = self.get_http_client(agent_id=self.system_id).get(self.format_url("/agents/"))
+        response.raise_for_status()
+
+        return [deserialize(item, AgentOutput) for item in response.json()]
 
-    def post_agent_create(self, agent_id: str) -> CreatedOutput:
+    def post_agent_create(self, agent_id: str, metadata: Dict | None = None) -> AgentCreatedOutput:
         """
         Create a new agent.
         :param agent_id: The ID of the agent.
-        :return: CreatedOutput, the details of the agent.
+        :param metadata: The metadata of the agent.
+        :return: AgentCreatedOutput, the details of the agent.
         """
-        response = self.get_http_client().post(
+        payload = {"agent_id": agent_id}
+        if metadata is not None:
+            payload["metadata"] = metadata  # type: ignore
+
+        return self.post_json(
             self.format_url("/agents/create/"),
-            json={
-                "agent_id": agent_id,
-            },
+            self.system_id,
+            output_class=AgentCreatedOutput,
+            payload=payload,
         )
-        response.raise_for_status()
-
-        return deserialize(response.json(), CreatedOutput)
 
     def post_agent_reset(self, agent_id: str) -> ResetOutput:
         """
@@ -56,16 +67,30 @@ class UtilsEndpoint(AbstractEndpoint):
         """
         return self.post_json(self.format_url("/agents/destroy/"), agent_id, output_class=ResetOutput)
 
-    def post_agent_clone(self, agent_id: str, new_agent_id: str) -> ClonedOutput:
+    def post_agent_clone(self, agent_id: str, new_agent_id: str) -> AgentClonedOutput:
         """
         Destroy an agent.
         :param agent_id: The ID of the agent.
         :param new_agent_id: The ID of the new cloned agent.
-        :return: ClonedOutput, the details of the cloning.
+        :return: AgentClonedOutput, the details of the cloning.
         """
         return self.post_json(
             self.format_url("/agents/clone/"),
             agent_id,
             payload={"agent_id": new_agent_id},
-            output_class=ClonedOutput
-        )
+            output_class=AgentClonedOutput,
+        )
+
+    def put_agent(self, agent_id: str, metadata: Dict) -> AgentUpdatedOutput:
+        """
+        Update the metadata of an agent.
+        :param agent_id: The ID of the agent.
+        :param metadata: The new metadata for the agent.
+        :return: AgentUpdatedOutput, the details of the update.
+        """
+        return self.put(
+            self.format_url("/agents/"),
+            agent_id,
+            payload={"metadata": metadata},
+            output_class=AgentUpdatedOutput,
+        )

cheshirecat_python_sdk/models/api/admins.py

@@ -4,7 +4,7 @@ from pydantic import BaseModel
 from cheshirecat_python_sdk.models.api.plugins import PluginToggleOutput
 
 
-class CreatedOutput(BaseModel):
+class AgentCreatedOutput(BaseModel):
     created: bool
 
 
@@ -32,5 +32,14 @@ class ResetOutput(BaseModel):
     deleted_plugin_folders: bool
 
 
-class ClonedOutput(BaseModel):
+class AgentClonedOutput(BaseModel):
     cloned: bool = False
+
+
+class AgentUpdatedOutput(BaseModel):
+    updated: bool
+
+
+class AgentOutput(BaseModel):
+    agent_id: str
+    metadata: Dict

{cheshirecat_python_sdk-1.8.5.dist-info → cheshirecat_python_sdk-1.8.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cheshirecat-python-sdk
-Version: 1.8.5
+Version: 1.8.6
 Summary: Python SDK for the Cloud-ready fork of the Cheshire Cat
 Project-URL: Repository, https://github.com/matteocacciola/cheshirecat-python-sdk
 Project-URL: Documentation, https://github.com/matteocacciola/cheshirecat-python-sdk#README

{cheshirecat_python_sdk-1.8.5.dist-info → cheshirecat_python_sdk-1.8.6.dist-info}/RECORD

@@ -28,11 +28,11 @@ cheshirecat_python_sdk/endpoints/memory.py,sha256=dgfoyjZH0gL3kln31gFJ42JHfL-S-o
 cheshirecat_python_sdk/endpoints/message.py,sha256=QTUSUJcj52VhfnX9hdjsD2aLeFC43O4eU8ld5lRvITU,2607
 cheshirecat_python_sdk/endpoints/plugins.py,sha256=FasjwNY2UVv7zfg9lvXEj2ATefPDOPQpJXzQ2BbQIYI,3474
 cheshirecat_python_sdk/endpoints/rabbit_hole.py,sha256=KjF5IRG6ghGEji-Js5try-9FhmWsuiM5v04jA68VoVY,7316
-cheshirecat_python_sdk/endpoints/users.py,sha256=ySpFi9ILyoT44fnAz8p4LSGTaEAMWv3RBWq72Hv73mI,5803
-cheshirecat_python_sdk/endpoints/utils.py,sha256=yKGrFRpu6ItLnBEuc98093LbXqGYaGdgCLiWuLQJVno,2551
+cheshirecat_python_sdk/endpoints/users.py,sha256=19kei56IDYlteq_KePgzjJne_NgJkaLDAyPVwV3_2i8,5742
+cheshirecat_python_sdk/endpoints/utils.py,sha256=etF2lipz-rGY8CAQve0zqDIgauJGeTdGYB3PtkcAb_4,3457
 cheshirecat_python_sdk/endpoints/vector_database.py,sha256=Xu6zcopjZgcFVVmyN547piiqFcaTUtF35X0QxQHycXQ,2149
 cheshirecat_python_sdk/models/dtos.py,sha256=0N6Auv43Gifo0ptdZxMZXAB6QGk8NNbcf8vQGm8HClo,869
-cheshirecat_python_sdk/models/api/admins.py,sha256=sIX4NdEntjORgVACsqkBkzte4zwD6w35BdPRRG0D7uo,661
+cheshirecat_python_sdk/models/api/admins.py,sha256=eIzK2MFjzF3w9Sqgt1Rkei-IW06sMpBhuOea7HenvSY,797
 cheshirecat_python_sdk/models/api/conversations.py,sha256=ifgCwbtYwPiilQ-pz1peXaR7yVDNXlmIy_eWp25LGDE,551
 cheshirecat_python_sdk/models/api/factories.py,sha256=_NWz0nKhaVYEbaphxe2YZGznDIUiMKb-xqA1xymME5A,594
 cheshirecat_python_sdk/models/api/file_managers.py,sha256=V0ZBjQWmYM1t0Qpfp3W2NDI_tScOtORB0i5YMEpTiRU,301
@@ -44,7 +44,7 @@ cheshirecat_python_sdk/models/api/tokens.py,sha256=GavSaCq0-SRWjvLtDgrYBOdUELd7V
 cheshirecat_python_sdk/models/api/users.py,sha256=_T3-8hfmlZK9cZcqBTbNMYS74yEnDrRACRQn9PNCEiQ,280
 cheshirecat_python_sdk/models/api/nested/memories.py,sha256=ddGDsmZa2refElM0sGz9QfKHGk50wsTpkwN5qDC5bF4,946
 cheshirecat_python_sdk/models/api/nested/plugins.py,sha256=7vrwgXh0csaUn11YKY_OWnGEnD8Fu_ewKxt024VOBEs,738
-cheshirecat_python_sdk-1.8.5.dist-info/METADATA,sha256=K5gpaytNWUZd-1rVwXykanO9bUztJYGQYIIZPvTXuws,44136
-cheshirecat_python_sdk-1.8.5.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-cheshirecat_python_sdk-1.8.5.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-cheshirecat_python_sdk-1.8.5.dist-info/RECORD,,
+cheshirecat_python_sdk-1.8.6.dist-info/METADATA,sha256=-bXe7LwP6dz8_a8LmjHqOEV8XYO9Svml0kXY6wD1DTI,44136
+cheshirecat_python_sdk-1.8.6.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+cheshirecat_python_sdk-1.8.6.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+cheshirecat_python_sdk-1.8.6.dist-info/RECORD,,