solana-agent 19.1.0__py3-none-any.whl → 20.0.0__py3-none-any.whl

solana_agent/client/solana_agent.py

@@ -97,7 +97,7 @@ class SolanaAgent(SolanaAgentInterface):
         page_num: int = 1,
         page_size: int = 20,
         sort_order: str = "desc"  # "asc" for oldest-first, "desc" for newest-first
-    ) -> Dict[str, Any]:
+    ) -> Dict[str, Any]:  # pragma: no cover
         """
         Get paginated message history for a user.
 
@@ -124,22 +124,11 @@ class SolanaAgent(SolanaAgentInterface):
         Returns:
             True if successful, False
         """
-
-        try:
-            print(f"Attempting to register tool: {tool.name}")
-            success = self.query_service.agent_service.tool_registry.register_tool(
-                tool)
-            if success:
-                print(f"Tool {tool.name} registered successfully")
-                # Get all agents and assign the tool to them
-                agents = self.query_service.agent_service.get_all_ai_agents()
-                for agent_name in agents:
-                    print(f"Assigning {tool.name} to agent {agent_name}")
-                    self.query_service.agent_service.assign_tool_for_agent(
-                        agent_name, tool.name)
-            return success
-        except Exception as e:
-            print(f"Error in register_tool: {str(e)}")
-            import traceback
-            print(traceback.format_exc())
-            return False
+        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)
+        return success

solana_agent/factories/agent_factory.py

@@ -109,8 +109,12 @@ class SolanaAgentFactory:
             config=config,
             tool_registry=agent_service.tool_registry
         )
-        loaded_plugins = agent_service.plugin_manager.load_plugins()
-        print(f"Loaded {loaded_plugins} plugins")
+        try:
+            loaded_plugins = agent_service.plugin_manager.load_plugins()
+            print(f"Loaded {loaded_plugins} plugins")
+        except Exception as e:
+            print(f"Error loading plugins: {e}")
+            loaded_plugins = 0
 
         # Register predefined agents
         for agent_config in config.get("agents", []):
@@ -125,25 +129,18 @@ class SolanaAgentFactory:
                 for tool_name in agent_config["tools"]:
                     print(
                         f"Available tools before registering {tool_name}: {agent_service.tool_registry.list_all_tools()}")
-                    try:
-                        agent_service.assign_tool_for_agent(
-                            agent_config["name"], tool_name
-                        )
-                        print(
-                            f"Successfully registered {tool_name} for agent {agent_config['name']}")
-                    except ValueError as e:
-                        print(
-                            f"Error registering tool {tool_name} for agent {agent_config['name']}: {e}")
+                    agent_service.assign_tool_for_agent(
+                        agent_config["name"], tool_name
+                    )
+                    print(
+                        f"Successfully registered {tool_name} for agent {agent_config['name']}")
 
         # Global tool registrations
         if "agent_tools" in config:
             for agent_name, tools in config["agent_tools"].items():
                 for tool_name in tools:
