solana-agent 28.3.3__tar.gz → 29.1.0__tar.gz

{solana_agent-28.3.3 → solana_agent-29.1.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: solana-agent
-Version: 28.3.3
+Version: 29.1.0
 Summary: AI Agents for Solana
 License: MIT
 Keywords: solana,solana ai,solana agent,ai,ai agent,ai agents
@@ -18,7 +18,7 @@ Requires-Dist: instructor (==1.7.9)
 Requires-Dist: llama-index-core (==0.12.32)
 Requires-Dist: llama-index-embeddings-openai (==0.3.1)
 Requires-Dist: logfire (==3.14.1)
-Requires-Dist: openai (==1.76.2)
+Requires-Dist: openai (==1.77.0)
 Requires-Dist: pillow (==11.2.1)
 Requires-Dist: pinecone (==6.0.2)
 Requires-Dist: pydantic (>=2)
@@ -27,7 +27,7 @@ Requires-Dist: pypdf (==5.4.0)
 Requires-Dist: rich (>=13,<14.0)
 Requires-Dist: scrubadub (==2.0.1)
 Requires-Dist: typer (==0.15.3)
-Requires-Dist: zep-cloud (==2.11.0)
+Requires-Dist: zep-cloud (==2.12.1)
 Project-URL: Documentation, https://docs.solana-agent.com
 Project-URL: Homepage, https://solana-agent.com
 Project-URL: Repository, https://github.com/truemagic-coder/solana-agent
@@ -55,6 +55,7 @@ Build your AI agents in three lines of code!
 * Three lines of code setup
 * Simple Agent Definition
 * Fast Responses
+* Multi-Vendor Support
 * Solana Ecosystem Integration
 * Multi-Agent Swarm
 * Multi-Modal (Images & Audio & Text)
@@ -79,6 +80,7 @@ Build your AI agents in three lines of code!
 * Easy three lines of code setup
 * Simple agent definition using JSON
 * Fast AI responses
+* Multi-vendor support including OpenAI, Grok, and Gemini AI services
 * Solana Ecosystem Integration via [AgentiPy](https://github.com/niceberginc/agentipy)
 * MCP tool usage with first-class support for [Zapier](https://zapier.com/mcp)
 * Integrated observability and tracing via [Pydantic Logfire](https://pydantic.dev/logfire)
@@ -112,7 +114,8 @@ Build your AI agents in three lines of code!
 
 ### AI Models Used
 
-* [gpt-4.1](https://platform.openai.com/docs/models/gpt-4.1) (agent)
+**OpenAI**
+* [gpt-4.1](https://platform.openai.com/docs/models/gpt-4.1) (agent - can be overridden)
 * [gpt-4.1-nano](https://platform.openai.com/docs/models/gpt-4.1-nano) (router)
 * [text-embedding-3-large](https://platform.openai.com/docs/models/text-embedding-3-large) (embedding)
 * [tts-1](https://platform.openai.com/docs/models/tts-1) (audio TTS)
@@ -120,6 +123,12 @@ Build your AI agents in three lines of code!
 * [gpt-image-1](https://platform.openai.com/docs/models/gpt-image-1) (image generation)
 * [gpt-4o-mini-search-preview](https://platform.openai.com/docs/models/gpt-4o-mini-search-preview) (Internet search)
 
+**Grok**
+* [grok-3-fast](https://x.ai/api#pricing) (agent - optional)
+
+**Gemini**
+* [gemini-2.5-pro-preview-03-25](https://ai.google.dev/gemini-api/docs/models#gemini-2.5-pro-preview-03-25) (agent - optional)
+
 ## Installation
 
 You can install Solana Agent using pip:
@@ -421,9 +430,33 @@ config = {
 }
 ```
 
+### Grok
+
+`grok-3-fast` can be used instead of `gpt-4.1` for the agent model
+
+```python
+config = {
+    "grok": {
+        "api_key": "your-grok-api-key",
+    },
+}
+```
+
+### Gemini
+
+`gemini-2.5-pro-preview-03-25` can be used instead of `gpt-4.1` for the agent model
+
+```python
+config = {
+    "gemini": {
+        "api_key": "your-gemini-api-key",
+    },
+}
+```
+
 ### 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 = {

{solana_agent-28.3.3 → solana_agent-29.1.0}/README.md

@@ -20,6 +20,7 @@ Build your AI agents in three lines of code!
 * Three lines of code setup
 * Simple Agent Definition
 * Fast Responses
+* Multi-Vendor Support
 * Solana Ecosystem Integration
 * Multi-Agent Swarm
 * Multi-Modal (Images & Audio & Text)
@@ -44,6 +45,7 @@ Build your AI agents in three lines of code!
 * Easy three lines of code setup
 * Simple agent definition using JSON
 * Fast AI responses
+* Multi-vendor support including OpenAI, Grok, and Gemini AI services
 * Solana Ecosystem Integration via [AgentiPy](https://github.com/niceberginc/agentipy)
 * MCP tool usage with first-class support for [Zapier](https://zapier.com/mcp)
 * Integrated observability and tracing via [Pydantic Logfire](https://pydantic.dev/logfire)
@@ -77,7 +79,8 @@ Build your AI agents in three lines of code!
 
 ### AI Models Used
 
-* [gpt-4.1](https://platform.openai.com/docs/models/gpt-4.1) (agent)
+**OpenAI**
+* [gpt-4.1](https://platform.openai.com/docs/models/gpt-4.1) (agent - can be overridden)
 * [gpt-4.1-nano](https://platform.openai.com/docs/models/gpt-4.1-nano) (router)
 * [text-embedding-3-large](https://platform.openai.com/docs/models/text-embedding-3-large) (embedding)
 * [tts-1](https://platform.openai.com/docs/models/tts-1) (audio TTS)
@@ -85,6 +88,12 @@ Build your AI agents in three lines of code!
 * [gpt-image-1](https://platform.openai.com/docs/models/gpt-image-1) (image generation)
 * [gpt-4o-mini-search-preview](https://platform.openai.com/docs/models/gpt-4o-mini-search-preview) (Internet search)
 
+**Grok**
+* [grok-3-fast](https://x.ai/api#pricing) (agent - optional)
+
+**Gemini**
+* [gemini-2.5-pro-preview-03-25](https://ai.google.dev/gemini-api/docs/models#gemini-2.5-pro-preview-03-25) (agent - optional)
+
 ## Installation
 
 You can install Solana Agent using pip:
@@ -386,9 +395,33 @@ config = {
 }
 ```
 
+### Grok
+
+`grok-3-fast` can be used instead of `gpt-4.1` for the agent model
+
+```python
+config = {
+    "grok": {
+        "api_key": "your-grok-api-key",
+    },
+}
+```
+
+### Gemini
+
+`gemini-2.5-pro-preview-03-25` can be used instead of `gpt-4.1` for the agent model
+
+```python
+config = {
+    "gemini": {
+        "api_key": "your-gemini-api-key",
+    },
+}
+```
+
 ### 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 = {

{solana_agent-28.3.3 → solana_agent-29.1.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "solana-agent"
-version = "28.3.3"
+version = "29.1.0"
 description = "AI Agents for Solana"
 authors = ["Bevan Hunt <bevan@bevanhunt.com>"]
 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"

{solana_agent-28.3.3 → solana_agent-29.1.0}/solana_agent/adapters/mongodb_adapter.py

@@ -32,6 +32,16 @@ class MongoDBAdapter(DataStorageProvider):
         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-28.3.3 → solana_agent-29.1.0}/solana_agent/client/solana_agent.py

@@ -227,48 +227,6 @@ class SolanaAgent(SolanaAgentInterface):
         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-28.3.3 → solana_agent-29.1.0}/solana_agent/factories/agent_factory.py

@@ -168,6 +168,30 @@ class SolanaAgentFactory:
             output_guardrails=output_guardrails,
         )
 
+        if "gemini" in config and "api_key" in config["gemini"]:
+            # Create primary services
+            agent_service = AgentService(
+                llm_provider=llm_adapter,
+                business_mission=business_mission,
+                config=config,
+                api_key=config["gemini"]["api_key"],
+                base_url="https://generativelanguage.googleapis.com/v1beta/openai/",
+                model="gemini-2.5-pro-preview-03-25",
+                output_guardrails=output_guardrails,
+            )  # pragma: no cover
+
+        elif "grok" in config and "api_key" in config["grok"]:
+            # Create primary services
+            agent_service = AgentService(
+                llm_provider=llm_adapter,
+                business_mission=business_mission,
+                config=config,
+                api_key=config["grok"]["api_key"],
+                base_url="https://api.x.ai/v1",
+                model="grok-3-fast",
+                output_guardrails=output_guardrails,
+            )  # pragma: no cover
+
         # Create routing service
         routing_service = RoutingService(
             llm_provider=llm_adapter,

{solana_agent-28.3.3 → solana_agent-29.1.0}/solana_agent/interfaces/client/client.py

@@ -91,27 +91,6 @@ class SolanaAgent(ABC):
         """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-28.3.3 → solana_agent-29.1.0}/solana_agent/interfaces/services/knowledge_base.py

@@ -44,31 +44,6 @@ class KnowledgeBaseService(ABC):
         """
         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,