solana-agent 27.3.5__py3-none-any.whl → 27.3.7__py3-none-any.whl

solana_agent/interfaces/services/agent.py

@@ -8,7 +8,9 @@ class AgentService(ABC):
     """Interface for agent management and response generation."""
 
     @abstractmethod
-    def register_ai_agent(self, name: str, instructions: str, specialization: str) -> None:
+    def register_ai_agent(
+        self, name: str, instructions: str, specialization: str
+    ) -> None:
         """Register an AI agent with its specialization."""
         pass
 
@@ -25,11 +27,22 @@ class AgentService(ABC):
         query: Union[str, bytes],
         memory_context: str = "",
         output_format: Literal["text", "audio"] = "text",
-        audio_voice: Literal["alloy", "ash", "ballad", "coral", "echo",
-                             "fable", "onyx", "nova", "sage", "shimmer"] = "nova",
+        audio_voice: Literal[
+            "alloy",
+            "ash",
+            "ballad",
+            "coral",
+            "echo",
+            "fable",
+            "onyx",
+            "nova",
+            "sage",
+            "shimmer",
+        ] = "nova",
         audio_instructions: str = "You speak in a friendly and helpful manner.",
-        audio_output_format: Literal['mp3', 'opus',
-                                     'aac', 'flac', 'wav', 'pcm'] = "aac",
+        audio_output_format: Literal[
+            "mp3", "opus", "aac", "flac", "wav", "pcm"
+        ] = "aac",
         prompt: Optional[str] = None,
     ) -> AsyncGenerator[Union[str, bytes], None]:
         """Generate a response from an agent."""
@@ -46,7 +59,9 @@ class AgentService(ABC):
         pass
 
     @abstractmethod
-    async def execute_tool(self, agent_name: str, tool_name: str, parameters: Dict[str, Any]) -> Dict[str, Any]:
+    async def execute_tool(
+        self, agent_name: str, tool_name: str, parameters: Dict[str, Any]
+    ) -> Dict[str, Any]:
         """Execute a tool on behalf of an agent."""
         pass
 

solana_agent/interfaces/services/knowledge_base.py

@@ -13,7 +13,7 @@ class KnowledgeBaseService(ABC):
         text: str,
         metadata: Dict[str, Any],
         document_id: Optional[str] = None,
-        namespace: Optional[str] = None
+        namespace: Optional[str] = None,
     ) -> str:
         """
         Add a document to the knowledge base.
@@ -28,7 +28,7 @@ class KnowledgeBaseService(ABC):
         top_k: int = 5,
         namespace: Optional[str] = None,
         include_content: bool = True,
-        include_metadata: bool = True
+        include_metadata: bool = True,
     ) -> List[Dict[str, Any]]:
         """
         Query the knowledge base with semantic search.
@@ -37,9 +37,7 @@ class KnowledgeBaseService(ABC):
 
     @abstractmethod
     async def delete_document(
-        self,
-        document_id: str,
-        namespace: Optional[str] = None
+        self, document_id: str, namespace: Optional[str] = None
     ) -> bool:
         """
         Delete a document from the knowledge base.
@@ -52,7 +50,7 @@ class KnowledgeBaseService(ABC):
         document_id: str,
         text: Optional[str] = None,
         metadata: Optional[Dict[str, Any]] = None,
-        namespace: Optional[str] = None
+        namespace: Optional[str] = None,
     ) -> bool:
         """
         Update an existing document in the knowledge base.
@@ -64,7 +62,7 @@ class KnowledgeBaseService(ABC):
         self,
         documents: List[Dict[str, Any]],
         namespace: Optional[str] = None,
-        batch_size: int = 50
+        batch_size: int = 50,
     ) -> List[str]:
         """
         Add multiple documents in batches.
@@ -78,7 +76,7 @@ class KnowledgeBaseService(ABC):
         metadata: Dict[str, Any],
         document_id: Optional[str] = None,
         namespace: Optional[str] = None,
-        chunk_batch_size: int = 50
+        chunk_batch_size: int = 50,
     ) -> str:
         """
         Add a PDF document to the knowledge base.

solana_agent/interfaces/services/query.py

@@ -13,11 +13,22 @@ class QueryService(ABC):
         user_id: str,
         query: Union[str, bytes],
         output_format: Literal["text", "audio"] = "text",
-        audio_voice: Literal["alloy", "ash", "ballad", "coral", "echo",
-                             "fable", "onyx", "nova", "sage", "shimmer"] = "nova",
+        audio_voice: Literal[
+            "alloy",
+            "ash",
+            "ballad",
+            "coral",
+            "echo",
+            "fable",
+            "onyx",
+            "nova",
+            "sage",
+            "shimmer",
+        ] = "nova",
         audio_instructions: str = "You speak in a friendly and helpful manner.",
-        audio_output_format: Literal['mp3', 'opus',
-                                     'aac', 'flac', 'wav', 'pcm'] = "aac",
+        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",
@@ -33,7 +44,7 @@ class QueryService(ABC):
         user_id: str,
         page_num: int = 1,
         page_size: int = 20,
