cheshirecat-python-sdk 1.7.11__tar.gz → 1.8.1__tar.gz

{cheshirecat_python_sdk-1.7.11 → cheshirecat_python_sdk-1.8.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cheshirecat-python-sdk
-Version: 1.7.11
+Version: 1.8.1
 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.11 → cheshirecat_python_sdk-1.8.1}/cheshirecat_python_sdk/builders/__init__.py

@@ -1,3 +1,3 @@
-from cheshirecat_python_sdk.builders.memory import MemoryBuilder, MemoryPointBuilder
+from cheshirecat_python_sdk.builders.memory import MemoryPointBuilder
 from cheshirecat_python_sdk.builders.settings_input import SettingInputBuilder
 from cheshirecat_python_sdk.builders.why import WhyBuilder

cheshirecat_python_sdk-1.8.1/cheshirecat_python_sdk/builders/memory.py

@@ -0,0 +1,25 @@
+from typing import Dict, Any
+
+from cheshirecat_python_sdk.builders.base import BaseBuilder
+from cheshirecat_python_sdk.models.dtos import MemoryPoint
+
+
+class MemoryPointBuilder(BaseBuilder):
+    def __init__(self):
+        self.content: str = ""
+        self.metadata: Dict[str, Any] = {}
+
+    @staticmethod
+    def create():
+        return MemoryPointBuilder()
+
+    def set_content(self, content: str):
+        self.content = content
+        return self
+
+    def set_metadata(self, metadata: Dict[str, Any]):
+        self.metadata = metadata
+        return self
+
+    def build(self):
+        return MemoryPoint(self.content, self.metadata)

{cheshirecat_python_sdk-1.7.11 → cheshirecat_python_sdk-1.8.1}/cheshirecat_python_sdk/builders/why.py

@@ -1,14 +1,14 @@
-from typing import Dict, Any
+from typing import List
 
 from cheshirecat_python_sdk.builders.base import BaseBuilder
-from cheshirecat_python_sdk.models.dtos import Memory, Why
+from cheshirecat_python_sdk.models.dtos import Why
 
 
 class WhyBuilder(BaseBuilder):
     def __init__(self):
         self.input: str | None = None
-        self.intermediate_steps: Dict[str, Any] | None = None
-        self.memory: Memory | None = None
+        self.intermediate_steps: List | None = []
+        self.memory: List | None = []
 
     @staticmethod
     def create() -> "WhyBuilder":
@@ -18,12 +18,12 @@ class WhyBuilder(BaseBuilder):
         self.input = input
         return self
 
-    def set_intermediate_steps(self, intermediate_steps: Dict[str, Any] | None = None) -> "WhyBuilder":
-        self.intermediate_steps = intermediate_steps
+    def set_intermediate_steps(self, intermediate_steps: List | None = None) -> "WhyBuilder":
+        self.intermediate_steps = intermediate_steps or []
         return self
 
-    def set_memory(self, memory: Memory) -> "WhyBuilder":
-        self.memory = memory
+    def set_memory(self, memory: List | None = None) -> "WhyBuilder":
+        self.memory = memory or []
         return self
 
     def build(self) -> Why:

{cheshirecat_python_sdk-1.7.11 → cheshirecat_python_sdk-1.8.1}/cheshirecat_python_sdk/client.py

@@ -18,6 +18,7 @@ from cheshirecat_python_sdk.endpoints import (
     UtilsEndpoint,
     VectorDatabaseEndpoint,
     HealthCheckEndpoint,
+    AgenticWorkflowEndpoint,
 )
 
 
@@ -64,6 +65,10 @@ class CheshireCatClient:
     def auth_handler(self):
         return AuthHandlerEndpoint(self)
 
+    @property
+    def agentic_workflow(self):
+        return AgenticWorkflowEndpoint(self)
+
     @property
     def chunker(self):
         return ChunkerEndpoint(self)

{cheshirecat_python_sdk-1.7.11 → cheshirecat_python_sdk-1.8.1}/cheshirecat_python_sdk/endpoints/__init__.py

@@ -1,4 +1,5 @@
 from cheshirecat_python_sdk.endpoints.admins import AdminsEndpoint
+from cheshirecat_python_sdk.endpoints.agentic_workflow import AgenticWorkflowEndpoint
 from cheshirecat_python_sdk.endpoints.auth import AuthEndpoint
 from cheshirecat_python_sdk.endpoints.auth_handler import AuthHandlerEndpoint
 from cheshirecat_python_sdk.endpoints.chunker import ChunkerEndpoint

