solana-agent 31.1.0__py3-none-any.whl → 31.1.1__py3-none-any.whl

solana_agent/factories/agent_factory.py

@@ -186,9 +186,16 @@ class SolanaAgentFactory:
         )
 
         # Create routing service
+        # Optional routing model override (use small, cheap model by default in service)
+        routing_model = (
+            config.get("openai", {}).get("routing_model")
+            if isinstance(config.get("openai"), dict)
+            else None
+        )
         routing_service = RoutingService(
             llm_provider=llm_adapter,
             agent_service=agent_service,
+            model=routing_model,
         )
 
         # Debug the agent service tool registry

solana_agent/services/query.py

@@ -7,6 +7,7 @@ clean separation of concerns.
 """
 
 import logging
+import re
 from typing import Any, AsyncGenerator, Dict, List, Literal, Optional, Type, Union
 
 from pydantic import BaseModel
@@ -22,9 +23,7 @@ from solana_agent.interfaces.providers.memory import (
 from solana_agent.interfaces.services.knowledge_base import (
     KnowledgeBaseService as KnowledgeBaseInterface,
 )
-from solana_agent.interfaces.guardrails.guardrails import (
-    InputGuardrail,
-)
+from solana_agent.interfaces.guardrails.guardrails import InputGuardrail
 
 from solana_agent.services.agent import AgentService
 from solana_agent.services.routing import RoutingService
@@ -44,16 +43,7 @@ class QueryService(QueryServiceInterface):
         kb_results_count: int = 3,
         input_guardrails: List[InputGuardrail] = None,
     ):
-        """Initialize the query service.
-
-        Args:
-            agent_service: Service for AI agent management
-            routing_service: Service for routing queries to appropriate agents
-            memory_provider: Optional provider for memory storage and retrieval
-            knowledge_base: Optional provider for knowledge base interactions
-            kb_results_count: Number of results to retrieve from knowledge base
-            input_guardrails: List of input guardrail instances
-        """
+        """Initialize the query service."""
         self.agent_service = agent_service
         self.routing_service = routing_service
         self.memory_provider = memory_provider
@@ -92,26 +82,9 @@ class QueryService(QueryServiceInterface):
         capture_schema: Optional[Dict[str, Any]] = None,
         capture_name: Optional[str] = None,
     ) -> AsyncGenerator[Union[str, bytes, BaseModel], None]:  # pragma: no cover
-        """Process the user request with appropriate agent and apply input guardrails.
-
-        Args:
-            user_id: User ID
-            query: Text query or audio bytes
-            images: Optional list of image URLs (str) or image bytes.
-            output_format: Response format ("text" or "audio")
-            audio_voice: Voice for TTS (text-to-speech)
-            audio_instructions: Audio voice instructions
-            audio_output_format: Audio output format
-            audio_input_format: Audio input format
-            prompt: Optional prompt for the agent
-            router: Optional routing service for processing
-            output_model: Optional Pydantic model for structured output
-
-        Yields:
-            Response chunks (text strings or audio bytes)
-        """
+        """Process the user request and generate a response."""
         try:
-            # --- 1. Handle Audio Input & Extract Text ---
+            # 1) Transcribe audio or accept text
             user_text = ""
             if not isinstance(query, str):
                 logger.info(
@@ -128,123 +101,386 @@ class QueryService(QueryServiceInterface):
                 user_text = query
                 logger.info(f"Received text input length: {len(user_text)}")
 
-            # --- 2. Apply Input Guardrails ---
+            # 2) Input guardrails
             original_text = user_text
-            processed_text = user_text
             for guardrail in self.input_guardrails:
                 try:
-                    processed_text = await guardrail.process(processed_text)
-                    logger.debug(
-                        f"Applied input guardrail: {guardrail.__class__.__name__}"
-                    )
+                    user_text = await guardrail.process(user_text)
                 except Exception as e:
-                    logger.error(
-                        f"Error applying input guardrail {guardrail.__class__.__name__}: {e}",
-                        exc_info=True,
-                    )
-            if processed_text != original_text:
+                    logger.debug(f"Guardrail error: {e}")
+            if user_text != original_text:
                 logger.info(
-                    f"Input guardrails modified user text. Original length: {len(original_text)}, New length: {len(processed_text)}"
+                    f"Input guardrails modified user text. Original length: {len(original_text)}, New length: {len(user_text)}"
                 )
-            user_text = processed_text  # Use the processed text going forward
-            # --- End Apply Input Guardrails ---
 
-            # --- 3. Handle Simple Greetings ---
-            # Simple greetings typically don't involve images
-            if not images and user_text.strip().lower() in [
-                "test",
-                "hello",
+            # 3) Greetings shortcut
+            if not images and user_text.strip().lower() in {
                 "hi",
+                "hello",
                 "hey",
                 "ping",
-            ]:
-                response = "Hello! How can I help you today?"
-                logger.info("Handling simple greeting.")
+                "test",
+            }:
+                greeting = "Hello! How can I help you today?"
                 if output_format == "audio":
                     async for chunk in self.agent_service.llm_provider.tts(
-                        text=response,
+                        text=greeting,
                         voice=audio_voice,
                         response_format=audio_output_format,
                         instructions=audio_instructions,
                     ):
                         yield chunk
                 else:
-                    yield response
-
-                # Store simple interaction in memory (using processed user_text)
+                    yield greeting
                 if self.memory_provider:
-                    await self._store_conversation(user_id, user_text, response)
+                    await self._store_conversation(user_id, original_text, greeting)
                 return
 
-            # --- 4. Get Memory Context ---
+            # 4) Memory context (conversation history)
             memory_context = ""
             if self.memory_provider:
                 try:
                     memory_context = await self.memory_provider.retrieve(user_id)
-                    logger.info(
-                        f"Retrieved memory context length: {len(memory_context)}"
-                    )
-                except Exception as e:
-                    logger.error(f"Error retrieving memory context: {e}", exc_info=True)
+                except Exception:
+                    memory_context = ""
 
-            # --- 5. Retrieve Relevant Knowledge ---
+            # 5) Knowledge base context
             kb_context = ""
             if self.knowledge_base:
                 try:
-                    # Use processed user_text for KB query
                     kb_results = await self.knowledge_base.query(
                         query_text=user_text,
                         top_k=self.kb_results_count,
                         include_content=True,
-                        include_metadata=False,  # Keep metadata minimal for context
+                        include_metadata=False,
                     )
-
                     if kb_results:
-                        kb_context = "**KNOWLEDGE BASE (CRITICAL: MAKE THIS INFORMATION THE TOP PRIORITY):**\n"
-                        for i, result in enumerate(kb_results, 1):
-                            content = result.get("content", "").strip()
-                            kb_context += f"[{i}] {content}\n\n"
-                        logger.info(
-                            f"Retrieved {len(kb_results)} results from Knowledge Base."
-                        )
-                    else:
-                        logger.info("No relevant results found in Knowledge Base.")
-                except Exception as e:
-                    logger.error(f"Error retrieving knowledge: {e}", exc_info=True)
-
-            # --- 6. Route Query ---
-            agent_name = "default"  # Fallback agent
+                        kb_lines = [
+                            "**KNOWLEDGE BASE (CRITICAL: MAKE THIS INFORMATION THE TOP PRIORITY):**"
+                        ]
+                        for i, r in enumerate(kb_results, 1):
+                            kb_lines.append(f"[{i}] {r.get('content', '').strip()}\n")
+                        kb_context = "\n".join(kb_lines)
+                except Exception:
+                    kb_context = ""
+
+            # 6) Route query (and fetch previous assistant message)
+            agent_name = "default"
+            prev_assistant = ""
+            routing_input = user_text
+            if self.memory_provider:
+                try:
+                    prev_docs = self.memory_provider.find(
+                        collection="conversations",
+                        query={"user_id": user_id},
+                        sort=[("timestamp", -1)],
+                        limit=1,
+                    )
+                    if prev_docs:
+                        prev_user_msg = (prev_docs[0] or {}).get(
+                            "user_message", ""
+                        ) or ""
+                        prev_assistant = (prev_docs[0] or {}).get(
+                            "assistant_message", ""
+                        ) or ""
+                        if prev_user_msg:
+                            routing_input = (
+                                f"previous_user_message: {prev_user_msg}\n"
+                                f"current_user_message: {user_text}"
+                            )
+                except Exception:
+                    pass
             try:
-                # Use processed user_text for routing (images generally don't affect routing logic here)
                 if router:
-                    agent_name = await router.route_query(user_text)
+                    agent_name = await router.route_query(routing_input)
                 else:
-                    agent_name = await self.routing_service.route_query(user_text)
-                logger.info(f"Routed query to agent: {agent_name}")
+                    agent_name = await self.routing_service.route_query(routing_input)
+            except Exception:
+                agent_name = "default"
+
+            # 7) Captured data context + incremental save using previous assistant message
+            capture_context = ""
+            form_complete = False
+
+            # Helpers
+            def _non_empty(v: Any) -> bool:
+                if v is None:
+                    return False
+                if isinstance(v, str):
+                    s = v.strip().lower()
+                    return s not in {"", "null", "none", "n/a", "na", "undefined", "."}
+                if isinstance(v, (list, dict, tuple, set)):
+                    return len(v) > 0
+                return True
+
+            def _parse_numbers_list(s: str) -> List[str]:
+                nums = re.findall(r"\b(\d+)\b", s)
+                # dedupe keep order
+                seen, out = set(), []
+                for n in nums:
+                    if n not in seen:
+                        seen.add(n)
+                        out.append(n)
+                return out
+
+            def _extract_numbered_options(text: str) -> Dict[str, str]:
+                """Parse previous assistant message for lines like:
+                '1) Foo', '2. Bar', '- 3) Baz', '* 4. Buzz'
+                Returns mapping '1' -> 'Foo', etc.
+                """
+                options: Dict[str, str] = {}
+                if not text:
+                    return options
+                for raw in text.splitlines():
+                    line = raw.strip()
+                    if not line:
+                        continue
+                    # Common Markdown patterns: "1. Label", "1) Label", "- 1) Label", "* 1. Label"
+                    m = re.match(r"^(?:[-*]\s*)?(\d+)[\.)]?\s+(.*)$", line)
+                    if m:
+                        idx, label = m.group(1), m.group(2).strip()
+                        # Strip trailing markdown soft-break spaces
+                        label = label.rstrip()
+                        # Ignore labels that are too short or look like continuations
+                        if len(label) >= 1:
+                            options[idx] = label
+                return options
+
+            def _detect_field_from_prev_question(
+                prev_text: str, schema: Optional[Dict[str, Any]]
+            ) -> Optional[str]:
+                if not prev_text or not isinstance(schema, dict):
+                    return None
+                t = prev_text.lower()
+                # Heuristic synonyms for your onboarding schema
+                patterns = [
+                    ("ideas", ["which ideas attract you", "ideas"]),
+                    ("description", ["please describe yourself", "describe yourself"]),
+                    ("myself", ["tell us about yourself", "about yourself"]),
+                    ("questions", ["do you have any questions"]),
+                    ("rating", ["rating", "1 to 5", "how satisfied", "how happy"]),
+                    ("email", ["email"]),
+                    ("phone", ["phone"]),
+                    ("name", ["name"]),
+                    ("city", ["city"]),
+                    ("state", ["state"]),
+                ]
+                candidates = set((schema.get("properties") or {}).keys())
+                for field, keys in patterns:
+                    if field in candidates and any(key in t for key in keys):
+                        return field
+                # Fallback: property name appears directly
+                for field in candidates:
+                    if field in t:
+                        return field
+                return None
+
+            # Resolve active capture from args or agent config
+            active_capture_name = capture_name
+            active_capture_schema = capture_schema
+            if not active_capture_name or not active_capture_schema:
+                try:
+                    cap_cfg = self.agent_service.get_agent_capture(agent_name)
+                    if cap_cfg:
+                        active_capture_name = active_capture_name or cap_cfg.get("name")
+                        active_capture_schema = active_capture_schema or cap_cfg.get(
+                            "schema"
+                        )
+                except Exception:
+                    pass
+
+            latest_by_name: Dict[str, Dict[str, Any]] = {}
+            if self.memory_provider:
+                try:
+                    docs = self.memory_provider.find(
+                        collection="captures",
+                        query={"user_id": user_id},
+                        sort=[("timestamp", -1)],
+                        limit=100,
+                    )
+                    for d in docs or []:
+                        name = (d or {}).get("capture_name")
+                        if not name or name in latest_by_name:
+                            continue
+                        latest_by_name[name] = {
+                            "data": (d or {}).get("data", {}) or {},
+                            "mode": (d or {}).get("mode", "once"),
+                            "agent": (d or {}).get("agent_name"),
+                        }
+                except Exception:
+                    pass
+
+            # Incremental save: use prev_assistant's numbered list to map numeric reply -> labels
+            incremental: Dict[str, Any] = {}
+            try:
+                if (
+                    self.memory_provider
+                    and active_capture_name
+                    and isinstance(active_capture_schema, dict)
+                ):
+                    props = (active_capture_schema or {}).get("properties", {})
+                    required_fields = list(
+                        (active_capture_schema or {}).get("required", []) or []
+                    )
+                    # Prefer a field detected from prev assistant; else if exactly one required missing, use it
+                    target_field: Optional[str] = _detect_field_from_prev_question(
+                        prev_assistant, active_capture_schema
+                    )
+                    active_data_existing = (
+                        latest_by_name.get(active_capture_name, {}) or {}
+                    ).get("data", {}) or {}
+
+                    def _missing_required() -> List[str]:
+                        return [
+                            f
+                            for f in required_fields
+                            if not _non_empty(active_data_existing.get(f))
+                        ]
+
+                    if not target_field:
+                        missing = _missing_required()
+                        if len(missing) == 1:
+                            target_field = missing[0]
+
+                    if target_field:
+                        f_schema = props.get(target_field, {}) or {}
+                        f_type = f_schema.get("type")
+                        number_to_label = _extract_numbered_options(prev_assistant)
+
+                        if number_to_label:
+                            # Map any numbers in user's reply to their labels
+                            nums = _parse_numbers_list(user_text)
+                            labels = [
+                                number_to_label[n] for n in nums if n in number_to_label
+                            ]
+                            if labels:
+                                if f_type == "array":
+                                    incremental[target_field] = labels
+                                else:
+                                    incremental[target_field] = labels[0]
+
+                        # If we didn't map via options, fallback to type-based parse
+                        if target_field not in incremental:
+                            if f_type == "number":
+                                m = re.search(r"\b([0-9]+(?:\.[0-9]+)?)\b", user_text)
+                                if m:
+                                    try:
+                                        incremental[target_field] = float(m.group(1))
+                                    except Exception:
+                                        pass
+                            elif f_type == "array":
+                                # Accept CSV-style input as array of strings
+                                parts = [
+                                    p.strip()
+                                    for p in re.split(r"[,\n;]+", user_text)
+                                    if p.strip()
+                                ]
+                                if parts:
+                                    incremental[target_field] = parts
+                            else:  # string/default
+                                if user_text.strip():
+                                    incremental[target_field] = user_text.strip()
+
+                    # Filter out empty junk and save
+                    if incremental:
+                        cleaned = {
+                            k: v for k, v in incremental.items() if _non_empty(v)
+                        }
+                        if cleaned:
+                            try:
+                                await self.memory_provider.save_capture(
+                                    user_id=user_id,
+                                    capture_name=active_capture_name,
+                                    agent_name=agent_name,
+                                    data=cleaned,
+                                    schema=active_capture_schema,
+                                )
+                            except Exception as se:
+                                logger.error(f"Error saving incremental capture: {se}")
             except Exception as e:
-                logger.error(
-                    f"Error during routing, falling back to default agent: {e}",
-                    exc_info=True,
+                logger.debug(f"Incremental extraction skipped: {e}")
+
+            # Build capture context, merging in incremental immediately (avoid read lag)
+            def _get_active_data(name: Optional[str]) -> Dict[str, Any]:
+                if not name:
+                    return {}
+                base = (latest_by_name.get(name, {}) or {}).get("data", {}) or {}
+                if incremental:
+                    base = {**base, **incremental}
+                return base
+
+            lines: List[str] = []
+            if active_capture_name and isinstance(active_capture_schema, dict):
+                active_data = _get_active_data(active_capture_name)
+                required_fields = list(
+                    (active_capture_schema or {}).get("required", []) or []
                 )
+                missing = [
+                    f for f in required_fields if not _non_empty(active_data.get(f))
+                ]
+                form_complete = len(missing) == 0 and len(required_fields) > 0
 
-            # --- 7. Combine Context ---
+                lines.append(
+                    "CAPTURED FORM STATE (Authoritative; do not re-ask filled values):"
+                )
+                lines.append(f"- form_name: {active_capture_name}")
+                if active_data:
+                    pairs = [
+                        f"{k}: {v}" for k, v in active_data.items() if _non_empty(v)
+                    ]
+                    lines.append(
+                        f"- filled_fields: {', '.join(pairs) if pairs else '(none)'}"
+                    )
+                else:
+                    lines.append("- filled_fields: (none)")
+                lines.append(
+                    f"- missing_required_fields: {', '.join(missing) if missing else '(none)'}"
+                )
+                lines.append("")
+
+            if latest_by_name:
+                lines.append("OTHER CAPTURED USER DATA (for reference):")
+                for cname, info in latest_by_name.items():
+                    if cname == active_capture_name:
+                        continue
+                    data = info.get("data", {}) or {}
+                    if data:
+                        pairs = [f"{k}: {v}" for k, v in data.items() if _non_empty(v)]
+                        lines.append(
+                            f"- {cname}: {', '.join(pairs) if pairs else '(none)'}"
+                        )
+                    else:
+                        lines.append(f"- {cname}: (none)")
+
+            if lines:
+                capture_context = "\n".join(lines) + "\n\n"
+
+            # Merge contexts
             combined_context = ""
+            if capture_context:
+                combined_context += capture_context
             if memory_context:
-                combined_context += f"CONVERSATION HISTORY (Use for context, but prioritize tools/KB for facts):\n{memory_context}\n\n"
+                combined_context += f"CONVERSATION HISTORY (Use for continuity; not authoritative for facts):\n{memory_context}\n\n"
             if kb_context:
-                combined_context += f"{kb_context}\n"
-
-            if memory_context or kb_context:
-                combined_context += "CRITICAL PRIORITIZATION GUIDE: For factual or current information, prioritize Knowledge Base results and Tool results (if applicable) over Conversation History.\n\n"
-            logger.debug(f"Combined context length: {len(combined_context)}")
+                combined_context += kb_context + "\n"
+            if combined_context:
+                combined_context += (
+                    "PRIORITIZATION GUIDE:\n"
+                    "- Prefer Captured User Data for user-specific fields.\n"
+                    "- Prefer KB/tools for facts.\n"
+                    "- History is for tone and continuity.\n\n"
+                    "FORM FLOW RULES:\n"
+                    "- Ask exactly one missing required field per turn.\n"
+                    "- Do NOT re-ask or verify values present in Captured User Data (auto-saved, authoritative).\n"
+                    "- If no required fields are missing, proceed without further capture questions.\n\n"
+                )
 
-            # --- 8. Generate Response ---
-            # Pass the processed user_text and images to the agent service
+            # 8) Generate response
             if output_format == "audio":
                 async for audio_chunk in self.agent_service.generate_response(
                     agent_name=agent_name,
                     user_id=user_id,
-                    query=user_text,  # Pass processed text
+                    query=user_text,
                     images=images,
                     memory_context=combined_context,
                     output_format="audio",
@@ -254,20 +490,17 @@ class QueryService(QueryServiceInterface):
                     prompt=prompt,
                 ):
                     yield audio_chunk
-
-                # Store conversation using processed user_text
-                # Note: Storing images in history is not directly supported by current memory provider interface
                 if self.memory_provider:
                     await self._store_conversation(
                         user_id=user_id,
-                        user_message=user_text,  # Store only text part of user query
+                        user_message=user_text,
                         assistant_message=self.agent_service.last_text_response,
                     )
             else:
                 full_text_response = ""
-                # If capture_schema is provided, we run a structured output pass first
                 capture_data: Optional[BaseModel] = None
-                # If no explicit capture provided, use the agent's configured capture
+
+                # Resolve agent capture if not provided
                 if not capture_schema or not capture_name:
                     try:
                         cap = self.agent_service.get_agent_capture(agent_name)
@@ -277,9 +510,9 @@ class QueryService(QueryServiceInterface):
                     except Exception:
                         pass
 
-                if capture_schema and capture_name:
+                # If form is complete, ask for structured output JSON
+                if capture_schema and capture_name and form_complete:
                     try:
-                        # Build a dynamic Pydantic model from JSON schema
                         DynamicModel = self._build_model_from_json_schema(
                             capture_name, capture_schema
                         )
@@ -293,14 +526,13 @@ class QueryService(QueryServiceInterface):
                             prompt=(
                                 (
                                     prompt
-                                    + "\n\nReturn only the JSON for the requested schema."
+                                    + "\n\nUsing the captured user data above, return only the JSON for the requested schema. Do not invent values."
                                 )
                                 if prompt
-                                else "Return only the JSON for the requested schema."
+                                else "Using the captured user data above, return only the JSON for the requested schema. Do not invent values."
                             ),
                             output_model=DynamicModel,
                         ):
-                            # This yields a pydantic model instance
                             capture_data = result  # type: ignore
                             break
                     except Exception as e:
@@ -309,8 +541,8 @@ class QueryService(QueryServiceInterface):
                 async for chunk in self.agent_service.generate_response(
                     agent_name=agent_name,
                     user_id=user_id,
-                    query=user_text,  # Pass processed text
-                    images=images,  # <-- Pass images
+                    query=user_text,
+                    images=images,
                     memory_context=combined_context,
                     output_format="text",
                     prompt=prompt,
@@ -320,16 +552,14 @@ class QueryService(QueryServiceInterface):
                     if output_model is None:
                         full_text_response += chunk
 
-                # Store conversation using processed user_text
-                # Note: Storing images in history is not directly supported by current memory provider interface
                 if self.memory_provider and full_text_response:
                     await self._store_conversation(
                         user_id=user_id,
-                        user_message=user_text,  # Store only text part of user query
+                        user_message=user_text,
                         assistant_message=full_text_response,
                     )
 
-                # Persist capture if available
+                # Save final capture data if the model returned it
                 if (
                     self.memory_provider
                     and capture_schema
@@ -337,11 +567,10 @@ class QueryService(QueryServiceInterface):
                     and capture_data is not None
                 ):
                     try:
-                        # pydantic v2: model_dump
                         data_dict = (
-                            capture_data.model_dump()  # type: ignore[attr-defined]
+                            capture_data.model_dump()
                             if hasattr(capture_data, "model_dump")
-                            else capture_data.dict()  # type: ignore
+                            else capture_data.dict()
                         )
                         await self.memory_provider.save_capture(
                             user_id=user_id,
@@ -371,52 +600,29 @@ class QueryService(QueryServiceInterface):
                         yield chunk
                 except Exception as tts_e:
                     logger.error(f"Error during TTS for error message: {tts_e}")
-                    # Fallback to yielding text error if TTS fails
                     yield error_msg + f" (TTS Error: {tts_e})"
             else:
                 yield error_msg
 
     async def delete_user_history(self, user_id: str) -> None:
-        """Delete all conversation history for a user.
-
-        Args:
-            user_id: User ID
-        """
+        """Delete all conversation history for a user."""
         if self.memory_provider:
             try:
                 await self.memory_provider.delete(user_id)
-                logger.info(f"Deleted conversation history for user: {user_id}")
             except Exception as e:
-                logger.error(
-                    f"Error deleting user history for {user_id}: {e}", exc_info=True
-                )
+                logger.error(f"Error deleting user history for {user_id}: {e}")
         else:
-            logger.warning(
-                "Attempted to delete user history, but no memory provider is configured."
-            )
+            logger.debug("No memory provider; skip delete_user_history")
 
     async def get_user_history(
         self,
         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",
     ) -> Dict[str, Any]:
-        """Get paginated message history for a user.
-
-        Args:
-            user_id: User ID
-            page_num: Page number (starting from 1)
-            page_size: Number of messages per page
-            sort_order: Sort order ("asc" or "desc")
-
-        Returns:
-            Dictionary with paginated results and metadata.
-        """
+        """Get paginated message history for a user."""
         if not self.memory_provider:
-            logger.warning(
-                "Attempted to get user history, but no memory provider is configured."
-            )
             return {
                 "data": [],
                 "total": 0,
@@ -425,20 +631,13 @@ class QueryService(QueryServiceInterface):
                 "total_pages": 0,
                 "error": "Memory provider not available",
             }
-
         try:
-            # Calculate skip and limit for pagination
             skip = (page_num - 1) * page_size
-
-            # Get total count of documents
             total = self.memory_provider.count_documents(
                 collection="conversations", query={"user_id": user_id}
             )
-
-            # Calculate total pages
             total_pages = (total + page_size - 1) // page_size if total > 0 else 0
 
-            # Get paginated results
             conversations = self.memory_provider.find(
                 collection="conversations",
                 query={"user_id": user_id},
@@ -447,39 +646,27 @@ class QueryService(QueryServiceInterface):
                 limit=page_size,
             )
 
-            # Format the results
-            formatted_conversations = []
+            formatted: List[Dict[str, Any]] = []
             for conv in conversations:
-                timestamp = (
-                    int(conv.get("timestamp").timestamp())
-                    if conv.get("timestamp")
-                    else None
-                )
-                # Assuming the stored format matches what _store_conversation saves
-                # (which currently only stores text messages)
-                formatted_conversations.append(
+                ts = conv.get("timestamp")
+                ts_epoch = int(ts.timestamp()) if ts else None
+                formatted.append(
                     {
                         "id": str(conv.get("_id")),
-                        "user_message": conv.get("user_message"),  # Or how it's stored
-                        "assistant_message": conv.get(
-                            "assistant_message"
-                        ),  # Or how it's stored
-                        "timestamp": timestamp,
+                        "user_message": conv.get("user_message"),
+                        "assistant_message": conv.get("assistant_message"),
+                        "timestamp": ts_epoch,
                     }
                 )
 
-            logger.info(
-                f"Retrieved page {page_num}/{total_pages} of history for user {user_id}"
-            )
             return {
-                "data": formatted_conversations,
+                "data": formatted,
                 "total": total,
                 "page": page_num,
                 "page_size": page_size,
                 "total_pages": total_pages,
                 "error": None,
             }
-
         except Exception as e:
             import traceback
 
@@ -498,48 +685,29 @@ class QueryService(QueryServiceInterface):
     async def _store_conversation(
         self, user_id: str, user_message: str, assistant_message: str
     ) -> None:
-        """Store conversation history in memory provider.
-
-        Args:
-            user_id: User ID
-            user_message: User message (text part, potentially processed by input guardrails)
-            assistant_message: Assistant message (potentially processed by output guardrails)
-        """
-        if self.memory_provider:
-            try:
-                # Store only the text parts for now, as memory provider interface
-                # doesn't explicitly handle image data storage in history.
-                await self.memory_provider.store(
-                    user_id,
-                    [
-                        {"role": "user", "content": user_message},
-                        {"role": "assistant", "content": assistant_message},
-                    ],
-                )
-                logger.info(f"Stored conversation for user {user_id}")
-            except Exception as e:
-                logger.error(
-                    f"Error storing conversation for user {user_id}: {e}", exc_info=True
-                )
-        else:
-            logger.debug(
-                "Memory provider not configured, skipping conversation storage."
+        """Store conversation history in memory provider."""
+        if not self.memory_provider:
+            return
+        try:
+            await self.memory_provider.store(
+                user_id,
+                [
+                    {"role": "user", "content": user_message},
+                    {"role": "assistant", "content": assistant_message},
+                ],
             )
+        except Exception as e:
+            logger.error(f"Store conversation error for {user_id}: {e}")
 
     def _build_model_from_json_schema(
         self, name: str, schema: Dict[str, Any]
     ) -> Type[BaseModel]:
-        """Create a Pydantic model dynamically from a JSON Schema subset.
-
-        Supports 'type' string, integer, number, boolean, object (flat), array (of simple types),
-        required fields, and default values. Nested objects/arrays can be extended later.
-        """
+        """Create a Pydantic model dynamically from a JSON Schema subset."""
         from pydantic import create_model
 
         def py_type(js: Dict[str, Any]):
             t = js.get("type")
             if isinstance(t, list):
-                # handle ["null", "string"] => Optional[str]
                 non_null = [x for x in t if x != "null"]
                 if not non_null:
                     return Optional[Any]
@@ -557,13 +725,12 @@ class QueryService(QueryServiceInterface):
                 items = js.get("items", {"type": "string"})
                 return List[py_type(items)]
             if t == "object":
-                # For now, represent as Dict[str, Any]
                 return Dict[str, Any]
             return Any
 
         properties: Dict[str, Any] = schema.get("properties", {})
         required = set(schema.get("required", []))
-        fields = {}
+        fields: Dict[str, Any] = {}
         for field_name, field_schema in properties.items():
             typ = py_type(field_schema)
             default = field_schema.get("default")

solana_agent/services/routing.py

@@ -1,8 +1,9 @@
 """
 Routing service implementation.
 