-        sort_order: str = "desc"  # "asc" for oldest-first, "desc" for newest-first
+        sort_order: str = "desc",  # "asc" for oldest-first, "desc" for newest-first
     ) -> Dict[str, Any]:
         """Get paginated message history for a user."""
         pass

solana_agent/interfaces/services/routing.py

@@ -1,5 +1,4 @@
 from abc import ABC, abstractmethod
-from typing import Any, Tuple
 
 
 class RoutingService(ABC):

solana_agent/plugins/manager.py

@@ -4,11 +4,14 @@ Plugin manager for the Solana Agent system.
 This module implements the concrete PluginManager that discovers,
 loads, and manages plugins.
 """
+
 import importlib
 from typing import Dict, List, Any, Optional
 import importlib.metadata
 
-from solana_agent.interfaces.plugins.plugins import PluginManager as PluginManagerInterface
+from solana_agent.interfaces.plugins.plugins import (
+    PluginManager as PluginManagerInterface,
+)
 from solana_agent.interfaces.plugins.plugins import Plugin
 from solana_agent.plugins.registry import ToolRegistry
 
@@ -19,7 +22,11 @@ class PluginManager(PluginManagerInterface):
     # Class variable to track loaded entry points
     _loaded_entry_points = set()
 
-    def __init__(self, config: Optional[Dict[str, Any]] = None, tool_registry: Optional[ToolRegistry] = None):
+    def __init__(
+        self,
+        config: Optional[Dict[str, Any]] = None,
+        tool_registry: Optional[ToolRegistry] = None,
+    ):
         """Initialize with optional configuration and tool registry."""
         self.config = config or {}
         self.tool_registry = tool_registry or ToolRegistry()
@@ -61,12 +68,13 @@ class PluginManager(PluginManagerInterface):
         loaded_plugins = []
 
         # Discover plugins through entry points
-        for entry_point in importlib.metadata.entry_points(group='solana_agent.plugins'):
+        for entry_point in importlib.metadata.entry_points(
+            group="solana_agent.plugins"
+        ):
             # Skip if this entry point has already been loaded
             entry_point_id = f"{entry_point.name}:{entry_point.value}"
             if entry_point_id in PluginManager._loaded_entry_points:
-                print(
-                    f"Skipping already loaded plugin: {entry_point.name}")
+                print(f"Skipping already loaded plugin: {entry_point.name}")
                 continue
 
             try:
@@ -103,10 +111,7 @@ class PluginManager(PluginManagerInterface):
             List of plugin details dictionaries
         """
         return [
-            {
-                "name": plugin.name,
-                "description": plugin.description
-            }
+            {"name": plugin.name, "description": plugin.description}
             for plugin in self._plugins.values()
         ]
 

solana_agent/plugins/registry.py

@@ -1,12 +1,15 @@
 """
 Tool registry for the Solana Agent system.
 
-This module implements the concrete ToolRegistry that manages tools 
+This module implements the concrete ToolRegistry that manages tools
 and their access permissions.
 """
+
 from typing import Dict, List, Any, Optional
 
-from solana_agent.interfaces.plugins.plugins import ToolRegistry as ToolRegistryInterface
+from solana_agent.interfaces.plugins.plugins import (
+    ToolRegistry as ToolRegistryInterface,
+)
 from solana_agent.interfaces.plugins.plugins import Tool
 
 
@@ -39,7 +42,8 @@ class ToolRegistry(ToolRegistryInterface):
         """Give an agent access to a specific tool."""
         if tool_name not in self._tools:
             print(
-                f"Error: Tool {tool_name} is not registered. Available tools: {list(self._tools.keys())}")
+                f"Error: Tool {tool_name} is not registered. Available tools: {list(self._tools.keys())}"
+            )
             return False
 
         # Initialize agent's tool list if not exists
@@ -47,12 +51,10 @@ class ToolRegistry(ToolRegistryInterface):
             self._agent_tools[agent_name] = [tool_name]
         elif tool_name not in self._agent_tools[agent_name]:
             # Add new tool to existing list
-            self._agent_tools[agent_name] = [
-                *self._agent_tools[agent_name], tool_name]
+            self._agent_tools[agent_name] = [*self._agent_tools[agent_name], tool_name]
 
         print(f"Successfully assigned tool {tool_name} to agent {agent_name}")
-        print(
-            f"Agent {agent_name} now has access to: {self._agent_tools[agent_name]}")
+        print(f"Agent {agent_name} now has access to: {self._agent_tools[agent_name]}")
 
         return True
 
@@ -63,12 +65,12 @@ class ToolRegistry(ToolRegistryInterface):
             {
                 "name": name,
                 "description": self._tools[name].description,
-                "parameters": self._tools[name].get_schema()
+                "parameters": self._tools[name].get_schema(),
             }
-            for name in tool_names if name in self._tools
+            for name in tool_names
+            if name in self._tools
         ]
