cheshirecat-python-sdk 1.7.4__py3-none-any.whl → 1.7.6__py3-none-any.whl

cheshirecat_python_sdk/endpoints/auth.py

@@ -1,7 +1,7 @@
 from typing import Any
 
 from cheshirecat_python_sdk.endpoints.base import AbstractEndpoint
-from cheshirecat_python_sdk.models.api.tokens import TokenOutput
+from cheshirecat_python_sdk.models.api.tokens import TokenOutput, MeOutput
 from cheshirecat_python_sdk.utils import deserialize
 
 
@@ -10,7 +10,7 @@ class AuthEndpoint(AbstractEndpoint):
         super().__init__(client)
         self.prefix = "/auth"
 
-    def token(self, username: str, password: str, agent_id: str | None = None) -> TokenOutput:
+    def token(self, username: str, password: str) -> TokenOutput:
         """
         This endpoint is used to get a token for the user. The token is used to authenticate the user in the system. When
         the token expires, the user must request a new token.
@@ -21,7 +21,6 @@ class AuthEndpoint(AbstractEndpoint):
                 "username": username,
                 "password": password,
             },
-            headers={"X-Agent-ID": agent_id} if agent_id else None,
         )
         response.raise_for_status()
 
@@ -40,3 +39,16 @@ class AuthEndpoint(AbstractEndpoint):
         response.raise_for_status()
 
         return response.json()
+
+    def me(self, token: str) -> MeOutput:
+        """
+        This endpoint is used to get the current user information. The user information includes the list of agents
+        the user has access to. This endpoint requires authentication.
+        """
+        self.client.add_token(token)
+
+        response = self.get_http_client().get("/me")
+        response.raise_for_status()
+
+        result = deserialize(response.json(), MeOutput)
+        return result

cheshirecat_python_sdk/endpoints/base.py

@@ -2,6 +2,7 @@ from abc import ABC
 from typing import Dict, Any, List, Tuple, Type
 import requests
 from pydantic import BaseModel
+from requests_toolbelt.sessions import BaseUrlSession
 from websockets import ClientConnection
 
 from cheshirecat_python_sdk.utils import T, deserialize
@@ -29,7 +30,7 @@ class AbstractEndpoint(ABC):
     ) -> requests.Session:
         return self.client.http_client.get_client(agent_id, user_id, chat_id)
 
-    def get_http_session(self):
+    def get_http_session(self) -> BaseUrlSession:
         return self.client.http_client.get_base_session()
 
     async def get_ws_client(self, agent_id: str, user_id: str, chat_id: str | None = None) -> ClientConnection:

cheshirecat_python_sdk/endpoints/utils.py

@@ -2,6 +2,7 @@ from typing import List
 
 from cheshirecat_python_sdk.endpoints.base import AbstractEndpoint
 from cheshirecat_python_sdk.models.api.admins import ResetOutput, ClonedOutput, CreatedOutput
+from cheshirecat_python_sdk.utils import deserialize
 
 
 class UtilsEndpoint(AbstractEndpoint):
@@ -29,7 +30,15 @@ class UtilsEndpoint(AbstractEndpoint):
         :param agent_id: The ID of the agent.
         :return: CreatedOutput, the details of the agent.
         """
