solana-agent 31.2.4__tar.gz → 31.2.5__tar.gz

{solana_agent-31.2.4 → solana_agent-31.2.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: solana-agent
-Version: 31.2.4
+Version: 31.2.5
 Summary: AI Agents for Solana
 License: MIT
 Keywords: solana,solana ai,solana agent,ai,ai agent,ai agents
@@ -17,12 +17,12 @@ Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Requires-Dist: instructor (==1.11.3)
 Requires-Dist: llama-index-core (==0.14.0)
 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 → solana_agent-31.2.5}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "solana-agent"
-version = "31.2.4"
+version = "31.2.5"
 description = "AI Agents for Solana"
 authors = ["Bevan Hunt <bevan@bevanhunt.com>"]
 license = "MIT"
@@ -24,9 +24,9 @@ testpaths = ["tests"]
 
 [tool.poetry.dependencies]
 python = ">=3.12,<4.0"
-openai = "1.107.0"
+openai = "1.107.2"
 pydantic = ">=2"
-pymongo = "4.14.1"
+pymongo = "4.15.0"
 zep-cloud = "3.4.3"
 instructor = "1.11.3"
 pinecone = { version = "7.3.0", extras = ["asyncio"] }
@@ -34,7 +34,7 @@ llama-index-core = "0.14.0"
 llama-index-embeddings-openai = "0.5.1"
 pypdf = "6.0.0"
 scrubadub = "2.0.1"
-logfire = "4.5.0"
+logfire = "4.7.0"
 typer = "0.17.4"
 rich = ">=13,<14.0"
 pillow = "11.3.0"
@@ -43,7 +43,7 @@ websockets = ">=13,<16"
 [tool.poetry.group.dev.dependencies]
 pytest = "^8.4.2"
 pytest-cov = "^7.0.0"
-pytest-asyncio = "^1.1.0"
+pytest-asyncio = "^1.2.0"
 pytest-mock = "^3.15.0"
 pytest-github-actions-annotate-failures = "^0.3.0"
 sphinx = "^8.2.3"
@@ -51,7 +51,7 @@ sphinx-rtd-theme = "^3.0.2"
 myst-parser = "^4.0.1"
 sphinx-autobuild = "^2025.08.25"
 mongomock = "^4.3.0"
-ruff = "^0.12.12"
+ruff = "^0.13.0"
 
 [tool.poetry.scripts]
 solana-agent = "solana_agent.cli:app"

{solana_agent-31.2.4 → solana_agent-31.2.5}/solana_agent/domains/routing.py

@@ -5,9 +5,13 @@ 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(
+        ...,
+        description="Names of secondary agents that might be helpful (must be from the available agent names)",
     )
     complexity_level: int = Field(..., description="Complexity level (1-5)")
     topics: List[str] = Field(..., description="Key topics in the query")

{solana_agent-31.2.4 → solana_agent-31.2.5}/solana_agent/services/routing.py

@@ -81,26 +81,34 @@ class RoutingService(RoutingServiceInterface):
 
         USER QUERY: {query}
 
+        INSTRUCTIONS:
+        - Look at the user query and match it to the most appropriate agent from the list above
+        - If the user mentions a specific topic or need that matches an agent's specialization, choose that agent
+        - Return the EXACT agent name (not the specialization description)
+
         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
+        1. primary_agent: The exact name of the best matching agent (e.g., "onboarding", "support")
+        2. secondary_agents: Names of other agents that might help (empty list if none)
+        3. complexity_level: 1-5 (5 being most complex)
+        4. topics: Key topics mentioned
+        5. confidence: 0.0-1.0 (how confident you are in this routing decision)
         """
 
         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,
@@ -143,9 +151,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 +181,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 +196,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 +223,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