-                    try:
-                        agent_service.assign_tool_for_agent(
-                            agent_name, tool_name)
-                    except ValueError as e:
-                        print(f"Error registering tool: {e}")
+                    agent_service.assign_tool_for_agent(
+                        agent_name, tool_name)
 
         # Create and return the query service
         query_service = QueryService(

solana_agent/plugins/manager.py

@@ -35,19 +35,21 @@ class PluginManager(PluginManagerInterface):
             True if registration succeeded, False otherwise
         """
         try:
-            # Store plugin by name
-            self._plugins[plugin.name] = plugin
-
-            # Initialize the plugin with the tool registry
+            # Initialize the plugin with the tool registry first
             plugin.initialize(self.tool_registry)
 
-            # *** ADD THIS LINE: Configure the plugin with our config ***
-            print(f"Configuring plugin {plugin.name} with config")
+            # Then configure the plugin
             plugin.configure(self.config)
 
+            # Only store plugin if both initialize and configure succeed
+            self._plugins[plugin.name] = plugin
+            print(f"Successfully registered plugin {plugin.name}")
             return True
+
         except Exception as e:
             print(f"Error registering plugin {plugin.name}: {e}")
+            # Remove plugin from registry if it was added
+            self._plugins.pop(plugin.name, None)
             return False
 
     def load_plugins(self) -> List[str]:
@@ -59,30 +61,27 @@ class PluginManager(PluginManagerInterface):
         loaded_plugins = []
 
         # Discover plugins through entry points
-        try:
-            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}")
-                    continue
-
-                try:
-                    print(f"Found plugin entry point: {entry_point.name}")
-                    PluginManager._loaded_entry_points.add(entry_point_id)
-                    plugin_factory = entry_point.load()
-                    plugin = plugin_factory()
-
-                    # Register the plugin
-                    if self.register_plugin(plugin):
-                        # Use entry_point.name instead of plugin.name
-                        loaded_plugins.append(entry_point.name)
-
-                except Exception as e:
-                    print(f"Error loading plugin {entry_point.name}: {e}")
-        except Exception as e:
-            print(f"Error discovering plugins: {e}")
+        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}")
+                continue
+
+            try:
+                print(f"Found plugin entry point: {entry_point.name}")
+                PluginManager._loaded_entry_points.add(entry_point_id)
+                plugin_factory = entry_point.load()
+                plugin = plugin_factory()
+
+                # Register the plugin
+                if self.register_plugin(plugin):
+                    # Use entry_point.name instead of plugin.name
+                    loaded_plugins.append(entry_point.name)
+
+            except Exception as e:
+                print(f"Error loading plugin {entry_point.name}: {e}")
 
         return loaded_plugins
 

solana_agent/plugins/registry.py

@@ -76,6 +76,23 @@ class ToolRegistry(ToolRegistryInterface):
         return list(self._tools.keys())
 
     def configure_all_tools(self, config: Dict[str, Any]) -> None:
-        """Configure all registered tools with the same config."""
-        for tool in self._tools.values():
-            tool.configure(config)
+        """Configure all registered tools with new configuration.
+
+        Args:
+            config: Configuration dictionary to apply
+        """
+        self._config.update(config)
+        configure_errors = []
+
+        for name, tool in self._tools.items():
+            try:
+                print(f"Configuring tool: {name}")
+                tool.configure(self._config)
+            except Exception as e:
+                print(f"Error configuring tool {name}: {e}")
+                configure_errors.append((name, str(e)))
+
+        if configure_errors:
+            print("The following tools failed to configure:")
+            for name, error in configure_errors:
+                print(f"- {name}: {error}")

solana_agent/plugins/tools/auto_tool.py

@@ -34,6 +34,8 @@ class AutoTool(Tool):
 
     def configure(self, config: Dict[str, Any]) -> None:
         """Configure the tool with settings from config."""
+        if config is None:
+            raise TypeError("Config cannot be None")
         self._config = config
 
     def get_schema(self) -> Dict[str, Any]:

solana_agent/repositories/memory.py

@@ -25,10 +25,13 @@ class MemoryRepository(MemoryProvider):
             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)])
+            try:
+                # 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)])
+            except Exception as e:
+                print(f"Error initializing MongoDB: {e}")
 
         # Initialize Zep
         if zep_api_key and not zep_base_url:
@@ -40,53 +43,78 @@ 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
-        if self.mongo:
+        if not user_id:
+            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):
+            raise ValueError(
+                "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.")
+
+        # Store in MongoDB
+        if self.mongo and len(messages) >= 2:
             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)
+                # Get last user and assistant messages
+                user_msg = None
+                assistant_msg = None
+                for msg in reversed(messages):
+                    if msg.get("role") == "user" and not user_msg:
+                        user_msg = msg.get("content")
+                    elif msg.get("role") == "assistant" and not assistant_msg:
+                        assistant_msg = msg.get("content")
+                    if user_msg and assistant_msg:
+                        break
+
+                if user_msg and assistant_msg:
+                    # Store truncated messages
+                    doc = {
+                        "user_id": user_id,
+                        "user_message": self._truncate(user_msg),
+                        "assistant_message": self._truncate(assistant_msg),
+                        "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
+        # Store in Zep
         if not self.zep:
             return
 
         try:
-            try:
-                await self.zep.user.add(user_id=user_id)
-            except Exception:
-                pass
-            try:
-                await self.zep.memory.add_session(
-                    session_id=user_id,
-                    user_id=user_id,
-                )
-            except Exception:
-                pass
+            await self.zep.user.add(user_id=user_id)
+        except Exception as e:
+            print(f"Zep user addition error: {e}")
+
+        try:
+            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}")
 
-            zep_messages = [
-                Message(
+        # Convert messages to Zep format
+        zep_messages = []
+        for msg in messages:
+            if "role" in msg and "content" in msg:
+                zep_msg = Message(
                     role=msg["role"],
+                    content=msg["content"],
                     role_type=msg["role"],
-                    content=self._truncate(msg["content"])
                 )
-                for msg in messages
-            ]
-            await self.zep.memory.add(session_id=user_id, messages=zep_messages)
-        except Exception as e:
-            print(f"Zep storage error: {e}")
+                zep_messages.append(zep_msg)
+
+        # 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:
+                print(f"Zep memory addition error: {e}")
 
     async def retrieve(self, user_id: str) -> str:
         """Retrieve memory context from Zep only."""
@@ -95,7 +123,8 @@ class MemoryRepository(MemoryProvider):
 
         try:
             memory = await self.zep.memory.get(session_id=user_id)
-
+            if memory is None or not hasattr(memory, 'context') or memory.context is None:
+                return ""
             return memory.context
 
         except Exception as e:
@@ -118,9 +147,13 @@ class MemoryRepository(MemoryProvider):
 
         try:
             await self.zep.memory.delete(session_id=user_id)
+        except Exception as e:
+            print(f"Zep memory deletion error: {e}")
+
+        try:
             await self.zep.user.delete(user_id=user_id)
         except Exception as e:
-            print(f"Zep deletion error: {e}")
+            print(f"Zep user deletion error: {e}")
 
     def find(
         self,
@@ -129,26 +162,38 @@ class MemoryRepository(MemoryProvider):
         sort: Optional[List[Tuple]] = None,
         limit: int = 0,
         skip: int = 0
-    ) -> List[Dict]:
-        """Find documents matching query."""
+    ) -> List[Dict]:  # pragma: no cover
+        """Find documents in MongoDB."""
         if not self.mongo:
             return []
-        return self.mongo.find(collection, query, sort=sort, limit=limit, skip=skip)
+
+        try:
+            return self.mongo.find(collection, query, sort=sort, limit=limit, skip=skip)
+        except Exception as e:
+            print(f"MongoDB find error: {e}")
+            return []
 
     def count_documents(self, collection: str, query: Dict) -> int:
-        """Count documents matching query."""
+        """Count documents in MongoDB."""
         if not self.mongo:
             return 0
         return self.mongo.count_documents(collection, query)
 
     def _truncate(self, text: str, limit: int = 2500) -> str:
         """Truncate text to be within limits."""
+        if text is None:
+            raise AttributeError("Cannot truncate None text")
+
+        if not text:
+            return ""
+
         if len(text) <= limit:
             return text
 
-        truncated = text[:limit]
-        last_period = truncated.rfind(".")
-        if last_period > limit * 0.8:
-            return truncated[:last_period + 1]
+        # Try to truncate at last period before limit
+        last_period = text.rfind('.', 0, limit)
+        if last_period > 0:
+            return text[:last_period + 1]
 
-        return truncated + "..."
+        # If no period found, truncate at limit and add ellipsis
+        return text[:limit] + "..."

solana_agent/services/agent.py

@@ -13,6 +13,7 @@ from typing import AsyncGenerator, Dict, List, Literal, Optional, Any, Union
 from solana_agent.interfaces.services.agent import AgentService as AgentServiceInterface
 from solana_agent.interfaces.providers.llm import LLMProvider
 from solana_agent.interfaces.plugins.plugins import ToolRegistry as ToolRegistryInterface
+from solana_agent.plugins.manager import PluginManager
 from solana_agent.plugins.registry import ToolRegistry
 from solana_agent.domains.agent import AIAgent, BusinessMission
 
@@ -40,8 +41,10 @@ class AgentService(AgentServiceInterface):
         self.tool_registry = ToolRegistry(config=self.config)
         self.agents: List[AIAgent] = []
 
-        # Will be set by factory if plugin system is enabled
-        self.plugin_manager = None
+        self.plugin_manager = PluginManager(
+            config=self.config,
+            tool_registry=self.tool_registry,
+        )
 
     def register_ai_agent(
         self, name: str, instructions: str, specialization: str,

solana_agent/services/routing.py

@@ -95,7 +95,7 @@ class RoutingService(RoutingServiceInterface):
                 "confidence": 0.0
             }
 
-    async def route_query(self, query: str) -> str:
+    async def route_query(self, query: str) -> str:  # pragma: no cover
         """Route a query to the appropriate agent.
 
         Args:

{solana_agent-19.1.0.dist-info → solana_agent-20.0.0.dist-info}/METADATA

@@ -1,23 +1,25 @@
 Metadata-Version: 2.3
 Name: solana-agent
-Version: 19.1.0
+Version: 20.0.0
 Summary: Agentic IQ
 License: MIT
 Keywords: ai,openai,ai agents,agi
 Author: Bevan Hunt
 Author-email: bevan@bevanhunt.com
 Requires-Python: >=3.12,<4.0
+Classifier: Development Status :: 5 - Production/Stable
+Classifier: Intended Audience :: Developers
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Programming Language :: Python :: 3
 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.69.0,<2.0.0)