-        return self.post_json(self.format_url("/agents/create/"), agent_id, output_class=CreatedOutput)
+        response = self.get_http_client().post(
+            self.format_url("/agents/create/"),
+            json={
+                "agent_id": agent_id,
+            },
+        )
+        response.raise_for_status()
+
+        return deserialize(response.json(), CreatedOutput)
 
     def post_agent_reset(self, agent_id: str) -> ResetOutput:
         """

cheshirecat_python_sdk/models/api/tokens.py

@@ -4,3 +4,22 @@ from pydantic import BaseModel
 class TokenOutput(BaseModel):
     access_token: str
     token_type: str | None = "bearer"
+
+
+class User(BaseModel):
+    id: str
+    username: str
+    permissions: dict[str, list[str]]
+
+
+class AgentMatch(BaseModel):
+    agent_id: str
+    agent_name: str
+    agent_description: str | None = None
+    user: User
+
+
+class MeOutput(BaseModel):
+    success : bool
+    agents: list[AgentMatch]
+    auto_selected: bool

{cheshirecat_python_sdk-1.7.4.dist-info → cheshirecat_python_sdk-1.7.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cheshirecat-python-sdk
-Version: 1.7.4
+Version: 1.7.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.7.4.dist-info → cheshirecat_python_sdk-1.7.6.dist-info}/RECORD

@@ -13,9 +13,9 @@ cheshirecat_python_sdk/clients/http_client.py,sha256=LnP0kr-kHRCwC-BZiysj1Zg-T3b
 cheshirecat_python_sdk/clients/websocket_client.py,sha256=xNjkLSEZhpHzRqIZL21V0oG3srAGFaqYveMqlma2gvI,1932
 cheshirecat_python_sdk/endpoints/__init__.py,sha256=uQPclok3DdKTibR7ywDVM67eqtxEMwx_iuza37Wqkr8,1247
 cheshirecat_python_sdk/endpoints/admins.py,sha256=rc6jEhYF8KTTWFXLinm3xPV7Rj87iKT8Kp0zHn1qUO4,7024
-cheshirecat_python_sdk/endpoints/auth.py,sha256=GcmN90uZIgeRqc_NjBhZx7bxYRSpfKL6ZOLHLom1aKQ,1658
+cheshirecat_python_sdk/endpoints/auth.py,sha256=_22fLGznoxo2pybcQGgcsHCpLoq76P6RplmSMMgajEQ,2031
 cheshirecat_python_sdk/endpoints/auth_handler.py,sha256=Jfi7E7L3SIzJyECbJnUenIyl6xxkJlyC31AHpAlmqSs,2117
-cheshirecat_python_sdk/endpoints/base.py,sha256=ldayH0QxQfzD5bYqlMBH1JVBOTd11RYDWxf-6kl0fng,4003
+cheshirecat_python_sdk/endpoints/base.py,sha256=W_ynJur_cbT31KLp6F4pJ212tMaAa1BxnG75YS1UgNU,4075
 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/custom_endpoint.py,sha256=HVRQMqA5JfPr87gHtWpYcsXsE-TA-coXLcvsPKabqds,2791
@@ -28,7 +28,7 @@ cheshirecat_python_sdk/endpoints/message.py,sha256=QTUSUJcj52VhfnX9hdjsD2aLeFC43
 cheshirecat_python_sdk/endpoints/plugins.py,sha256=FasjwNY2UVv7zfg9lvXEj2ATefPDOPQpJXzQ2BbQIYI,3474
 cheshirecat_python_sdk/endpoints/rabbit_hole.py,sha256=DH5KGpk0Ur8RPi1fZIO9D925mc2D0GHQpsOeIKIa1G0,7213
 cheshirecat_python_sdk/endpoints/users.py,sha256=PXlPPobLoWpiuAzi0k89YUyqSp24Sc5DEcDQYGRXRBA,5369
-cheshirecat_python_sdk/endpoints/utils.py,sha256=UhhWCC4h9c7lbPxAaV3nHjLYwod4z0tHzrE-2tuKs8M,2328
+cheshirecat_python_sdk/endpoints/utils.py,sha256=yKGrFRpu6ItLnBEuc98093LbXqGYaGdgCLiWuLQJVno,2551
 cheshirecat_python_sdk/endpoints/vector_database.py,sha256=Xu6zcopjZgcFVVmyN547piiqFcaTUtF35X0QxQHycXQ,2149
 cheshirecat_python_sdk/models/dtos.py,sha256=yK6CfCGdwLvZtYFPoImDEgpV5eyHB2iwji2t6dh6gxc,668
 cheshirecat_python_sdk/models/api/admins.py,sha256=K1HzZrjEk--rmBSxEg3cbWm2MVQaSxTheMBhhBr54_4,767
@@ -39,11 +39,11 @@ cheshirecat_python_sdk/models/api/memories.py,sha256=4rqFuFNzNbmwIG1KGiSyfnv40nd
 cheshirecat_python_sdk/models/api/messages.py,sha256=oslVFn8sMNOZyQkOzAncI2W2WLX0Tmd8Avc81iEIpWo,609
 cheshirecat_python_sdk/models/api/plugins.py,sha256=uX8SYbcTxQSxv1IaVChEBd3rr6kc83rwq3IILnmq5EE,1996
 cheshirecat_python_sdk/models/api/rabbit_holes.py,sha256=01FcPSIG6nq9WJhL_rQopLMSD81CU9lfP-t0xtf35Gw,285
-cheshirecat_python_sdk/models/api/tokens.py,sha256=5vzQqmcYWg593U1cTnnmLPZMNkgNNY5egQBQTE-NDoE,123
+cheshirecat_python_sdk/models/api/tokens.py,sha256=tJRx96ZrOS66anE9F5EvBLgd5aKNbrgn4EwpIvYaruQ,442
 cheshirecat_python_sdk/models/api/users.py,sha256=2l86y6ol0R0823gGl4bmc1TzGP4ZyvTm9AR8riKiatA,160
 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.7.4.dist-info/METADATA,sha256=R1wHR1rsC5gYTtJFJ-zM02uo_qY2S2yJa-i8Bh5i4eM,44136
-cheshirecat_python_sdk-1.7.4.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-cheshirecat_python_sdk-1.7.4.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-cheshirecat_python_sdk-1.7.4.dist-info/RECORD,,
+cheshirecat_python_sdk-1.7.6.dist-info/METADATA,sha256=IHhoqsj7YiRvzbu-DDWoweHhwPQb7lVLmJs0t9rTKXk,44136
+cheshirecat_python_sdk-1.7.6.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+cheshirecat_python_sdk-1.7.6.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+cheshirecat_python_sdk-1.7.6.dist-info/RECORD,,