-This service manages query routing to appropriate agents based on
-specializations and query analysis.
+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
@@ -28,7 +29,7 @@ class RoutingService(RoutingServiceInterface):
         api_key: Optional[str] = None,
         base_url: Optional[str] = None,
         model: Optional[str] = None,
-    ):
+    ) -> None:
         """Initialize the routing service.
 
         Args:
@@ -39,7 +40,10 @@ class RoutingService(RoutingServiceInterface):
         self.agent_service = agent_service
         self.api_key = api_key
         self.base_url = base_url
-        self.model = model
+        # Use a small, cheap model for routing unless explicitly provided
+        self.model = model or "gpt-4.1-mini"
+        # Simple sticky session: remember last routed agent in-process
+        self._last_agent = None
 
     async def _analyze_query(self, query: str) -> Dict[str, Any]:
         """Analyze a query to determine routing information.
@@ -82,9 +86,6 @@ class RoutingService(RoutingServiceInterface):
         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
-
-        Think carefully about whether the query is more technical/development-focused or more
-        financial/market-focused to match with the appropriate agent.
         """
 
         try:
@@ -131,18 +132,23 @@ class RoutingService(RoutingServiceInterface):
         if len(agents) == 1:
             agent_name = next(iter(agents.keys()))
             logger.info(f"Only one agent available: {agent_name}")  # Use logger.info
+            self._last_agent = agent_name
             return agent_name
 
-        # Analyze query
-        analysis = await self._analyze_query(query)
+        # Short reply bypass and default stickiness
+        short = query.strip().lower()
+        short_replies = {"", "yes", "no", "ok", "k", "y", "n", "1", "0"}
+        if short in short_replies and self._last_agent:
+            return self._last_agent
 
-        # Find best agent based on analysis
+        # Always analyze with a small model to select the best agent
+        analysis = await self._analyze_query(query)
         best_agent = await self._find_best_ai_agent(
             analysis["primary_specialization"], analysis["secondary_specializations"]
         )
-
-        # Return best agent
-        return best_agent
+        chosen = best_agent or next(iter(agents.keys()))
+        self._last_agent = chosen
+        return chosen
 
     async def _find_best_ai_agent(
         self,

{solana_agent-31.1.0.dist-info → solana_agent-31.1.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: solana-agent
-Version: 31.1.0
+Version: 31.1.1
 Summary: AI Agents for Solana
 License: MIT
 Keywords: solana,solana ai,solana agent,ai,ai agent,ai agents
@@ -14,11 +14,11 @@ 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: instructor (==1.10.0)
+Requires-Dist: instructor (==1.11.2)
 Requires-Dist: llama-index-core (==0.13.3)
 Requires-Dist: llama-index-embeddings-openai (==0.5.0)
-Requires-Dist: logfire (==4.3.5)
-Requires-Dist: openai (==1.101.0)
+Requires-Dist: logfire (==4.3.6)
+Requires-Dist: openai (==1.102.0)
 Requires-Dist: pillow (==11.3.0)
 Requires-Dist: pinecone[asyncio] (==7.3.0)
 Requires-Dist: pydantic (>=2)
@@ -26,8 +26,8 @@ Requires-Dist: pymongo (==4.14.1)
 Requires-Dist: pypdf (==6.0.0)
 Requires-Dist: rich (>=13,<14.0)
 Requires-Dist: scrubadub (==2.0.1)
-Requires-Dist: typer (==0.16.1)
-Requires-Dist: zep-cloud (==3.4.1)
+Requires-Dist: typer (==0.17.3)
+Requires-Dist: zep-cloud (==3.4.3)
 Project-URL: Documentation, https://docs.solana-agent.com
 Project-URL: Homepage, https://solana-agent.com
 Project-URL: Repository, https://github.com/truemagic-coder/solana-agent

{solana_agent-31.1.0.dist-info → solana_agent-31.1.1.dist-info}/RECORD

@@ -10,7 +10,7 @@ solana_agent/domains/__init__.py,sha256=HiC94wVPRy-QDJSSRywCRrhrFfTBeHjfi5z-QfZv
 solana_agent/domains/agent.py,sha256=8pAi1-kIgzFNANt3dyQjw-1zbThcNdpEllbAGWi79uI,2841
 solana_agent/domains/routing.py,sha256=1yR4IswGcmREGgbOOI6TKCfuM7gYGOhQjLkBqnZ-rNo,582
 solana_agent/factories/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-solana_agent/factories/agent_factory.py,sha256=_C6YO3rtfa20i09Wtrwy9go8DZGGe6r0Ko_r3hVfqRU,14379
+solana_agent/factories/agent_factory.py,sha256=i20C57hxF7juWuYy9lru28OwCC-xD4g5UhzmQtPe474,14671
 solana_agent/guardrails/pii.py,sha256=FCz1IC3mmkr41QFFf5NaC0fwJrVkwFsxgyOCS2POO5I,4428
 solana_agent/interfaces/__init__.py,sha256=IQs1WIM1FeKP1-kY2FEfyhol_dB-I-VAe2rD6jrVF6k,355
 solana_agent/interfaces/client/client.py,sha256=9hg35-hp_CI-WVGOXehBE1ZCKYahLmbeAvtQOYmML4o,3245
@@ -34,10 +34,10 @@ solana_agent/repositories/memory.py,sha256=F46vZ-Uhj7PX2uFGCRKYsZ8JLmKteMN1d30qG
 solana_agent/services/__init__.py,sha256=iko0c2MlF8b_SA_nuBGFllr2E3g_JowOrOzGcnU9tkA,162
 solana_agent/services/agent.py,sha256=dotuINMtW3TQDLq2eNM5r1cAUwhzxbHBotw8p5CLsYU,20983
 solana_agent/services/knowledge_base.py,sha256=ZvOPrSmcNDgUzz4bJIQ4LeRl9vMZiK9hOfs71IpB7Bk,32735
-solana_agent/services/query.py,sha256=sqWpan0qVrRm0MRxa_14ViuUTP2xsvxbaOl8EPgWJbg,23894
-solana_agent/services/routing.py,sha256=C5Ku4t9TqvY7S8wlUPMTC04HCrT4Ib3E8Q8yX0lVU_s,7137
-solana_agent-31.1.0.dist-info/LICENSE,sha256=BnSRc-NSFuyF2s496l_4EyrwAP6YimvxWcjPiJ0J7g4,1057
-solana_agent-31.1.0.dist-info/METADATA,sha256=GgLLBonDRWgNcbWYU0fRtgu_or5UrbRaei8lAEsEDS4,30013
-solana_agent-31.1.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-solana_agent-31.1.0.dist-info/entry_points.txt,sha256=-AuT_mfqk8dlZ0pHuAjx1ouAWpTRjpqvEUa6YV3lmc0,53
-solana_agent-31.1.0.dist-info/RECORD,,
+solana_agent/services/query.py,sha256=FwqZMmuLUI7TmAkzv91D363PfnWajww35s8j5ubatY4,31544
+solana_agent/services/routing.py,sha256=hsHe8HSGO_xFc0A17WIOGTidLTfLSfApQw3l2HHqkLo,7614
+solana_agent-31.1.1.dist-info/LICENSE,sha256=BnSRc-NSFuyF2s496l_4EyrwAP6YimvxWcjPiJ0J7g4,1057
+solana_agent-31.1.1.dist-info/METADATA,sha256=Ke1JR65TFAZ3YAE4V_x-WbVXvMfkF-2aj_Hs2xlgcxQ,30013
+solana_agent-31.1.1.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+solana_agent-31.1.1.dist-info/entry_points.txt,sha256=-AuT_mfqk8dlZ0pHuAjx1ouAWpTRjpqvEUa6YV3lmc0,53
+solana_agent-31.1.1.dist-info/RECORD,,