cheshirecat-python-sdk 1.8.2__tar.gz → 1.8.4__tar.gz

{cheshirecat_python_sdk-1.8.2 → cheshirecat_python_sdk-1.8.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cheshirecat-python-sdk
-Version: 1.8.2
+Version: 1.8.4
 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.2 → cheshirecat_python_sdk-1.8.4}/cheshirecat_python_sdk/endpoints/memory.py

@@ -12,7 +12,7 @@ from cheshirecat_python_sdk.models.api.memories import (
     MemoryPointsOutput,
 )
 from cheshirecat_python_sdk.models.api.nested.memories import CollectionsItem
-from cheshirecat_python_sdk.models.dtos import Why, MemoryPoint
+from cheshirecat_python_sdk.models.dtos import Why, MemoryPoint, FilterSource
 
 
 class MemoryEndpoint(AbstractEndpoint):
@@ -242,4 +242,31 @@ class MemoryEndpoint(AbstractEndpoint):
             query=query,
         )
 
+    def has_source(self, agent_id: str, filter_source: FilterSource, chat_id: str | None = None) -> bool:
+        """
+        Checks if the given filter source exists for a specified agent.
+
+        This method determines whether a specific source, defined by the given filter_source, is associated with the
+        provided agent in the memory database.
+        It optionally considers a specific chat ID when filtering the data. The result indicates the existence of such a
+        source.
+
+        Args:
+            agent_id: Unique identifier of the agent for which the check is performed.
+            filter_source: FilterSource object that defines the source or hash to check.
+            chat_id: An optional chat identifier to narrow the scope of the check. If not provided, the check will
+                operate in a broader context.
+
+        Returns:
+            A boolean value indicating whether the specified filter source exists in the agent's memory database.
+        """
+        metadata = {"source": filter_source.source} if filter_source.source else {"hash": filter_source.hash}
+        if chat_id:
+            metadata["chat_id"] = chat_id
+
+        collection_name = "declarative" if chat_id is None else "episodic"
+        points = self.get_memory_points(collection_name, agent_id, metadata=metadata)
+
+        return len(points.points) > 0
+
     # END Memory Points API

{cheshirecat_python_sdk-1.8.2 → cheshirecat_python_sdk-1.8.4}/cheshirecat_python_sdk/models/dtos.py

@@ -1,5 +1,5 @@
 from typing import Dict, List, Any
-from pydantic import BaseModel, Field
+from pydantic import BaseModel, Field, model_validator
 
 
 class MemoryPoint(BaseModel):
@@ -26,3 +26,13 @@ class Why(BaseModel):
     input: str | None = None
     intermediate_steps: List | None = Field(default_factory=list)
     memory: List | None = Field(default_factory=list)
+
+
+class FilterSource(BaseModel):
+    source: str | None = None
+    hash: str | None = None
+
+    @model_validator(mode="after")
+    def validate_source_or_hash(self):
+        assert self.source or self.hash, "Either source or hash must be provided"
+        return self

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