solana-agent 20.1.5__tar.gz → 21.1.0__tar.gz

{solana_agent-20.1.5 → solana_agent-21.1.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: solana-agent
-Version: 20.1.5
+Version: 21.1.0
 Summary: Agentic IQ
 License: MIT
 Keywords: ai,openai,ai agents,agi
@@ -14,10 +14,10 @@ Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
-Requires-Dist: openai (>=1.68.2,<2.0.0)
-Requires-Dist: pydantic (>=2.11.1,<3.0.0)
+Requires-Dist: openai (>=1.70.0,<2.0.0)
+Requires-Dist: pydantic (>=2.11.2,<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-cloud (>=2.9.0,<3.0.0)
 Requires-Dist: zep-python (>=2.0.2,<3.0.0)
 Project-URL: Documentation, https://docs.solana-agent.com
 Project-URL: Repository, https://github.com/truemagic-coder/solana-agent
@@ -44,6 +44,7 @@ Build your AI business in three lines of code!
 * Multi-Agent Swarm
 * Multi-Modal Streaming
 * Conversational Memory & History
+* Built-in Internet Search
 * Intelligent Routing
 * Business Alignment
 * Extensible Tooling
@@ -59,6 +60,7 @@ Build your AI business in three lines of code!
 * Designed for a multi-agent swarm 
 * Seamless text and audio streaming with real-time multi-modal processing
 * Persistent memory that preserves context across all agent interactions
+* Quick built-in Internet Search to answer your queries
 * Streamlined message history for all agent interactions
 * Intelligent query routing to agents with optimal domain expertise or your own custom routing
 * Unified value system ensuring brand-aligned agent responses
@@ -250,42 +252,6 @@ Plugins like Solana Agent Kit (sakit) integrate automatically with Solana Agent.
 
 `pip install sakit`
 
-### Internet Search
-```python
-from solana_agent import SolanaAgent
-
-config = {
-    "openai": {
-        "api_key": "your-openai-api-key",
-    },
-    "tools": {
-        "search_internet": {
-            "api_key": "your-perplexity-key", # Required
-            "citations": True, # Optional, defaults to True
-            "model": "sonar"  # Optional, defaults to "sonar"
-        },
-    },
-    "agents": [
-        {
-            "name": "research_specialist",
-            "instructions": "You are an expert researcher who synthesizes complex information clearly.",
-            "specialization": "Research and knowledge synthesis",
-            "tools": ["search_internet"],
-        },
-        {
-            "name": "customer_support",
-            "instructions": "You provide friendly, helpful customer support responses.",
-            "specialization": "Customer inquiries",
-        }
-    ],
-}
-
-solana_agent = SolanaAgent(config=config)
-
-async for response in solana_agent.process("user123", "What are the latest AI developments?"):
-    print(response, end="")
-```
-
 ### MCP
 ```python
 from solana_agent import SolanaAgent
@@ -402,7 +368,7 @@ solana_agent = SolanaAgent(config=config)
 
 test_tool = TestTool()
 
-solana_agent.register_tool(test_tool)
+solana_agent.register_tool("customer_support", test_tool)
 
 async for response in solana_agent.process("user123", "What are the latest AI developments?"):
     print(response, end="")
@@ -410,6 +376,8 @@ async for response in solana_agent.process("user123", "What are the latest AI de
 
 ### Custom Prompt Injection at Runtime
 
+Useful for Knowledge Base answers and FAQs
+
 ```python
 from solana_agent import SolanaAgent
 

{solana_agent-20.1.5 → solana_agent-21.1.0}/README.md

@@ -19,6 +19,7 @@ Build your AI business in three lines of code!
 * Multi-Agent Swarm
 * Multi-Modal Streaming
 * Conversational Memory & History
+* Built-in Internet Search
 * Intelligent Routing
 * Business Alignment
 * Extensible Tooling
@@ -34,6 +35,7 @@ Build your AI business in three lines of code!
 * Designed for a multi-agent swarm 
 * Seamless text and audio streaming with real-time multi-modal processing
 * Persistent memory that preserves context across all agent interactions
+* Quick built-in Internet Search to answer your queries
 * Streamlined message history for all agent interactions
 * Intelligent query routing to agents with optimal domain expertise or your own custom routing
 * Unified value system ensuring brand-aligned agent responses
@@ -225,42 +227,6 @@ Plugins like Solana Agent Kit (sakit) integrate automatically with Solana Agent.
 
 `pip install sakit`
 
-### Internet Search
-```python
-from solana_agent import SolanaAgent
-
-config = {
-    "openai": {
-        "api_key": "your-openai-api-key",
-    },
-    "tools": {
-        "search_internet": {
-            "api_key": "your-perplexity-key", # Required
-            "citations": True, # Optional, defaults to True
-            "model": "sonar"  # Optional, defaults to "sonar"
-        },
-    },
-    "agents": [
-        {
-            "name": "research_specialist",
-            "instructions": "You are an expert researcher who synthesizes complex information clearly.",
-            "specialization": "Research and knowledge synthesis",
-            "tools": ["search_internet"],
-        },
-        {
-            "name": "customer_support",
-            "instructions": "You provide friendly, helpful customer support responses.",
-            "specialization": "Customer inquiries",
-        }
-    ],
-}
-
-solana_agent = SolanaAgent(config=config)
-
-async for response in solana_agent.process("user123", "What are the latest AI developments?"):
-    print(response, end="")
-```
-
 ### MCP
 ```python
 from solana_agent import SolanaAgent
@@ -377,7 +343,7 @@ solana_agent = SolanaAgent(config=config)
 
 test_tool = TestTool()
 
-solana_agent.register_tool(test_tool)
+solana_agent.register_tool("customer_support", test_tool)
 
 async for response in solana_agent.process("user123", "What are the latest AI developments?"):
     print(response, end="")
@@ -385,6 +351,8 @@ async for response in solana_agent.process("user123", "What are the latest AI de
 
 ### Custom Prompt Injection at Runtime
 
+Useful for Knowledge Base answers and FAQs
+
 ```python
 from solana_agent import SolanaAgent
 

{solana_agent-20.1.5 → solana_agent-21.1.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "solana-agent"
-version = "20.1.5"
+version = "21.1.0"
 description = "Agentic IQ"
 authors = ["Bevan Hunt <bevan@bevanhunt.com>"]
 license = "MIT"
@@ -23,15 +23,15 @@ python_paths = [".", "tests"]
 
 [tool.poetry.dependencies]
 python = ">=3.12,<4.0"
-openai = "^1.68.2"
-pydantic = "^2.11.1"
+openai = "^1.70.0"
+pydantic = "^2.11.2"
 pymongo = "^4.11.3"
-zep-cloud = "^2.8.0"
+zep-cloud = "^2.9.0"
 zep-python = "^2.0.2"
 
 [tool.poetry.group.dev.dependencies]
 pytest = "^8.3.5"
-pytest-cov = "^6.0.0"
+pytest-cov = "^6.1.0"
 pytest-asyncio = "^0.26.0"
 pytest-mock = "^3.14.0"
 pytest-github-actions-annotate-failures = "^0.3.0"

{solana_agent-20.1.5 → solana_agent-21.1.0}/solana_agent/adapters/llm_adapter.py

@@ -20,6 +20,7 @@ class OpenAIAdapter(LLMProvider):
         self.client = OpenAI(api_key=api_key)
         self.parse_model = "gpt-4o-mini"
         self.text_model = "gpt-4o-mini"
+        self.search_model = "gpt-4o-mini-search-preview"
         self.transcription_model = "gpt-4o-mini-transcribe"
         self.tts_model = "tts-1"
 
@@ -105,6 +106,7 @@ class OpenAIAdapter(LLMProvider):
         self,
         prompt: str,
         system_prompt: str = "",
+        search: bool = False,
     ) -> AsyncGenerator[str, None]:  # pragma: no cover
         """Generate text from OpenAI models."""
         messages = []
@@ -114,11 +116,15 @@ class OpenAIAdapter(LLMProvider):
 
         messages.append({"role": "user", "content": prompt})
 
+        model = self.text_model
+        if search:
+            model = self.search_model
+
         # Prepare request parameters
         request_params = {
             "messages": messages,
             "stream": True,
-            "model": self.text_model,
+            "model": model,
         }
         try:
             response = self.client.chat.completions.create(**request_params)

{solana_agent-20.1.5 → solana_agent-21.1.0}/solana_agent/client/solana_agent.py

@@ -56,6 +56,7 @@ class SolanaAgent(SolanaAgentInterface):
             "flac", "mp3", "mp4", "mpeg", "mpga", "m4a", "ogg", "wav", "webm"
         ] = "mp4",
         router: Optional[RoutingInterface] = None,
+        use_openai_search: bool = True,
     ) -> AsyncGenerator[Union[str, bytes], None]:  # pragma: no cover
         """Process a user message and return the response stream.
 
@@ -69,6 +70,7 @@ class SolanaAgent(SolanaAgentInterface):
             audio_output_format: Audio output format
             audio_input_format: Audio input format
             router: Optional routing service for processing
+            use_openai_search: Flag to use OpenAI search
 
         Returns:
             Async generator yielding response chunks (text strings or audio bytes)
@@ -83,6 +85,7 @@ class SolanaAgent(SolanaAgentInterface):
             audio_input_format=audio_input_format,
             prompt=prompt,
             router=router,
+            use_openai_search=use_openai_search,
         ):
             yield chunk
 
@@ -118,11 +121,12 @@ class SolanaAgent(SolanaAgentInterface):
             user_id, page_num, page_size, sort_order
         )
 
-    def register_tool(self, tool: Tool) -> bool:
+    def register_tool(self, agent_name: str, tool: Tool) -> bool:
         """
         Register a tool with the agent system.
 
         Args:
+            agent_name: Name of the agent to register the tool with
             tool: Tool instance to register
 
         Returns:
@@ -131,8 +135,6 @@ class SolanaAgent(SolanaAgentInterface):
         success = self.query_service.agent_service.tool_registry.register_tool(
             tool)
         if success:
-            agents = self.query_service.agent_service.get_all_ai_agents()
-            for agent_name in agents:
-                self.query_service.agent_service.assign_tool_for_agent(
-                    agent_name, tool.name)
+            self.query_service.agent_service.assign_tool_for_agent(
+                agent_name, tool.name)
         return success

{solana_agent-20.1.5 → solana_agent-21.1.0}/solana_agent/interfaces/client/client.py

@@ -2,6 +2,7 @@ from abc import ABC, abstractmethod
 from typing import Any, AsyncGenerator, Dict, Literal, Optional, Union
 
 from solana_agent.interfaces.plugins.plugins import Tool
+from solana_agent.interfaces.services.routing import RoutingService as RoutingInterface
 
 
 class SolanaAgent(ABC):
@@ -12,6 +13,7 @@ class SolanaAgent(ABC):
         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",
@@ -21,7 +23,8 @@ class SolanaAgent(ABC):
         audio_input_format: Literal[
             "flac", "mp3", "mp4", "mpeg", "mpga", "m4a", "ogg", "wav", "webm"
         ] = "mp4",
