cheshirecat-python-sdk 1.7.1__tar.gz → 1.7.3__tar.gz

{cheshirecat_python_sdk-1.7.1 → cheshirecat_python_sdk-1.7.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cheshirecat-python-sdk
-Version: 1.7.1
+Version: 1.7.3
 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.3/cheshirecat_python_sdk/endpoints/health_check.py

@@ -0,0 +1,22 @@
+from cheshirecat_python_sdk.endpoints.base import AbstractEndpoint
+
+class HealthCheckEndpoint(AbstractEndpoint):
+    def liveness(self):
+        """
+        This endpoint is used to check if the server is running.
+        :return: dict, the status of the server.
+        """
+        response = self.get_http_session().get("/health/liveness")
+        response.raise_for_status()
+
+        return response.json()
+
+    def readiness(self):
+        """
+        This endpoint is used to check if the server is running.
+        :return: dict, the status of the server.
+        """
+        response = self.get_http_session().get("/health/readiness")
+        response.raise_for_status()
+
+        return response.json()

{cheshirecat_python_sdk-1.7.1 → cheshirecat_python_sdk-1.7.3}/cheshirecat_python_sdk/endpoints/message.py

@@ -2,7 +2,7 @@ from typing import Callable
 import json
 
 from cheshirecat_python_sdk.endpoints.base import AbstractEndpoint
-from cheshirecat_python_sdk.models.api.messages import MessageOutput
+from cheshirecat_python_sdk.models.api.messages import ChatOutput
 from cheshirecat_python_sdk.models.dtos import Message
 from cheshirecat_python_sdk.utils import deserialize
 
@@ -14,19 +14,19 @@ class MessageEndpoint(AbstractEndpoint):
         agent_id: str,
         user_id: str,
         chat_id: str | None = None,
-    ) -> MessageOutput:
+    ) -> ChatOutput:
         """
         This endpoint sends a message to the agent identified by the agentId parameter. The message is sent via HTTP.
         :param message: Message object, the message to send
         :param agent_id: the agent id
         :param user_id: the user id
         :param chat_id: the chat id (optional)
-        :return: MessageOutput object
+        :return: ChatOutput object
         """
         return self.post_json(
             '/message',
             agent_id,
-            output_class=MessageOutput,
+            output_class=ChatOutput,
             payload=message.model_dump(),
             user_id=user_id,
             chat_id=chat_id,
@@ -39,7 +39,7 @@ class MessageEndpoint(AbstractEndpoint):
         user_id: str,
         chat_id: str | None = None,
         callback: Callable[[str], None] | None = None
-    ) -> MessageOutput:
+    ) -> ChatOutput:
         """
         This endpoint sends a message to the agent identified by the agentId parameter. The message is sent via WebSocket.
         :param message: Message object, the message to send
@@ -47,7 +47,7 @@ class MessageEndpoint(AbstractEndpoint):
         :param user_id: the user id
         :param chat_id: the chat id
         :param callback: callable, a callback function that will be called for each message received
-        :return: MessageOutput object
+        :return: ChatOutput object
         """
         try:
             json_data = json.dumps(message.model_dump())
@@ -74,4 +74,4 @@ class MessageEndpoint(AbstractEndpoint):
             raise Exception(f"WebSocket error: {str(e)}")
 
         await client.close()
-        return deserialize(json.loads(response), MessageOutput)
+        return deserialize(json.loads(response), ChatOutput)

{cheshirecat_python_sdk-1.7.1 → cheshirecat_python_sdk-1.7.3}/cheshirecat_python_sdk/models/api/messages.py

@@ -1,4 +1,4 @@
-from pydantic import Field
+from pydantic import Field, BaseModel
 
 from cheshirecat_python_sdk.models.dtos import MessageBase, Why
 
@@ -12,3 +12,10 @@ class MessageOutput(MessageBase):
     def __init__(self, **data):
         super().__init__(**data)
         self.content = self.text
+
+
+class ChatOutput(BaseModel):
+    agent_id: str
+    user_id: str
+    chat_id: str
+    message: MessageOutput

{cheshirecat_python_sdk-1.7.1 → cheshirecat_python_sdk-1.7.3}/pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 name = "cheshirecat-python-sdk"
 description = "Python SDK for the Cloud-ready fork of the Cheshire Cat"
-version = "1.7.1"
+version = "1.7.3"
 requires-python = ">=3.10"
 license = { file = "LICENSE" }
 authors = [

cheshirecat_python_sdk-1.7.1/cheshirecat_python_sdk/endpoints/health_check.py

@@ -1,12 +0,0 @@
-from cheshirecat_python_sdk.endpoints.base import AbstractEndpoint
-
-class HealthCheckEndpoint(AbstractEndpoint):
-    def home(self):
-        """
-        This endpoint is used to check if the server is running.
-        :return: dict, the status of the server.
-        """
-        response = self.get_http_session().get("/")
-        response.raise_for_status()
-
-        return response.json()