solana-agent 19.0.1__tar.gz → 19.1.0__tar.gz

{solana_agent-19.0.1 → solana_agent-19.1.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: solana-agent
-Version: 19.0.1
+Version: 19.1.0
 Summary: Agentic IQ
 License: MIT
 Keywords: ai,openai,ai agents,agi
@@ -13,8 +13,8 @@ Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Classifier: Programming Language :: Python :: 3 :: Only
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
-Requires-Dist: openai (>=1.68.2,<2.0.0)
-Requires-Dist: pydantic (>=2.10.6,<3.0.0)
+Requires-Dist: openai (>=1.69.0,<2.0.0)
+Requires-Dist: pydantic (>=2.11.0,<3.0.0)
 Requires-Dist: pymongo (>=4.11.3,<5.0.0)
 Requires-Dist: zep-cloud (>=2.8.0,<3.0.0)
 Requires-Dist: zep-python (>=2.0.2,<3.0.0)
@@ -35,41 +35,36 @@ Description-Content-Type: text/markdown
 
 ## Agentic IQ
 
-Power your business using Solana Agent!
+Build your AI business in three lines of code!
 
-* **Brand:** AI agents will speak in your brand voice
-* **Expert:** The AI agent with the most expertise will answer your users' inquiries
-* **Extend:** Assign tools for your AI agents to perform actions
+## Why?
+* Three lines of code required
+* Multi-Modal Streaming
+* Conversational Memory & History
+* Intelligent Routing
+* Business Alignment
+* Extensible Tooling
+* Simple Business Definition
+* Tested & Secure
+* Built in Python
+* Deployed by [CometHeart](https://cometheart.com) & [WalletBubbles](https://walletbubbles.com)
 
 ## Features
 
 * Seamless text and audio streaming with real-time multi-modal processing
-* Persistent memory that preserves context across all agent interactions (optional)
+* Persistent memory that preserves context across all agent interactions
+* Streamlined message history for all agent interactions
 * Intelligent query routing to agents with optimal domain expertise
 * Unified value system ensuring brand-aligned agent responses
 * Powerful tool integration using standard Python packages and/or inline classes
 * Assigned tools are utilized by agents automatically and effectively
 * Simple business definition using JSON
 
-## Why?
-* Multi-Modal Streaming
-* Conversational Memory
-* Intelligent Routing
-* Business Alignment
-* Extensible Tooling
-* Simple Business Definition
-* Built using OOP & SOLID in Python - the language of AI
-* Batteries Included
-* Small Library: ~2,000 LOC of code & ~1,200 LOC of tests
-* Few Dependencies
-* Well Tested
-* Used in production by [CometHeart](https://cometheart.com) and [WalletBubbles](https://walletbubbles.com)
-
 ## Stack
 
 * [Python](https://python.org) - Programming Language
 * [OpenAI](https://openai.com) - LLMs
-* [MongoDB](https://mongodb.com) - Database
+* [MongoDB](https://mongodb.com) - Conversational History (optional)
 * [Zep](https://getzep.com) - Conversational Memory (optional)
 
 ## Installation
@@ -84,7 +79,7 @@ You can install Solana Agent using pip:
 from solana_agent import SolanaAgent
 
 config = {
-    "business": {
+    "business": { # optional
         "mission": "To provide users with a one-stop shop for their queries.",
         "values": {
             "Friendliness": "Users must be treated fairly, openly, and with friendliness.",
@@ -95,17 +90,17 @@ config = {
         ],
         "voice": "The voice of the brand is that of a research business."
     },
-    "mongo": {
+    "mongo": { # optional
         "connection_string": "mongodb://localhost:27017",
         "database": "solana_agent"
     },
-    "openai": {
-        "api_key": "your-openai-api-key",
-    },
     "zep": { # optional
         "api_key": "your-zep-api-key",
         "base_url": "your-zep-base-url", # not applicable if using Zep Cloud
     },
+    "openai": {
+        "api_key": "your-openai-api-key",
+    },
     "agents": [
         {
             "name": "research_specialist",
@@ -136,7 +131,7 @@ Plugins like Solana Agent Kit (sakit) integrate automatically with Solana Agent.
 from solana_agent import SolanaAgent
 
 config = {
-    "business": {
+    "business": { # optional
         "mission": "To provide users with a one-stop shop for their queries.",
         "values": {
             "Friendliness": "Users must be treated fairly, openly, and with friendliness.",
@@ -147,23 +142,26 @@ config = {
         ],
         "voice": "The voice of the brand is that of a research business."
     },
-    "mongo": {
+    "openai": { # optional
+        "api_key": "your-openai-api-key",
+    },
+    "ollama": { # optional
+        "url": "your-ollama-url",
+    },
+    "mongo": { # optional
         "connection_string": "mongodb://localhost:27017",
         "database": "solana_agent"
     },
-    "openai": {
-        "api_key": "your-openai-api-key",
-    },
     "zep": { # optional
         "api_key": "your-zep-api-key",
         "base_url": "your-zep-base-url", # not applicable if using Zep Cloud
     },
     "tools": {
-    "search_internet": {
-        "api_key": "your-perplexity-key", # Required
-        "citations": True, # Optional, defaults to True
-        "model": "sonar"  # Optional, defaults to "sonar"
-    },
+        "search_internet": {
+            "api_key": "your-perplexity-key", # Required
+            "citations": True, # Optional, defaults to True
+            "model": "sonar"  # Optional, defaults to "sonar"
+        },
     },
     "agents": [
         {
@@ -241,7 +239,7 @@ class TestTool(Tool):
             }
 
 config = {
-    "business": {
+    "business": { # optional
         "mission": "To provide users with a one-stop shop for their queries.",
         "values": {
             "Friendliness": "Users must be treated fairly, openly, and with friendliness.",
@@ -252,13 +250,16 @@ config = {
         ],
         "voice": "The voice of the brand is that of a research business."
     },
-    "mongo": {
+    "openai": { # optional
+        "api_key": "your-openai-api-key",
+    },
+    "ollama": { # optional
+        "url": "your-ollama-url",
+    },
+    "mongo": { # optional
         "connection_string": "mongodb://localhost:27017",
         "database": "solana_agent"
     },
-    "openai": {
-        "api_key": "your-openai-api-key",
-    },
     "zep": { # optional
         "api_key": "your-zep-api-key",
         "base_url": "your-zep-base-url", # not applicable if using Zep Cloud
@@ -287,24 +288,32 @@ async for response in solana_agent.process("user123", "What are the latest AI de
     print(response, end="")
 ```
 
-## Notes on Tools
+## Notes
 * Solana Agent agents can only call one tool per response.
 * Solana Agent agents choose the best tool for the job.
 * Solana Agent tools do not use OpenAI function calling.
 * Solana Agent tools are async functions.
+* Solana Agent will use OpenAI for audio and Ollama and for text if both config vars are set
+
+## Local Setup
+
+A Docker Compose and Zep Config file is available at the root of this project
 
 ## API Documentation
-* Available at [Solana Agent Documentation Site](https://docs.solana-agent.com)
 
-## Solana Agent Kit
+The official up-to-date documentation site
+
+[Solana Agent Documentation Site](https://docs.solana-agent.com)
+
+## Official Tools
 
-A collection of Solana Agent tools
+The official collection of tools in one plugin
 
 [Solana Agent Kit](https://github.com/truemagic-coder/solana-agent-kit)
 
 ## Example App
 
-A Solana Agent example app written in FastAPI and Next.js
+The official example app written in FastAPI and Next.js
 
 [Solana Agent Example App](https://github.com/truemagic-coder/solana-agent-app)
 

{solana_agent-19.0.1 → solana_agent-19.1.0}/README.md

@@ -12,41 +12,36 @@
 
 ## Agentic IQ
 
-Power your business using Solana Agent!
+Build your AI business in three lines of code!
 
-* **Brand:** AI agents will speak in your brand voice
-* **Expert:** The AI agent with the most expertise will answer your users' inquiries
-* **Extend:** Assign tools for your AI agents to perform actions
+## Why?
+* Three lines of code required
+* Multi-Modal Streaming
+* Conversational Memory & History
+* Intelligent Routing
+* Business Alignment
+* Extensible Tooling
+* Simple Business Definition
+* Tested & Secure
+* Built in Python
+* Deployed by [CometHeart](https://cometheart.com) & [WalletBubbles](https://walletbubbles.com)
 
 ## Features
 
 * Seamless text and audio streaming with real-time multi-modal processing
-* Persistent memory that preserves context across all agent interactions (optional)
+* Persistent memory that preserves context across all agent interactions
+* Streamlined message history for all agent interactions
 * Intelligent query routing to agents with optimal domain expertise
 * Unified value system ensuring brand-aligned agent responses
 * Powerful tool integration using standard Python packages and/or inline classes
 * Assigned tools are utilized by agents automatically and effectively
 * Simple business definition using JSON
 
-## Why?
-* Multi-Modal Streaming
-* Conversational Memory
-* Intelligent Routing
-* Business Alignment
-* Extensible Tooling
-* Simple Business Definition
-* Built using OOP & SOLID in Python - the language of AI
-* Batteries Included
-* Small Library: ~2,000 LOC of code & ~1,200 LOC of tests
-* Few Dependencies
-* Well Tested
-* Used in production by [CometHeart](https://cometheart.com) and [WalletBubbles](https://walletbubbles.com)
-
 ## Stack
 
 * [Python](https://python.org) - Programming Language
 * [OpenAI](https://openai.com) - LLMs
-* [MongoDB](https://mongodb.com) - Database
+* [MongoDB](https://mongodb.com) - Conversational History (optional)
 * [Zep](https://getzep.com) - Conversational Memory (optional)
 
 ## Installation
@@ -61,7 +56,7 @@ You can install Solana Agent using pip:
 from solana_agent import SolanaAgent
 
 config = {
-    "business": {
+    "business": { # optional
         "mission": "To provide users with a one-stop shop for their queries.",
         "values": {
             "Friendliness": "Users must be treated fairly, openly, and with friendliness.",
@@ -72,17 +67,17 @@ config = {
         ],
         "voice": "The voice of the brand is that of a research business."
     },
-    "mongo": {
+    "mongo": { # optional
         "connection_string": "mongodb://localhost:27017",
         "database": "solana_agent"
     },
-    "openai": {
-        "api_key": "your-openai-api-key",
-    },
     "zep": { # optional
         "api_key": "your-zep-api-key",
         "base_url": "your-zep-base-url", # not applicable if using Zep Cloud
     },
+    "openai": {
+        "api_key": "your-openai-api-key",
+    },
     "agents": [
         {
             "name": "research_specialist",
@@ -113,7 +108,7 @@ Plugins like Solana Agent Kit (sakit) integrate automatically with Solana Agent.
 from solana_agent import SolanaAgent
 
 config = {
-    "business": {
+    "business": { # optional
         "mission": "To provide users with a one-stop shop for their queries.",
         "values": {
             "Friendliness": "Users must be treated fairly, openly, and with friendliness.",
@@ -124,23 +119,26 @@ config = {
         ],
         "voice": "The voice of the brand is that of a research business."
     },
-    "mongo": {
+    "openai": { # optional
+        "api_key": "your-openai-api-key",
+    },
+    "ollama": { # optional
+        "url": "your-ollama-url",
+    },
+    "mongo": { # optional
         "connection_string": "mongodb://localhost:27017",
         "database": "solana_agent"
     },
-    "openai": {
-        "api_key": "your-openai-api-key",
-    },
     "zep": { # optional
         "api_key": "your-zep-api-key",
         "base_url": "your-zep-base-url", # not applicable if using Zep Cloud
     },
     "tools": {
-    "search_internet": {
-        "api_key": "your-perplexity-key", # Required
-        "citations": True, # Optional, defaults to True
-        "model": "sonar"  # Optional, defaults to "sonar"
-    },
+        "search_internet": {
+            "api_key": "your-perplexity-key", # Required
+            "citations": True, # Optional, defaults to True
+            "model": "sonar"  # Optional, defaults to "sonar"
+        },
     },
     "agents": [
         {
@@ -218,7 +216,7 @@ class TestTool(Tool):
             }
 
 config = {
-    "business": {
+    "business": { # optional
         "mission": "To provide users with a one-stop shop for their queries.",
         "values": {
             "Friendliness": "Users must be treated fairly, openly, and with friendliness.",
@@ -229,13 +227,16 @@ config = {
         ],
         "voice": "The voice of the brand is that of a research business."
     },
-    "mongo": {
+    "openai": { # optional
+        "api_key": "your-openai-api-key",
+    },
+    "ollama": { # optional
+        "url": "your-ollama-url",
+    },
+    "mongo": { # optional
         "connection_string": "mongodb://localhost:27017",
         "database": "solana_agent"
     },
-    "openai": {
-        "api_key": "your-openai-api-key",
-    },
     "zep": { # optional
         "api_key": "your-zep-api-key",
         "base_url": "your-zep-base-url", # not applicable if using Zep Cloud
@@ -264,24 +265,32 @@ async for response in solana_agent.process("user123", "What are the latest AI de
     print(response, end="")
 ```
 
-## Notes on Tools
+## Notes
 * Solana Agent agents can only call one tool per response.
 * Solana Agent agents choose the best tool for the job.
 * Solana Agent tools do not use OpenAI function calling.
 * Solana Agent tools are async functions.
+* Solana Agent will use OpenAI for audio and Ollama and for text if both config vars are set
+
+## Local Setup
+
+A Docker Compose and Zep Config file is available at the root of this project
 
 ## API Documentation
-* Available at [Solana Agent Documentation Site](https://docs.solana-agent.com)
 
-## Solana Agent Kit
+The official up-to-date documentation site
+
+[Solana Agent Documentation Site](https://docs.solana-agent.com)
+
+## Official Tools
 
-A collection of Solana Agent tools
+The official collection of tools in one plugin
 
 [Solana Agent Kit](https://github.com/truemagic-coder/solana-agent-kit)
 
 ## Example App
 
-A Solana Agent example app written in FastAPI and Next.js
+The official example app written in FastAPI and Next.js
 
 [Solana Agent Example App](https://github.com/truemagic-coder/solana-agent-app)
 

{solana_agent-19.0.1 → solana_agent-19.1.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "solana-agent"
-version = "19.0.1"
+version = "19.1.0"
 description = "Agentic IQ"
 authors = ["Bevan Hunt <bevan@bevanhunt.com>"]
 license = "MIT"
@@ -18,8 +18,8 @@ python_paths = [".", "tests"]
 
 [tool.poetry.dependencies]
 python = ">=3.12,<4.0"
-openai = "^1.68.2"
-pydantic = "^2.10.6"
+openai = "^1.69.0"
+pydantic = "^2.11.0"
 pymongo = "^4.11.3"
 zep-cloud = "^2.8.0"
 zep-python = "^2.0.2"

{solana_agent-19.0.1 → solana_agent-19.1.0}/solana_agent/adapters/llm_adapter.py

@@ -135,19 +135,6 @@ class OpenAIAdapter(LLMProvider):
             print(traceback.format_exc())
             yield f"I apologize, but I encountered an error: {str(e)}"
 
-    def generate_embedding(self, text: str) -> List[float]:  # pragma: no cover
-        """Generate embeddings for a given text using OpenAI's embedding model."""
-        try:
-            response = self.client.embeddings.create(
-                model="text-embedding-3-small",
-                input=text
-            )
-            return response.data[0].embedding
-        except Exception as e:
-            print(f"Error generating embedding: {e}")
-            # Return a zero vector as fallback (not ideal but prevents crashing)
-            return [0.0] * 1536  # Standard size for text-embedding-3-small
-
     async def parse_structured_output(
         self,
         prompt: str,

{solana_agent-19.0.1 → solana_agent-19.1.0}/solana_agent/client/solana_agent.py

@@ -44,6 +44,7 @@ class SolanaAgent(SolanaAgentInterface):
         self,
         user_id: str,
         message: Union[str, bytes],
+        prompt: Optional[str] = None,
         output_format: Literal["text", "audio"] = "text",
         audio_voice: Literal["alloy", "ash", "ballad", "coral", "echo",
                              "fable", "onyx", "nova", "sage", "shimmer"] = "nova",
@@ -59,6 +60,7 @@ class SolanaAgent(SolanaAgentInterface):
         Args:
             user_id: User ID
             message: Text message or audio bytes
+            prompt: Optional prompt for the agent
             output_format: Response format ("text" or "audio")
             audio_voice: Voice to use for audio output
             audio_instructions: Optional instructions for audio synthesis
@@ -76,6 +78,7 @@ class SolanaAgent(SolanaAgentInterface):
             audio_instructions=audio_instructions,
             audio_output_format=audio_output_format,
             audio_input_format=audio_input_format,
+            prompt=prompt,
         ):
             yield chunk
 

{solana_agent-19.0.1 → solana_agent-19.1.0}/solana_agent/factories/agent_factory.py

@@ -37,10 +37,19 @@ class SolanaAgentFactory:
             Configured QueryService instance
         """
         # Create adapters
-        db_adapter = MongoDBAdapter(
-            connection_string=config["mongo"]["connection_string"],
-            database_name=config["mongo"]["database"],
-        )
+
+        if "mongo" in config:
+            # MongoDB connection string and database name
+            if "connection_string" not in config["mongo"]:
+                raise ValueError("MongoDB connection string is required.")
+            if "database" not in config["mongo"]:
+                raise ValueError("MongoDB database name is required.")
+            db_adapter = MongoDBAdapter(
+                connection_string=config["mongo"]["connection_string"],
+                database_name=config["mongo"]["database"],
+            )
+        else:
+            db_adapter = None
 
         llm_adapter = OpenAIAdapter(
             api_key=config["openai"]["api_key"],
@@ -59,12 +68,25 @@ class SolanaAgentFactory:
             )
 
         # Create repositories
-        if "zep" in config:
+        memory_provider = None
+
+        if "zep" in config and "mongo" in config:
+            if "api_key" not in config["zep"]:
+                raise ValueError("Zep API key is required.")
             memory_provider = MemoryRepository(
                 db_adapter, config["zep"].get("api_key"), config["zep"].get("base_url"))
-        else:
+
+        if "mongo" in config and not "zep" in config:
             memory_provider = MemoryRepository(db_adapter)
 
+        if "zep" in config and not "mongo" in config:
+            if "api_key" not in config["zep"]:
+                raise ValueError("Zep API key is required.")
+            memory_provider = MemoryRepository(
+                zep_api_key=config["zep"].get("api_key"),
+                zep_base_url=config["zep"].get("base_url")
+            )
+
         # Create primary services
         agent_service = AgentService(
             llm_provider=llm_adapter,

{solana_agent-19.0.1 → solana_agent-19.1.0}/solana_agent/interfaces/client/client.py

@@ -1,5 +1,5 @@
 from abc import ABC, abstractmethod
-from typing import Any, AsyncGenerator, Dict, Literal, Union
+from typing import Any, AsyncGenerator, Dict, Literal, Optional, Union
 
 from solana_agent.interfaces.plugins.plugins import Tool
 
@@ -15,12 +15,13 @@ class SolanaAgent(ABC):
         output_format: Literal["text", "audio"] = "text",
         audio_voice: Literal["alloy", "ash", "ballad", "coral", "echo",
                              "fable", "onyx", "nova", "sage", "shimmer"] = "nova",
-        audio_instructions: str = None,
+        audio_instructions: Optional[str] = None,
         audio_output_format: Literal['mp3', 'opus',
                                      'aac', 'flac', 'wav', 'pcm'] = "aac",
         audio_input_format: Literal[
             "flac", "mp3", "mp4", "mpeg", "mpga", "m4a", "ogg", "wav", "webm"
         ] = "mp4",
+        prompt: Optional[str] = None,
     ) -> AsyncGenerator[Union[str, bytes], None]:
         """Process a user message and return the response stream."""
         pass

{solana_agent-19.0.1 → solana_agent-19.1.0}/solana_agent/interfaces/providers/llm.py

@@ -19,11 +19,6 @@ class LLMProvider(ABC):
         """Generate text from the language model."""
         pass
 
-    @abstractmethod
-    def generate_embedding(self, text: str) -> List[float]:
-        """Generate embedding vector for text."""
-        pass
-
     @abstractmethod
     async def parse_structured_output(
         self, prompt: str, system_prompt: str, model_class: Type[T],

{solana_agent-19.0.1 → solana_agent-19.1.0}/solana_agent/interfaces/services/agent.py

@@ -1,5 +1,5 @@
 from abc import ABC, abstractmethod
-from typing import Any, AsyncGenerator, Dict, List, Literal, Union
+from typing import Any, AsyncGenerator, Dict, List, Literal, Optional, Union
 
 from solana_agent.domains.agent import AIAgent
 
@@ -17,11 +17,6 @@ class AgentService(ABC):
         """Get the system prompt for an agent."""
         pass
 
-    @abstractmethod
-    def get_specializations(self) -> Dict[str, str]:
-        """Get all registered specializations."""
-        pass
-
     @abstractmethod
     async def generate_response(
         self,
@@ -32,12 +27,13 @@ class AgentService(ABC):
         output_format: Literal["text", "audio"] = "text",
         audio_voice: Literal["alloy", "ash", "ballad", "coral", "echo",
                              "fable", "onyx", "nova", "sage", "shimmer"] = "nova",
-        audio_instructions: str = None,
+        audio_instructions: Optional[str] = None,
         audio_output_format: Literal['mp3', 'opus',
                                      'aac', 'flac', 'wav', 'pcm'] = "aac",
         audio_input_format: Literal[
             "flac", "mp3", "mp4", "mpeg", "mpga", "m4a", "ogg", "wav", "webm"
         ] = "mp4",
+        prompt: Optional[str] = None,
     ) -> AsyncGenerator[Union[str, bytes], None]:
         """Generate a response from an agent."""
         pass

{solana_agent-19.0.1 → solana_agent-19.1.0}/solana_agent/interfaces/services/query.py

@@ -19,6 +19,7 @@ class QueryService(ABC):
         audio_input_format: Literal[
             "flac", "mp3", "mp4", "mpeg", "mpga", "m4a", "ogg", "wav", "webm"
         ] = "mp4",
+        prompt: Optional[str] = None,
     ) -> AsyncGenerator[Union[str, bytes], None]:
         """Process the user request and generate a response."""
         pass

{solana_agent-19.0.1 → solana_agent-19.1.0}/solana_agent/repositories/memory.py

@@ -12,19 +12,23 @@ class MemoryRepository(MemoryProvider):
 
     def __init__(
         self,
-        mongo_adapter: MongoDBAdapter,
+        mongo_adapter: Optional[MongoDBAdapter] = None,
         zep_api_key: Optional[str] = None,
         zep_base_url: Optional[str] = None
     ):
         """Initialize the combined memory provider."""
-        # Initialize MongoDB
-        self.mongo = mongo_adapter
-        self.collection = "conversations"
+        if not mongo_adapter:
+            self.mongo = None
+            self.collection = None
+        else:
+            # Initialize MongoDB
+            self.mongo = mongo_adapter
+            self.collection = "conversations"
 
-        # Ensure MongoDB collection and indexes
-        self.mongo.create_collection(self.collection)
-        self.mongo.create_index(self.collection, [("user_id", 1)])
-        self.mongo.create_index(self.collection, [("timestamp", 1)])
+            # Ensure MongoDB collection and indexes
+            self.mongo.create_collection(self.collection)
+            self.mongo.create_index(self.collection, [("user_id", 1)])
+            self.mongo.create_index(self.collection, [("timestamp", 1)])
 
         # Initialize Zep
         if zep_api_key and not zep_base_url:
@@ -37,22 +41,23 @@ class MemoryRepository(MemoryProvider):
     async def store(self, user_id: str, messages: List[Dict[str, Any]]) -> None:
         """Store messages in both Zep and MongoDB."""
         # Store in MongoDB as single document
-        try:
-            # Extract user and assistant messages
-            user_message = next(msg["content"]
-                                for msg in messages if msg["role"] == "user")
-            assistant_message = next(
-                msg["content"] for msg in messages if msg["role"] == "assistant")
-
-            doc = {
-                "user_id": user_id,
-                "user_message": user_message,
-                "assistant_message": assistant_message,
-                "timestamp": datetime.now(timezone.utc)
-            }
-            self.mongo.insert_one(self.collection, doc)
-        except Exception as e:
-            print(f"MongoDB storage error: {e}")
+        if self.mongo:
+            try:
+                # Extract user and assistant messages
+                user_message = next(msg["content"]
+                                    for msg in messages if msg["role"] == "user")
+                assistant_message = next(
+                    msg["content"] for msg in messages if msg["role"] == "assistant")
+
+                doc = {
+                    "user_id": user_id,
+                    "user_message": user_message,
+                    "assistant_message": assistant_message,
+                    "timestamp": datetime.now(timezone.utc)
+                }
+                self.mongo.insert_one(self.collection, doc)
+            except Exception as e:
+                print(f"MongoDB storage error: {e}")
 
         # Store in Zep with role-based format
         if not self.zep:
@@ -99,13 +104,14 @@ class MemoryRepository(MemoryProvider):
 
     async def delete(self, user_id: str) -> None:
         """Delete memory from both systems."""
-        try:
-            self.mongo.delete_all(
-                self.collection,
-                {"user_id": user_id}
-            )
-        except Exception as e:
-            print(f"MongoDB deletion error: {e}")
+        if self.mongo:
+            try:
+                self.mongo.delete_all(
+                    self.collection,
+                    {"user_id": user_id}
+                )
+            except Exception as e:
+                print(f"MongoDB deletion error: {e}")
 
         if not self.zep:
             return
@@ -125,9 +131,14 @@ class MemoryRepository(MemoryProvider):
         skip: int = 0
     ) -> List[Dict]:
         """Find documents matching query."""
+        if not self.mongo:
+            return []
         return self.mongo.find(collection, query, sort=sort, limit=limit, skip=skip)
 
     def count_documents(self, collection: str, query: Dict) -> int:
+        """Count documents matching query."""
+        if not self.mongo:
+            return 0
         return self.mongo.count_documents(collection, query)
 
     def _truncate(self, text: str, limit: int = 2500) -> str:

{solana_agent-19.0.1 → solana_agent-19.1.0}/solana_agent/services/agent.py

@@ -108,20 +108,6 @@ class AgentService(AgentServiceInterface):
         """
         return {agent.name: agent for agent in self.agents}
 
-    def get_specializations(self) -> Dict[str, str]:
-        """Get all registered specializations.
-
-        Returns:
-            Dictionary mapping specialization names to descriptions
-        """
-        specializations = {}
-
-        for agent in self.agents:
-            if agent.specialization:
-                specializations[agent.specialization] = f"AI expertise in {agent.specialization}"
-
-        return specializations
-
     def assign_tool_for_agent(self, agent_name: str, tool_name: str) -> bool:
         """Assign a tool to an agent.
 
@@ -187,6 +173,7 @@ class AgentService(AgentServiceInterface):
         audio_input_format: Literal[
             "flac", "mp3", "mp4", "mpeg", "mpga", "m4a", "ogg", "wav", "webm"
         ] = "mp4",
+        prompt: Optional[str] = None,
     ) -> AsyncGenerator[Union[str, bytes], None]:  # pragma: no cover
         """Generate a response with support for text/audio input/output.
 
@@ -200,6 +187,7 @@ class AgentService(AgentServiceInterface):
             audio_instructions: Optional instructions for audio synthesis
             audio_output_format: Audio output format
             audio_input_format: Audio input format
+            prompt: Optional prompt for the agent
 
         Yields:
             Text chunks or audio bytes depending on output_format
@@ -233,7 +221,9 @@ class AgentService(AgentServiceInterface):
             # Add User ID and memory context
             system_prompt += f"\n\nUser ID: {user_id}"
             if memory_context:
-                system_prompt += f"\n\nMemory Context: {memory_context}"
+                system_prompt += f"\n\nMEMORY CONTEXT: {memory_context}"
+            if prompt:
+                system_prompt += f"\n\nADDITIONAL PROMPT: {prompt}"
 
             # Keep track of the complete text response
             complete_text_response = ""

{solana_agent-19.0.1 → solana_agent-19.1.0}/solana_agent/services/query.py

@@ -46,14 +46,19 @@ class QueryService(QueryServiceInterface):
         audio_input_format: Literal[
             "flac", "mp3", "mp4", "mpeg", "mpga", "m4a", "ogg", "wav", "webm"
         ] = "mp4",
+        prompt: Optional[str] = None,
     ) -> AsyncGenerator[Union[str, bytes], None]:  # pragma: no cover
         """Process the user request with appropriate agent.
 
         Args:
             user_id: User ID
             query: Text query or audio bytes
-            output_format: Response format ("text" or "audio") 
-            voice: Voice to use for audio output
+            output_format: Response format ("text" or "audio")
+            audio_voice: Voice for TTS (text-to-speech)
+            audio_instructions: Optional instructions for TTS
+            audio_output_format: Audio output format
+            audio_input_format: Audio input format
+            prompt: Optional prompt for the agent
 
         Yields:
             Response chunks (text strings or audio bytes)
@@ -105,7 +110,8 @@ class QueryService(QueryServiceInterface):
                     audio_voice=audio_voice,
                     audio_input_format=audio_input_format,
                     audio_output_format=audio_output_format,
-                    audio_instructions=audio_instructions
+                    audio_instructions=audio_instructions,
+                    prompt=prompt,
                 ):
                     yield audio_chunk
 
@@ -122,7 +128,8 @@ class QueryService(QueryServiceInterface):
                     user_id=user_id,
                     query=user_text,
                     memory_context=memory_context,
-                    output_format="text"
+                    output_format="text",
+                    prompt=prompt,
                 ):
                     yield chunk
                     full_text_response += chunk