-        prompt: Optional[str] = None,
+        router: Optional[RoutingInterface] = None,
+        use_openai_search: bool = True,
     ) -> AsyncGenerator[Union[str, bytes], None]:
         """Process a user message and return the response stream."""
         pass
@@ -38,6 +41,6 @@ class SolanaAgent(ABC):
         pass
 
     @abstractmethod
-    def register_tool(self, tool: Tool) -> bool:
+    def register_tool(self, agent_name: str, tool: Tool) -> bool:
         """Register a tool in the registry."""
         pass

{solana_agent-20.1.5 → solana_agent-21.1.0}/solana_agent/interfaces/providers/llm.py

@@ -15,6 +15,7 @@ class LLMProvider(ABC):
         self,
         prompt: str,
         system_prompt: str = "",
+        search: bool = False,
     ) -> AsyncGenerator[str, None]:
         """Generate text from the language model."""
         pass

{solana_agent-20.1.5 → solana_agent-21.1.0}/solana_agent/interfaces/services/agent.py

@@ -34,6 +34,7 @@ class AgentService(ABC):
             "flac", "mp3", "mp4", "mpeg", "mpga", "m4a", "ogg", "wav", "webm"
         ] = "mp4",
         prompt: Optional[str] = None,
+        use_openai_search: bool = True,
     ) -> AsyncGenerator[Union[str, bytes], None]:
         """Generate a response from an agent."""
         pass

{solana_agent-20.1.5 → solana_agent-21.1.0}/solana_agent/interfaces/services/query.py

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

{solana_agent-20.1.5 → solana_agent-21.1.0}/solana_agent/services/agent.py

@@ -176,6 +176,7 @@ class AgentService(AgentServiceInterface):
             "flac", "mp3", "mp4", "mpeg", "mpga", "m4a", "ogg", "wav", "webm"
         ] = "mp4",
         prompt: Optional[str] = None,
+        use_openai_search: bool = True,
     ) -> AsyncGenerator[Union[str, bytes], None]:  # pragma: no cover
         """Generate a response with support for text/audio input/output.
 
@@ -190,6 +191,7 @@ class AgentService(AgentServiceInterface):
             audio_output_format: Audio output format
             audio_input_format: Audio input format
             prompt: Optional prompt for the agent
+            use_openai_search: Flag to use OpenAI search
 
         Yields:
             Text chunks or audio bytes depending on output_format
@@ -237,6 +239,7 @@ class AgentService(AgentServiceInterface):
             async for chunk in self.llm_provider.generate_text(
                 prompt=query_text,
                 system_prompt=system_prompt,
+                search=use_openai_search,
             ):
                 # Check for JSON start
                 if chunk.strip().startswith("{") and not is_json:

{solana_agent-20.1.5 → solana_agent-21.1.0}/solana_agent/services/query.py

@@ -49,6 +49,7 @@ class QueryService(QueryServiceInterface):
         ] = "mp4",
         prompt: Optional[str] = None,
         router: Optional[RoutingServiceInterface] = None,
+        use_openai_search: bool = True,
     ) -> AsyncGenerator[Union[str, bytes], None]:  # pragma: no cover
         """Process the user request with appropriate agent.
 
@@ -62,6 +63,7 @@ class QueryService(QueryServiceInterface):
             audio_input_format: Audio input format
             prompt: Optional prompt for the agent
             router: Optional routing service for processing
+            use_openai_search: Flag to use OpenAI search
 
         Yields:
             Response chunks (text strings or audio bytes)
@@ -119,6 +121,7 @@ class QueryService(QueryServiceInterface):
                     audio_output_format=audio_output_format,
                     audio_instructions=audio_instructions,
                     prompt=prompt,
+                    use_openai_search=use_openai_search,
                 ):
                     yield audio_chunk
 
@@ -137,6 +140,7 @@ class QueryService(QueryServiceInterface):
                     memory_context=memory_context,
                     output_format="text",
                     prompt=prompt,
+                    use_openai_search=use_openai_search,
                 ):
                     yield chunk
                     full_text_response += chunk