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

cheshirecat_python_sdk/endpoints/conversation.py

@@ -1,11 +1,11 @@
-from typing import List
+from typing import List, Dict
 
 from cheshirecat_python_sdk.endpoints.base import AbstractEndpoint
 from cheshirecat_python_sdk.models.api.conversations import (
     ConversationHistoryOutput,
     ConversationDeleteOutput,
     ConversationsResponse,
-    ConversationNameChangeOutput,
+    ConversationAttributesChangeOutput,
 )
 from cheshirecat_python_sdk.utils import deserialize
 
@@ -13,7 +13,7 @@ from cheshirecat_python_sdk.utils import deserialize
 class ConversationEndpoint(AbstractEndpoint):
     def __init__(self, client: "CheshireCatClient"):
         super().__init__(client)
-        self.prefix = "/conversation"
+        self.prefix = "/conversations"
 
     def get_conversation_history(self, agent_id: str, user_id: str, chat_id: str) -> ConversationHistoryOutput:
         """
@@ -24,7 +24,7 @@ class ConversationEndpoint(AbstractEndpoint):
         :return: ConversationHistoryOutput, a list of conversation history entries.
         """
         return self.get(
-            self.format_url(chat_id),
+            self.format_url(f"{chat_id}/history"),
             agent_id,
             user_id=user_id,
             output_class=ConversationHistoryOutput,
@@ -42,6 +42,22 @@ class ConversationEndpoint(AbstractEndpoint):
 
         return [deserialize(item, ConversationsResponse) for item in response.json()]
 
+    def get_conversation(self, agent_id: str, user_id: str, chat_id: str) -> ConversationsResponse:
+        """
+        This endpoint returns the attributes of a specific conversation, given the `agent_id`, the `user_id` and the
+        `chat_id`.
+        :param agent_id: The agent ID.
+        :param user_id: The user ID to filter the conversation history.
+        :param chat_id: The chat ID to filter the conversation history.
+        :return: ConversationsResponse, the conversation attributes.
+        """
+        return self.get(
+            self.format_url(chat_id),
+            agent_id,
+            user_id=user_id,
+            output_class=ConversationsResponse,
+        )
+
     def delete_conversation(self, agent_id: str, user_id: str, chat_id: str) -> ConversationDeleteOutput:
         """
         This endpoint deletes the conversation.
@@ -57,27 +73,36 @@ class ConversationEndpoint(AbstractEndpoint):
             user_id=user_id,
         )
 
-    def post_conversation_name(
+    def put_conversation_attributes(
         self,
-        name: str,
         agent_id: str,
         user_id: str,
         chat_id: str,
-    ) -> ConversationNameChangeOutput:
+        name: str | None,
+        metadata: Dict | None = None,
+    ) -> ConversationAttributesChangeOutput:
         """
         This endpoint creates a new element in the conversation history.
-        :param name: The new name to assign to the conversation
         :param agent_id: The agent ID.
         :param user_id: The user ID to filter the conversation history.
         :param chat_id: The chat ID to filter the conversation history.
+        :param name: The new name to assign to the conversation
+        :param metadata: The metadata to assign to the conversation
         :return: ConversationNameChangeOutput, a message indicating whether the conversation name was changed.
         """
-        payload = {"name": name}
+        if not name and not metadata:
+            raise ValueError("Either name or metadata must be provided")
+
+        payload = {}
+        if name:
+            payload = {"name": name}
+        if metadata:
+            payload["metadata"] = metadata
 
-        return self.post_json(
+        return self.put(
             self.format_url(chat_id),
             agent_id,
-            output_class=ConversationNameChangeOutput,
+            output_class=ConversationAttributesChangeOutput,
             payload=payload,
             user_id=user_id,
         )

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/models/api/conversations.py

@@ -1,4 +1,4 @@
-from typing import List
+from typing import List, Any, Dict
 from pydantic import BaseModel
 
 from cheshirecat_python_sdk.models.api.nested.memories import ConversationMessage
@@ -16,9 +16,10 @@ class ConversationsResponse(BaseModel):
     chat_id: str
     name: str
     num_messages: int
+    metadata: Dict[str, Any]
     created_at: float | None
     updated_at: float | None
 
 
-class ConversationNameChangeOutput(BaseModel):
+class ConversationAttributesChangeOutput(BaseModel):
     changed: bool

{cheshirecat_python_sdk-1.8.4.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.4
+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.4.dist-info → cheshirecat_python_sdk-1.8.6.dist-info}/RECORD

@@ -18,7 +18,7 @@ cheshirecat_python_sdk/endpoints/auth.py,sha256=_22fLGznoxo2pybcQGgcsHCpLoq76P6R
 cheshirecat_python_sdk/endpoints/auth_handler.py,sha256=Jfi7E7L3SIzJyECbJnUenIyl6xxkJlyC31AHpAlmqSs,2117
 cheshirecat_python_sdk/endpoints/base.py,sha256=0X2EQxRGqR6VG5gXE5XKcscmQ7rNM9de61GBrQlKD90,4210
 cheshirecat_python_sdk/endpoints/chunker.py,sha256=bgp-2gQpKt7W_B46eEb-KFRkd4eZJH0euwsf3N0jfC8,1979
-cheshirecat_python_sdk/endpoints/conversation.py,sha256=jG1z7ivdD53H8JZe7WZWihoBWy2_-W951SwKXmnpmTk,3282
+cheshirecat_python_sdk/endpoints/conversation.py,sha256=OGxG312kQBzL5629vCmTreltUrEnoUV4JshZUcj4L2U,4335
 cheshirecat_python_sdk/endpoints/custom_endpoint.py,sha256=HVRQMqA5JfPr87gHtWpYcsXsE-TA-coXLcvsPKabqds,2791
 cheshirecat_python_sdk/endpoints/embedder.py,sha256=4B2yc3KBThKGreq_KPTTCsxt2Ekktw_muZ5gBfxrrwY,1741
 cheshirecat_python_sdk/endpoints/file_manager.py,sha256=U4kc2oR3UszK0EKzYnES_RkgF1zhaPINAW_3X2WgX7A,4614
@@ -28,12 +28,12 @@ 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/conversations.py,sha256=Ej8677MLBqZEME1lZFd4UhHokdja762zYNjV9ZaEg-I,505
+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
 cheshirecat_python_sdk/models/api/memories.py,sha256=4rqFuFNzNbmwIG1KGiSyfnv40nds5g7fSkov3drj0cc,948
@@ -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.4.dist-info/METADATA,sha256=f5sjZkDGm4AXlxsMLbEjOEEgD2Pyke2NMqM7zi0Ag_4,44136
-cheshirecat_python_sdk-1.8.4.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-cheshirecat_python_sdk-1.8.4.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-cheshirecat_python_sdk-1.8.4.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,,