-Requires-Dist: pydantic (>=2.11.0,<3.0.0)
+Requires-Dist: openai (>=1.68.2,<2.0.0)
+Requires-Dist: pydantic (>=2.11.1,<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)
+Project-URL: Documentation, https://docs.solana-agent.com
 Project-URL: Repository, https://github.com/truemagic-coder/solana-agent
 Description-Content-Type: text/markdown
 

{solana_agent-19.1.0.dist-info → solana_agent-20.0.0.dist-info}/RECORD

@@ -3,12 +3,12 @@ solana_agent/adapters/__init__.py,sha256=tiEEuuy0NF3ngc_tGEcRTt71zVI58v3dYY9RvMr
 solana_agent/adapters/llm_adapter.py,sha256=3_7whVsPSJdlzBVUBlV7RBRCCo2dMXNmlACIrCoQxQ4,5426
 solana_agent/adapters/mongodb_adapter.py,sha256=qqEFbY_v1XGyFXBmwd5HSXSSHnA9wWo-Hm1vGEyIG0k,2718
 solana_agent/client/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-solana_agent/client/solana_agent.py,sha256=sN8imsCMm2SqsF4c4moumno1V6OPPAABv1IJw9P7QbU,5476
+solana_agent/client/solana_agent.py,sha256=CwUQHA76QiVmp9aNZcH1BVzD7I3LffwkvxtP5yLLmew,5003
 solana_agent/domains/__init__.py,sha256=HiC94wVPRy-QDJSSRywCRrhrFfTBeHjfi5z-QfZv46U,168
 solana_agent/domains/agent.py,sha256=WTo-pEc66V6D_35cpDE-kTsw1SJM-dtylPZ7em5em7Q,2659
 solana_agent/domains/routing.py,sha256=UDlgTjUoC9xIBVYu_dnf9-KG_bBgdEXAv_UtDOrYo0w,650
 solana_agent/factories/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-solana_agent/factories/agent_factory.py,sha256=v_Q0bKPoZOQ8e6pS4gcYeCpE-jSbktOhVF5oaAXn_-I,5779