-        print(
-            f"Tools available to agent {agent_name}: {[t['name'] for t in tools]}")
+        print(f"Tools available to agent {agent_name}: {[t['name'] for t in tools]}")
         return tools
 
     def list_all_tools(self) -> List[str]:

solana_agent/plugins/tools/__init__.py

@@ -3,8 +3,3 @@ Tools for the Solana Agent system.
 
 This package contains the base AutoTool class and built-in tool implementations.
 """
-
-from solana_agent.plugins.tools.auto_tool import *
-
-# Version of the domain model
-__version__ = '14.0.0'

solana_agent/plugins/tools/auto_tool.py

@@ -4,6 +4,7 @@ AutoTool implementation for the Solana Agent system.
 This module provides the base AutoTool class that implements the Tool interface
 and can be extended to create custom tools.
 """
+
 from typing import Dict, Any
 
 from solana_agent.interfaces.plugins.plugins import Tool

solana_agent/repositories/memory.py

@@ -43,13 +43,18 @@ class MemoryRepository(MemoryProvider):
             raise ValueError("User ID cannot be None or empty")
         if not messages or not isinstance(messages, list):
             raise ValueError("Messages must be a non-empty list")
-        if not all(isinstance(msg, dict) and "role" in msg and "content" in msg for msg in messages):
+        if not all(
+            isinstance(msg, dict) and "role" in msg and "content" in msg
+            for msg in messages
+        ):
             raise ValueError(
-                "All messages must be dictionaries with 'role' and 'content' keys")
+                "All messages must be dictionaries with 'role' and 'content' keys"
+            )
         for msg in messages:
             if msg["role"] not in ["user", "assistant"]:
                 raise ValueError(
-                    f"Invalid role '{msg['role']}' in message. Only 'user' and 'assistant' roles are accepted.")
+                    f"Invalid role '{msg['role']}' in message. Only 'user' and 'assistant' roles are accepted."
+                )
 
         # Store in MongoDB
         if self.mongo and len(messages) >= 2:
@@ -71,7 +76,7 @@ class MemoryRepository(MemoryProvider):
                         "user_id": user_id,
                         "user_message": user_msg,
                         "assistant_message": assistant_msg,
-                        "timestamp": datetime.now(timezone.utc)
+                        "timestamp": datetime.now(timezone.utc),
                     }
                     self.mongo.insert_one(self.collection, doc)
             except Exception as e:
@@ -96,11 +101,8 @@ class MemoryRepository(MemoryProvider):
         # Add messages to Zep memory
         if zep_messages:
             try:
-                await self.zep.memory.add(
-                    session_id=user_id,
-                    messages=zep_messages
-                )
-            except Exception as e:
+                await self.zep.memory.add(session_id=user_id, messages=zep_messages)
+            except Exception:
                 try:
                     try:
                         await self.zep.user.add(user_id=user_id)
@@ -108,13 +110,12 @@ class MemoryRepository(MemoryProvider):
                         print(f"Zep user addition error: {e}")
 
                     try:
-                        await self.zep.memory.add_session(session_id=user_id, user_id=user_id)
+                        await self.zep.memory.add_session(
+                            session_id=user_id, user_id=user_id
+                        )
                     except Exception as e:
                         print(f"Zep session creation error: {e}")
-                    await self.zep.memory.add(
-                        session_id=user_id,
-                        messages=zep_messages
-                    )
+                    await self.zep.memory.add(session_id=user_id, messages=zep_messages)
                 except Exception as e:
                     print(f"Zep memory addition error: {e}")
                     return
@@ -137,10 +138,7 @@ class MemoryRepository(MemoryProvider):
         """Delete memory from both systems."""
         if self.mongo:
             try:
-                self.mongo.delete_all(
-                    self.collection,
-                    {"user_id": user_id}
-                )
+                self.mongo.delete_all(self.collection, {"user_id": user_id})
             except Exception as e:
                 print(f"MongoDB deletion error: {e}")
 
@@ -163,7 +161,7 @@ class MemoryRepository(MemoryProvider):
         query: Dict,
         sort: Optional[List[Tuple]] = None,
         limit: int = 0,
-        skip: int = 0
+        skip: int = 0,
     ) -> List[Dict]:  # pragma: no cover
         """Find documents in MongoDB."""
         if not self.mongo:
@@ -193,9 +191,9 @@ class MemoryRepository(MemoryProvider):
             return text
 
         # Try to truncate at last period before limit
-        last_period = text.rfind('.', 0, limit)
+        last_period = text.rfind(".", 0, limit)
         if last_period > 0:
-            return text[:last_period + 1]
+            return text[: last_period + 1]
 
         # If no period found, truncate at limit and add ellipsis
-        return text[:limit-3] + "..."
+        return text[: limit - 3] + "..."

solana_agent/services/__init__.py

@@ -1,6 +1,6 @@
 """
 Service implementations for the Solana Agent system.
 
-These services implement the business logic interfaces defined in 
+These services implement the business logic interfaces defined in
 solana_agent.interfaces.services.
 """