v29.0.0 (#104)

* wip

* update deps

* wip

* wip

* need to test

* pdfs fully working

* fix bug with doc text add

* fix document deletion

* fix failing test

Author: truemagic-coder

README.md

@@ -388,7 +388,7 @@ config = {
 
 ### Knowledge Base
 
-The Knowledge Base (KB) is meant to store text values and/or small PDFs.
+The Knowledge Base (KB) is meant to store text values and/or PDFs (extracts text) - can handle very large PDFs.
 
 ```python
 config = {

poetry.lock

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "solana-agent"
-version = "28.3.3"
+version = "29.0.0"
 description = "AI Agents for Solana"
 authors = ["Bevan Hunt <[email protected]>"]
 license = "MIT"
@@ -24,10 +24,10 @@ python_paths = [".", "tests"]
 
 [tool.poetry.dependencies]
 python = ">=3.12,<4.0"
-openai = "1.76.2"
+openai = "1.77.0"
 pydantic = ">=2"
 pymongo = "4.12.1"
-zep-cloud = "2.11.0"
+zep-cloud = "2.12.1"
 instructor = "1.7.9"
 pinecone = "6.0.2"
 llama-index-core = "0.12.32"

pyproject.toml

@@ -32,6 +32,16 @@ def insert_one(self, collection: str, document: Dict) -> str:
         self.db[collection].insert_one(document)
         return document["_id"]
 
+    def insert_many(self, collection: str, documents: List[Dict]) -> List[str]:
+        for document in documents:
+            if "_id" not in document:
+                document["_id"] = str(uuid.uuid4())
+        result = self.db[collection].insert_many(documents)
+        return [str(doc_id) for doc_id in result.inserted_ids]
+
+    def delete_many(self, collection: str, query: Dict):
+        return self.db[collection].delete_many(query)
+
     def find_one(self, collection: str, query: Dict) -> Optional[Dict]:
         return self.db[collection].find_one(query)
 

solana_agent/adapters/mongodb_adapter.py

@@ -227,48 +227,6 @@ async def kb_delete_document(
         kb = self._ensure_kb()
         return await kb.delete_document(document_id, namespace)
 
-    async def kb_update_document(
-        self,
-        document_id: str,
-        text: Optional[str] = None,
-        metadata: Optional[Dict[str, Any]] = None,
-        namespace: Optional[str] = None,
-    ) -> bool:
-        """
-        Update an existing document in the knowledge base.
-
-        Args:
-            document_id: ID of document to update.
-            text: Optional new text content.
-            metadata: Optional metadata to update.
-            namespace: Optional Pinecone namespace.
-
-        Returns:
-            True if successful.
-        """
-        kb = self._ensure_kb()
-        return await kb.update_document(document_id, text, metadata, namespace)
-
-    async def kb_add_documents_batch(
-        self,
-        documents: List[Dict[str, Any]],
-        namespace: Optional[str] = None,
-        batch_size: int = 50,
-    ) -> List[str]:
-        """
-        Add multiple documents to the knowledge base in batches.
-
-        Args:
-            documents: List of documents ({'text': ..., 'metadata': ...}).
-            namespace: Optional Pinecone namespace.
-            batch_size: Number of documents per batch.
-
-        Returns:
-            List of added document IDs.
-        """
-        kb = self._ensure_kb()
-        return await kb.add_documents_batch(documents, namespace, batch_size)
-
     async def kb_add_pdf_document(
         self,
         pdf_data: Union[bytes, str],

solana_agent/client/solana_agent.py

@@ -91,27 +91,6 @@ async def kb_delete_document(
         """Delete a document from the knowledge base."""
         pass
 
-    @abstractmethod
-    async def kb_update_document(
-        self,
-        document_id: str,
-        text: Optional[str] = None,
-        metadata: Optional[Dict[str, Any]] = None,
-        namespace: Optional[str] = None,
-    ) -> bool:
-        """Update an existing document in the knowledge base."""
-        pass
-
-    @abstractmethod
-    async def kb_add_documents_batch(
-        self,
-        documents: List[Dict[str, Any]],
-        namespace: Optional[str] = None,
-        batch_size: int = 50,
-    ) -> List[str]:
-        """Add multiple documents to the knowledge base in batches."""
-        pass
-
     @abstractmethod
     async def kb_add_pdf_document(
         self,

solana_agent/interfaces/client/client.py

@@ -44,31 +44,6 @@ async def delete_document(
         """
         pass
 
-    @abstractmethod
-    async def update_document(
-        self,
-        document_id: str,
-        text: Optional[str] = None,
-        metadata: Optional[Dict[str, Any]] = None,
-        namespace: Optional[str] = None,
-    ) -> bool:
-        """
-        Update an existing document in the knowledge base.
-        """
-        pass
-
-    @abstractmethod
-    async def add_documents_batch(
-        self,
-        documents: List[Dict[str, Any]],
-        namespace: Optional[str] = None,
-        batch_size: int = 50,
-    ) -> List[str]:
-        """
-        Add multiple documents in batches.
-        """
-        pass
-
     @abstractmethod
     async def add_pdf_document(
         self,