+solana_agent/factories/agent_factory.py,sha256=c5TgZpDZ6m_q8F-Bv36Rl4dXq5yWNvAN2q9Nq_PtqqM,5547
 solana_agent/interfaces/__init__.py,sha256=IQs1WIM1FeKP1-kY2FEfyhol_dB-I-VAe2rD6jrVF6k,355
 solana_agent/interfaces/client/client.py,sha256=c9n5uF7UzwqCTU-jFz-33eq9TRyora_A5UbO4-EYlJU,1503
 solana_agent/interfaces/plugins/plugins.py,sha256=T8HPBsekmzVwfU_Rizp-vtzAeYkMlKMYD7U9d0Wjq9c,3338
@@ -19,17 +19,17 @@ solana_agent/interfaces/services/agent.py,sha256=34luGrUF5FNXLhF6JXwbfOSuo_SbMOm
 solana_agent/interfaces/services/query.py,sha256=m8Uc0uXT3apSOhX3N1QjLMPk1KdJhj7HDrJjWUpDPBc,1309
 solana_agent/interfaces/services/routing.py,sha256=gohkt5f9uYDLpu4iDVDk9yj8js9P56R6QHSIDNylgwA,438
 solana_agent/plugins/__init__.py,sha256=coZdgJKq1ExOaj6qB810i3rEhbjdVlrkN76ozt_Ojgo,193