cheshirecat_python_sdk-1.8.1/cheshirecat_python_sdk/endpoints/agentic_workflow.py

@@ -0,0 +1,52 @@
+from typing import Dict, Any
+
+from cheshirecat_python_sdk.endpoints.base import AbstractEndpoint
+from cheshirecat_python_sdk.models.api.factories import FactoryObjectSettingsOutput, FactoryObjectSettingOutput
+
+
+class AgenticWorkflowEndpoint(AbstractEndpoint):
+    def __init__(self, client: "CheshireCatClient"):
+        super().__init__(client)
+        self.prefix = "/agentic_workflow"
+
+    def get_agentic_workflows_settings(self, agent_id: str) -> FactoryObjectSettingsOutput:
+        """
+        Get all agentic workflow settings for the agent with the given ID.
+        :param agent_id: The ID of the agent.
+        :return: FactoryObjectSettingsOutput, containing the settings for all agentic workflows.
+        """
+        return self.get(
+            self.format_url("/settings"),
+            agent_id,
+            output_class=FactoryObjectSettingsOutput,
+        )
+
+    def get_agentic_workflow_settings(self, agentic_workflow: str, agent_id: str) -> FactoryObjectSettingOutput:
+        """
+        Get the settings for the agentic workflow with the given name.
+        :param agentic_workflow: The name of the agentic workflow.
+        :param agent_id: The ID of the agent.
+        :return: FactoryObjectSettingOutput, containing the settings for the agentic workflow.
+        """
+        return self.get(
+            self.format_url(f"/settings/{agentic_workflow}"),
+            agent_id,
+            output_class=FactoryObjectSettingOutput,
+        )
+
+    def put_agentic_workflow_settings(
+        self, agentic_workflow: str, agent_id: str, values: Dict[str, Any]
+    ) -> FactoryObjectSettingOutput:
+        """
+        Update the settings for the agentic workflow with the given name.
+        :param agentic_workflow: The name of the agentic workflow.
+        :param agent_id: The ID of the agent.
+        :param values: The new settings for the agentic workflow.
+        :return: FactoryObjectSettingOutput, containing the updated settings for the agentic workflow.
+        """
+        return self.put(
+            self.format_url(f"/settings/{agentic_workflow}"),
+            agent_id,
+            output_class=FactoryObjectSettingOutput,
+            payload=values,
+        )

{cheshirecat_python_sdk-1.7.11 → cheshirecat_python_sdk-1.8.1}/cheshirecat_python_sdk/models/dtos.py

@@ -2,10 +2,6 @@ from typing import Dict, List, Any
 from pydantic import BaseModel, Field
 
 
-class Memory(BaseModel):
-    declarative: List | None = Field(default_factory=list)
-
-
 class MemoryPoint(BaseModel):
     content: str
     metadata: Dict[str, Any]
@@ -29,4 +25,4 @@ class SettingInput(BaseModel):
 class Why(BaseModel):
     input: str | None = None
     intermediate_steps: List | None = Field(default_factory=list)
-    memory: Memory = Field(default_factory=Memory)
+    memory: List | None = Field(default_factory=list)

{cheshirecat_python_sdk-1.7.11 → cheshirecat_python_sdk-1.8.1}/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.11"
+version = "1.8.1"
 requires-python = ">=3.10"
 license = { file = "LICENSE" }
 authors = [

cheshirecat_python_sdk-1.7.11/cheshirecat_python_sdk/builders/memory.py

@@ -1,44 +0,0 @@
-from typing import Dict, Any
-
-from cheshirecat_python_sdk.builders.base import BaseBuilder
-from cheshirecat_python_sdk.models.dtos import Memory, MemoryPoint
-
-
-class MemoryBuilder(BaseBuilder):
-    def __init__(self):
-        self.declarative: Dict[str, Any] | None = None
-
-    @staticmethod
-    def create():
-        return MemoryBuilder()
-
-    def set_declarative(self, declarative: Dict[str, Any] | None = None):
-        self.declarative = declarative or {}
-        return self
-
-    def build(self):
-        memory = Memory()
-        memory.declarative = self.declarative
-
-        return memory
-
-
-class MemoryPointBuilder(BaseBuilder):
-    def __init__(self):
-        self.content: str = ""
-        self.metadata: Dict[str, Any] = {}
-
-    @staticmethod
-    def create():
-        return MemoryPointBuilder()
-
-    def set_content(self, content: str):
-        self.content = content
-        return self
-
-    def set_metadata(self, metadata: Dict[str, Any]):
-        self.metadata = metadata
-        return self
-
-    def build(self):
-        return MemoryPoint(self.content, self.metadata)