solana-agent 31.2.4__py3-none-any.whl → 31.2.6__py3-none-any.whl

solana_agent/domains/routing.py

@@ -5,10 +5,20 @@ from pydantic import BaseModel, Field
 class QueryAnalysis(BaseModel):
     """Analysis of a user query for routing purposes."""
 
-    primary_specialization: str = Field(..., description="Main specialization needed")
-    secondary_specializations: List[str] = Field(
-        ..., description="Other helpful specializations"
+    primary_agent: str = Field(
+        ...,
+        description="Name of the primary agent that should handle this query (must be one of the available agent names)",
+    )
+    secondary_agents: List[str] = Field(
+        default_factory=list,
+        description="Names of secondary agents that might be helpful (must be from the available agent names)",
+    )
+    complexity_level: int = Field(
+        default=1, description="Complexity level (1-5)", ge=1, le=5
+    )
+    topics: List[str] = Field(
+        default_factory=list, description="Key topics in the query"
+    )
+    confidence: float = Field(
+        default=0.5, description="Confidence in the analysis", ge=0.0, le=1.0
     )
-    complexity_level: int = Field(..., description="Complexity level (1-5)")
-    topics: List[str] = Field(..., description="Key topics in the query")
-    confidence: float = Field(..., description="Confidence in the analysis")

solana_agent/services/routing.py

@@ -1,11 +1,3 @@
-"""
-Routing service implementation.
-
-This service manages query routing to appropriate agents using an LLM-based
-analysis. It defaults to a small, low-cost model for routing to minimize
-overhead while maintaining quality.
-"""
-
 import logging
 from typing import Dict, List, Optional, Any
 from solana_agent.interfaces.services.routing import (
@@ -81,37 +73,48 @@ class RoutingService(RoutingServiceInterface):
 
         USER QUERY: {query}
 
-        Please determine:
-        1. Which agent is the primary best match for this query (must be one of the listed agents)
-        2. Any secondary agents that might be helpful (must be from the listed agents)
-        3. The complexity level (1-5, where 5 is most complex)
-        4. Any key topics or technologies mentioned
+        ROUTING RULES:
+        - Match the user query to the agent whose specialization best fits the user's intent
+        - Return the EXACT agent name that matches best
+
+        INSTRUCTIONS:
+        - primary_agent: The exact name of the best matching agent (e.g., "onboarding", "event_feedback")
+        - secondary_agents: Other agents that might help (usually empty)
+        - complexity_level: 1-5 (5 being most complex)
+        - topics: Key topics mentioned
+        - confidence: 0.0-1.0 (how confident you are in this routing decision)
+
+        For the query "{query}", which agent should handle it?
         """
 
         try:
             analysis = await self.llm_provider.parse_structured_output(
                 prompt=prompt,
-                system_prompt="Match user queries to the most appropriate agent based on specializations.",
+                system_prompt="You are an expert at routing user queries to the most appropriate AI agent. Always return the exact agent name that best matches the user's needs based on the specializations provided. If the user mentions a specific topic, prioritize agents whose specialization matches that topic.",
                 model_class=QueryAnalysis,
                 api_key=self.api_key,
                 base_url=self.base_url,
                 model=self.model,
             )
 
+            logger.debug(f"LLM analysis result: {analysis}")
+
             return {
-                "primary_specialization": analysis.primary_specialization,
-                "secondary_specializations": analysis.secondary_specializations,
+                "primary_specialization": analysis.primary_agent,
+                "secondary_specializations": analysis.secondary_agents,
                 "complexity_level": analysis.complexity_level,
                 "topics": analysis.topics,
                 "confidence": analysis.confidence,
             }
         except Exception as e:
-            logger.error(f"Error analyzing query: {e}")  # Use logger.error
+            logger.error(f"Error analyzing query: {e}")
+            logger.debug(f"Query that failed: {query}")
+            logger.debug(f"Available agents: {list(agents.keys())}")
             # Return default analysis on error
+            first_agent = list(agents.keys())[0] if agents else "general"
+            logger.debug(f"Defaulting to first agent: {first_agent}")
             return {
-                "primary_specialization": list(agents.keys())[0]
-                if agents
-                else "general",
+                "primary_specialization": first_agent,
                 "secondary_specializations": [],
                 "complexity_level": 1,
                 "topics": [],
@@ -143,9 +146,11 @@ class RoutingService(RoutingServiceInterface):
 
         # Always analyze with a small model to select the best agent
         analysis = await self._analyze_query(query)
+        logger.debug(f"Routing analysis for query '{query}': {analysis}")
         best_agent = await self._find_best_ai_agent(
             analysis["primary_specialization"], analysis["secondary_specializations"]
         )
+        logger.debug(f"Selected agent: {best_agent}")
         chosen = best_agent or next(iter(agents.keys()))
         self._last_agent = chosen
         return chosen
@@ -171,6 +176,7 @@ class RoutingService(RoutingServiceInterface):
 
         # First, check if primary_specialization is directly an agent name
         if primary_specialization in ai_agents:
+            logger.debug(f"Direct agent match: {primary_specialization}")
             return primary_specialization
 
         # If not a direct agent name match, use specialization matching
@@ -185,6 +191,9 @@ class RoutingService(RoutingServiceInterface):
                 or primary_specialization.lower() in agent.specialization.lower()
             ):
                 score += 10
+                logger.debug(
+                    f"Specialization match for {agent_id}: '{agent.specialization}' matches '{primary_specialization}'"
+                )
 
             # Check secondary specializations
             for sec_spec in secondary_specializations:
@@ -209,6 +218,8 @@ class RoutingService(RoutingServiceInterface):
 
         # If no match found, return first agent as fallback
         if ai_agents:
-            return next(iter(ai_agents.keys()))
+            fallback_agent = next(iter(ai_agents.keys()))
+            logger.debug(f"No match found, using fallback agent: {fallback_agent}")
+            return fallback_agent
 
         return None

{solana_agent-31.2.4.dist-info → solana_agent-31.2.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: solana-agent
-Version: 31.2.4
+Version: 31.2.6
 Summary: AI Agents for Solana
 License: MIT
 Keywords: solana,solana ai,solana agent,ai,ai agent,ai agents
@@ -15,14 +15,14 @@ Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Requires-Dist: instructor (==1.11.3)
-Requires-Dist: llama-index-core (==0.14.0)
+Requires-Dist: llama-index-core (==0.14.1)
 Requires-Dist: llama-index-embeddings-openai (==0.5.1)
-Requires-Dist: logfire (==4.5.0)
-Requires-Dist: openai (==1.107.0)
+Requires-Dist: logfire (==4.7.0)
+Requires-Dist: openai (==1.107.2)
 Requires-Dist: pillow (==11.3.0)
 Requires-Dist: pinecone[asyncio] (==7.3.0)
 Requires-Dist: pydantic (>=2)
-Requires-Dist: pymongo (==4.14.1)
+Requires-Dist: pymongo (==4.15.0)
 Requires-Dist: pypdf (==6.0.0)
 Requires-Dist: rich (>=13,<14.0)
 Requires-Dist: scrubadub (==2.0.1)

{solana_agent-31.2.4.dist-info → solana_agent-31.2.6.dist-info}/RECORD

@@ -10,7 +10,7 @@ solana_agent/client/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSu
 solana_agent/client/solana_agent.py,sha256=hLtiR3xD1eFww7XRdg4dTvxlJnTCepilYmEfABn9L7E,10344
 solana_agent/domains/__init__.py,sha256=HiC94wVPRy-QDJSSRywCRrhrFfTBeHjfi5z-QfZv46U,168
 solana_agent/domains/agent.py,sha256=8pAi1-kIgzFNANt3dyQjw-1zbThcNdpEllbAGWi79uI,2841
-solana_agent/domains/routing.py,sha256=1yR4IswGcmREGgbOOI6TKCfuM7gYGOhQjLkBqnZ-rNo,582
+solana_agent/domains/routing.py,sha256=rb7YyeH4CJmjtuoBMk7f8kUFddcT5tUBZ9Bj8mJ4VV8,845
 solana_agent/factories/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 solana_agent/factories/agent_factory.py,sha256=d9VuD5E9khqVXU_Qu67zKU2yVvXHK2EmercDmSZ4stk,14226
 solana_agent/guardrails/pii.py,sha256=FCz1IC3mmkr41QFFf5NaC0fwJrVkwFsxgyOCS2POO5I,4428
@@ -40,9 +40,9 @@ solana_agent/services/agent.py,sha256=LWjsdmCeygwmjFoazOCVhrb0hdZHQDEQo_DFWZe57L
 solana_agent/services/knowledge_base.py,sha256=ZvOPrSmcNDgUzz4bJIQ4LeRl9vMZiK9hOfs71IpB7Bk,32735
 solana_agent/services/query.py,sha256=oqNFbQsz2FiSswGkt8ZlNaOR8DAz66hgWXD5kHc7c-M,71428
 solana_agent/services/realtime.py,sha256=6_44-JaKN0V4gkizaisGLPsopM5Z8xymQcCbq5V3yEc,21054
-solana_agent/services/routing.py,sha256=hsHe8HSGO_xFc0A17WIOGTidLTfLSfApQw3l2HHqkLo,7614
-solana_agent-31.2.4.dist-info/LICENSE,sha256=BnSRc-NSFuyF2s496l_4EyrwAP6YimvxWcjPiJ0J7g4,1057
-solana_agent-31.2.4.dist-info/METADATA,sha256=zRrJ4pjQ3CEo0DpUT21txy1_vg2JWB4A4BDeQcAw_tI,31168
-solana_agent-31.2.4.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-solana_agent-31.2.4.dist-info/entry_points.txt,sha256=-AuT_mfqk8dlZ0pHuAjx1ouAWpTRjpqvEUa6YV3lmc0,53
-solana_agent-31.2.4.dist-info/RECORD,,
+solana_agent/services/routing.py,sha256=FXVeOwcJiZ77JEcr2Xbd_tJaWCTqcct7KJmwKtyl-io,8602
+solana_agent-31.2.6.dist-info/LICENSE,sha256=BnSRc-NSFuyF2s496l_4EyrwAP6YimvxWcjPiJ0J7g4,1057
+solana_agent-31.2.6.dist-info/METADATA,sha256=e8tQOCFEg87jPmG_e22ZsrgknGkqREuJn3LaHuYGOWw,31168
+solana_agent-31.2.6.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+solana_agent-31.2.6.dist-info/entry_points.txt,sha256=-AuT_mfqk8dlZ0pHuAjx1ouAWpTRjpqvEUa6YV3lmc0,53
+solana_agent-31.2.6.dist-info/RECORD,,