-solana_agent/plugins/manager.py,sha256=sdHq6Cqn4juvgS6_3Iu_yjLVAcTRoIkPCqw8fWPkxgw,5096
-solana_agent/plugins/registry.py,sha256=ZUeczZ9bkdm9G6NBMDpY7sglOW2IrJLiLhboMNAHors,3072
+solana_agent/plugins/manager.py,sha256=Il49hXeqvu0b02pURNNp7mY8kp9_sqpi_vJIWBW5Hc0,5044
+solana_agent/plugins/registry.py,sha256=5S0DlUQKogsg1zLiRUIGMHEmGYHtOovU-S-5W1Mwo1A,3639
 solana_agent/plugins/tools/__init__.py,sha256=c0z7ij42gs94_VJrcn4Y8gUlTxMhsFNY6ahIsNswdLk,231
-solana_agent/plugins/tools/auto_tool.py,sha256=eDq2P-_D2PM7Dafpn55b1QU5LkM6BLdJGiWB9Wz8d_M,1495
+solana_agent/plugins/tools/auto_tool.py,sha256=DgES_cZ6xKSf_HJpFINpvJxrjVlk5oeqa7pZRBsR9SM,1575
 solana_agent/repositories/__init__.py,sha256=fP83w83CGzXLnSdq-C5wbw9EhWTYtqE2lQTgp46-X_4,163
-solana_agent/repositories/memory.py,sha256=xvgFscvnIGicIB9r1RwavhXSXGt2KRrloIheG3OWs-I,5221
+solana_agent/repositories/memory.py,sha256=eecl1P0fr_xFSWFKIJg99q90oCS9--ihPrMLH3G2AzM,7136
 solana_agent/services/__init__.py,sha256=ab_NXJmwYUCmCrCzuTlZ47bJZINW0Y0F5jfQ9OovidU,163
-solana_agent/services/agent.py,sha256=o8RJg1pBREfTK5gCLS2A0Lprg36gvn0Hzh6_TS1Hx6Y,16448
+solana_agent/services/agent.py,sha256=Nx3D7HrL9YRP1qI9Yzhm8HR8AXBfavFYnujXsSLdZ4A,16540
 solana_agent/services/query.py,sha256=4MEGR_Q2THcCCjln-IcSAw7GGNqJ1q2cyEVRx69OQ7I,10746
-solana_agent/services/routing.py,sha256=IPvBicgTYXqQ8iIRaatCsBGQVsOBGdAkq2i6U8hZlOY,6479
-solana_agent-19.1.0.dist-info/LICENSE,sha256=BnSRc-NSFuyF2s496l_4EyrwAP6YimvxWcjPiJ0J7g4,1057
-solana_agent-19.1.0.dist-info/METADATA,sha256=DD7fvgmWqOjDOTYePcppCLZP-gs0kOu-S9mUyOFxV7w,10859
-solana_agent-19.1.0.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
-solana_agent-19.1.0.dist-info/RECORD,,
+solana_agent/services/routing.py,sha256=PMCSG5m3uLMaHMj3dxNvNfcFZaeaDi7kMr7AEBCzwDE,6499
+solana_agent-20.0.0.dist-info/LICENSE,sha256=BnSRc-NSFuyF2s496l_4EyrwAP6YimvxWcjPiJ0J7g4,1057
+solana_agent-20.0.0.dist-info/METADATA,sha256=JQWWGntl9adbkC5gSghwYANhwM7TIOMgsEZz-BDG0Pw,10961
+solana_agent-20.0.0.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
+solana_agent-20.0.0.dist-info/RECORD,,