solana-agent 27.3.6__py3-none-any.whl → 27.3.8__py3-none-any.whl

solana_agent/services/agent.py

@@ -4,8 +4,8 @@ Agent service implementation.
 This service manages AI and human agents, their registration, tool assignments,
 and response generation.
 """
+
 import asyncio
-from copy import deepcopy
 import datetime as main_datetime
 from datetime import datetime
 import json
@@ -53,7 +53,10 @@ class AgentService(AgentServiceInterface):
         )
 
     def register_ai_agent(
-        self, name: str, instructions: str, specialization: str,
+        self,
+        name: str,
+        instructions: str,
+        specialization: str,
     ) -> None:
         """Register an AI agent with its specialization.
 
@@ -95,16 +98,19 @@ class AgentService(AgentServiceInterface):
             system_prompt += f"\n\nVOICE OF THE BRAND:\n{self.business_mission.voice}"
 
             if self.business_mission.values:
-                values_text = "\n".join([
-                    f"- {value.get('name', '')}: {value.get('description', '')}"
-                    for value in self.business_mission.values
-                ])
+                values_text = "\n".join(
+                    [
+                        f"- {value.get('name', '')}: {value.get('description', '')}"
+                        for value in self.business_mission.values
+                    ]
+                )
                 system_prompt += f"\n\nBUSINESS VALUES:\n{values_text}"
 
             # Add goals if available
             if self.business_mission.goals:
                 goals_text = "\n".join(
-                    [f"- {goal}" for goal in self.business_mission.goals])
+                    [f"- {goal}" for goal in self.business_mission.goals]
+                )
                 system_prompt += f"\n\nBUSINESS GOALS:\n{goals_text}"
 
         return system_prompt
@@ -140,7 +146,9 @@ class AgentService(AgentServiceInterface):
         """
         return self.tool_registry.get_agent_tools(agent_name)
 
-    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."""
 
         if not self.tool_registry:
@@ -156,7 +164,7 @@ class AgentService(AgentServiceInterface):
         if not any(t.get("name") == tool_name for t in agent_tools):
             return {
                 "status": "error",
-                "message": f"Agent '{agent_name}' doesn't have access to tool '{tool_name}'"
+                "message": f"Agent '{agent_name}' doesn't have access to tool '{tool_name}'",
             }
 
         try:
@@ -164,6 +172,7 @@ class AgentService(AgentServiceInterface):
             return result
         except Exception as e:
             import traceback
+
             print(traceback.format_exc())
             return {"status": "error", "message": f"Error executing tool: {str(e)}"}
 
@@ -174,11 +183,22 @@ class AgentService(AgentServiceInterface):
         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]:  # pragma: no cover
         """Generate a response with support for text/audio input/output."""
@@ -186,8 +206,12 @@ class AgentService(AgentServiceInterface):
         if not agent:
             error_msg = f"Agent '{agent_name}' not found."
             if output_format == "audio":
-                async for chunk in self.llm_provider.tts(error_msg, instructions=audio_instructions,
-                                                         response_format=audio_output_format, voice=audio_voice):
+                async for chunk in self.llm_provider.tts(
+                    error_msg,
+                    instructions=audio_instructions,
+                    response_format=audio_output_format,
+                    voice=audio_voice,
+                ):
                     yield chunk
             else:
                 yield error_msg
@@ -200,30 +224,33 @@ class AgentService(AgentServiceInterface):
             # --- 2. Add Tool Usage Instructions EARLY ---
             tool_usage_prompt_text = ""
             if self.tool_registry:
-                tool_usage_prompt_text = self._get_tool_usage_prompt(
-                    agent_name)
+                tool_usage_prompt_text = self._get_tool_usage_prompt(agent_name)
                 if tool_usage_prompt_text:
                     system_prompt_parts.append(
-                        f"\n\n--- TOOL USAGE INSTRUCTIONS ---{tool_usage_prompt_text}")
+                        f"\n\n--- TOOL USAGE INSTRUCTIONS ---{tool_usage_prompt_text}"
+                    )
                     print(
-                        f"Tools available to agent {agent_name}: {[t.get('name') for t in self.get_agent_tools(agent_name)]}")
+                        f"Tools available to agent {agent_name}: {[t.get('name') for t in self.get_agent_tools(agent_name)]}"
+                    )
 
             # --- 3. Add User ID ---
-            system_prompt_parts.append(f"\n\n--- USER & SESSION INFO ---")
+            system_prompt_parts.append("\n\n--- USER & SESSION INFO ---")
             system_prompt_parts.append(f"User ID: {user_id}")
 
             # --- 4. Add Memory Context ---
             if memory_context:
                 # Make the header clearly separate it
                 system_prompt_parts.append(
-                    f"\n\n--- CONVERSATION HISTORY (Memory Context) ---")
+                    "\n\n--- CONVERSATION HISTORY (Memory Context) ---"
+                )
                 system_prompt_parts.append(memory_context)
 
             # --- 5. Add Additional Prompt (if provided) ---
             if prompt:
                 # Make the header clearly separate it
                 system_prompt_parts.append(
-                    f"\n\n--- ADDITIONAL INSTRUCTIONS FOR THIS TURN ---")
+                    "\n\n--- ADDITIONAL INSTRUCTIONS FOR THIS TURN ---"
+                )
                 system_prompt_parts.append(prompt)
 
             # --- Assemble the final system prompt ---
@@ -237,15 +264,13 @@ class AgentService(AgentServiceInterface):
             tool_buffer = ""
             pending_chunk = ""  # To hold text that might contain partial markers
             is_tool_call = False
-            window_size = 30  # Increased window size for better detection
 
             # Define start and end markers
             start_marker = "[TOOL]"
             end_marker = "[/TOOL]"
 
             # Generate and stream response (ALWAYS use non-realtime for text generation)
-            print(
-                f"Generating response with {len(query)} characters of query text")
+            print(f"Generating response with {len(query)} characters of query text")
             async for chunk in self.llm_provider.generate_text(
                 prompt=query,
                 system_prompt=final_system_prompt,
@@ -263,7 +288,8 @@ class AgentService(AgentServiceInterface):
                 # STEP 1: Check for tool call start marker
                 if start_marker in combined_chunk and not is_tool_call:
                     print(
-                        f"Found tool start marker in chunk of length {len(combined_chunk)}")
+                        f"Found tool start marker in chunk of length {len(combined_chunk)}"
+                    )
                     is_tool_call = True
 
                     # Extract text before the marker and the marker itself with everything after
@@ -285,20 +311,18 @@ class AgentService(AgentServiceInterface):
 
                     # Check if the tool call is complete
                     if end_marker in tool_buffer:
-                        print(
-                            f"Tool call complete, buffer size: {len(tool_buffer)}")
+                        print(f"Tool call complete, buffer size: {len(tool_buffer)}")
 
                         # Process the tool call
                         response_text = await self._handle_tool_call(
-                            agent_name=agent_name,
-                            tool_text=tool_buffer
+                            agent_name=agent_name, tool_text=tool_buffer
                         )
 
                         # Clean the response to remove any markers or formatting
-                        response_text = self._clean_tool_response(
-                            response_text)
+                        response_text = self._clean_tool_response(response_text)
                         print(
-                            f"Tool execution complete, result size: {len(response_text)}")
+                            f"Tool execution complete, result size: {len(response_text)}"
+                        )
 
                         # Create new prompt with search/tool results
                         # Ensure query is string
@@ -307,36 +331,42 @@ class AgentService(AgentServiceInterface):
                         # --- REBUILD the system prompt for the follow-up call ---
                         # Start with base prompt again
                         follow_up_system_prompt_parts = [
-                            self.get_agent_system_prompt(agent_name)]
+                            self.get_agent_system_prompt(agent_name)
+                        ]
                         # Add the instruction NOT to use tools again
                         follow_up_system_prompt_parts.append(
-                            "\n\nCRITICAL: You have received the results from a tool. Base your response on the 'Search Result' provided in the user prompt. DO NOT use the tool calling format again for this turn.")
-                        follow_up_system_prompt_parts.append(
-                            f"\n\n--- USER & SESSION INFO ---")
+                            "\n\nCRITICAL: You have received the results from a tool. Base your response on the 'Search Result' provided in the user prompt. DO NOT use the tool calling format again for this turn."
+                        )
                         follow_up_system_prompt_parts.append(
-                            f"User ID: {user_id}")
+                            "\n\n--- USER & SESSION INFO ---"
+                        )
+                        follow_up_system_prompt_parts.append(f"User ID: {user_id}")
                         if memory_context:
                             # Make the header clearly separate it
                             follow_up_system_prompt_parts.append(
-                                f"\n\n--- CONVERSATION HISTORY (Memory Context) ---")
-                            follow_up_system_prompt_parts.append(
-                                memory_context)
+                                "\n\n--- CONVERSATION HISTORY (Memory Context) ---"
+                            )
+                            follow_up_system_prompt_parts.append(memory_context)
                         if prompt:
                             # Make the header clearly separate it
                             follow_up_system_prompt_parts.append(
-                                f"\n\n--- ADDITIONAL INSTRUCTIONS FOR THIS TURN ---")
+                                "\n\n--- ADDITIONAL INSTRUCTIONS FOR THIS TURN ---"
+                            )
                             follow_up_system_prompt_parts.append(prompt)
 
                         # --- Assemble the final follow_up prompt ---
                         final_follow_up_system_prompt = "\n".join(
-                            follow_up_system_prompt_parts)
+                            follow_up_system_prompt_parts
+                        )
                         # --- End Rebuild ---"
 
                         # Generate a new response with the tool results
                         print("Generating new response with tool results")
                         if output_format == "text":
                             # Stream the follow-up response for text output
-                            async for processed_chunk in self.llm_provider.generate_text(
+                            async for (
+                                processed_chunk
+                            ) in self.llm_provider.generate_text(
                                 prompt=user_prompt,
                                 system_prompt=final_follow_up_system_prompt,
                                 api_key=self.api_key,
@@ -348,15 +378,16 @@ class AgentService(AgentServiceInterface):
                         else:
                             # For audio output, collect the full response first
                             tool_response = ""
-                            async for processed_chunk in self.llm_provider.generate_text(
+                            async for (
+                                processed_chunk
+                            ) in self.llm_provider.generate_text(
                                 prompt=user_prompt,
                                 system_prompt=final_follow_up_system_prompt,
                             ):
                                 tool_response += processed_chunk
 
                             # Clean and add to our complete text record and audio buffer
-                            tool_response = self._clean_for_audio(
-                                tool_response)
+                            tool_response = self._clean_for_audio(tool_response)
                             complete_text_response += tool_response
                             full_response_buffer += tool_response
 
@@ -380,8 +411,7 @@ class AgentService(AgentServiceInterface):
                         # Everything except the partial marker
                         chunk_to_yield = combined_chunk[:-i]
                         potential_marker = True
-                        print(
-                            f"Potential partial marker detected: '{pending_chunk}'")
+                        print(f"Potential partial marker detected: '{pending_chunk}'")
                         break
 
                 if potential_marker:
@@ -405,7 +435,8 @@ class AgentService(AgentServiceInterface):
             # Process any incomplete tool call as regular text
             if is_tool_call and tool_buffer:
                 print(
-                    f"Incomplete tool call detected, returning as regular text: {len(tool_buffer)} chars")
+                    f"Incomplete tool call detected, returning as regular text: {len(tool_buffer)} chars"
+                )
                 if output_format == "text":
                     yield tool_buffer
 
@@ -417,28 +448,28 @@ class AgentService(AgentServiceInterface):
             if output_format == "audio" and full_response_buffer:
                 # Clean text before TTS
                 print(
-                    f"Processing {len(full_response_buffer)} characters for audio output")
-                full_response_buffer = self._clean_for_audio(
-                    full_response_buffer)
+                    f"Processing {len(full_response_buffer)} characters for audio output"
+                )
+                full_response_buffer = self._clean_for_audio(full_response_buffer)
 
                 # Process the entire response with TTS
                 async for audio_chunk in self.llm_provider.tts(
                     text=full_response_buffer,
                     voice=audio_voice,
                     response_format=audio_output_format,
-                    instructions=audio_instructions
+                    instructions=audio_instructions,
                 ):
                     yield audio_chunk
 
             # Store the complete text response
             self.last_text_response = complete_text_response
-            print(
-                f"Response generation complete: {len(complete_text_response)} chars")
+            print(f"Response generation complete: {len(complete_text_response)} chars")
 
         except Exception as e:
             error_msg = f"I apologize, but I encountered an error: {str(e)}"
             print(f"Error in generate_response: {str(e)}")
             import traceback
+
             print(traceback.format_exc())
 
             if output_format == "audio":
@@ -446,7 +477,7 @@ class AgentService(AgentServiceInterface):
                     error_msg,
                     voice=audio_voice,
                     response_format=audio_output_format,
-                    instructions=audio_instructions
+                    instructions=audio_instructions,
                 ):
                     yield chunk
             else:
@@ -465,7 +496,7 @@ class AgentService(AgentServiceInterface):
         chunk_size = 4096
 
         for i in range(0, len(data), chunk_size):
-            yield data[i:i + chunk_size]
+            yield data[i : i + chunk_size]
             # Small delay to simulate streaming
             await asyncio.sleep(0.01)
 
@@ -514,6 +545,7 @@ class AgentService(AgentServiceInterface):
 
         except Exception as e:
             import traceback
+
             print(traceback.format_exc())
             return f"Error processing tool call: {str(e)}"
 
@@ -524,8 +556,6 @@ class AgentService(AgentServiceInterface):
         if not tools:
             return ""
 
-        # Get actual tool names
-        available_tool_names = [tool.get("name", "") for tool in tools]
         tools_json = json.dumps(tools, indent=2)
 
         return f"""
@@ -578,62 +608,62 @@ class AgentService(AgentServiceInterface):
             return ""
 
         # Remove Markdown links - [text](url) -> text
-        text = re.sub(r'\[([^\]]+)\]\([^\)]+\)', r'\1', text)
+        text = re.sub(r"\[([^\]]+)\]\([^\)]+\)", r"\1", text)
 
         # Remove inline code with backticks
-        text = re.sub(r'`([^`]+)`', r'\1', text)
+        text = re.sub(r"`([^`]+)`", r"\1", text)
 
         # Remove bold formatting - **text** or __text__ -> text
-        text = re.sub(r'(\*\*|__)(.*?)\1', r'\2', text)
+        text = re.sub(r"(\*\*|__)(.*?)\1", r"\2", text)
 
         # Remove italic formatting - *text* or _text_ -> text
-        text = re.sub(r'(\*|_)(.*?)\1', r'\2', text)
+        text = re.sub(r"(\*|_)(.*?)\1", r"\2", text)
 
         # Remove headers - ## Header -> Header
-        text = re.sub(r'^\s*#+\s*(.*?)$', r'\1', text, flags=re.MULTILINE)
+        text = re.sub(r"^\s*#+\s*(.*?)$", r"\1", text, flags=re.MULTILINE)
 
         # Remove blockquotes - > Text -> Text
-        text = re.sub(r'^\s*>\s*(.*?)$', r'\1', text, flags=re.MULTILINE)
+        text = re.sub(r"^\s*>\s*(.*?)$", r"\1", text, flags=re.MULTILINE)
 
         # Remove horizontal rules (---, ***, ___)
-        text = re.sub(r'^\s*[-*_]{3,}\s*$', '', text, flags=re.MULTILINE)
+        text = re.sub(r"^\s*[-*_]{3,}\s*$", "", text, flags=re.MULTILINE)
 
         # Remove list markers - * Item or - Item or 1. Item -> Item
-        text = re.sub(r'^\s*[-*+]\s+(.*?)$', r'\1', text, flags=re.MULTILINE)
-        text = re.sub(r'^\s*\d+\.\s+(.*?)$', r'\1', text, flags=re.MULTILINE)
+        text = re.sub(r"^\s*[-*+]\s+(.*?)$", r"\1", text, flags=re.MULTILINE)
+        text = re.sub(r"^\s*\d+\.\s+(.*?)$", r"\1", text, flags=re.MULTILINE)
 
         # Remove multiple consecutive newlines (keep just one)
-        text = re.sub(r'\n{3,}', '\n\n', text)
+        text = re.sub(r"\n{3,}", "\n\n", text)
 
         # Remove emojis and other non-pronounceable characters
         # Common emoji Unicode ranges
         emoji_pattern = re.compile(
             "["
-            "\U0001F600-\U0001F64F"  # emoticons
-            "\U0001F300-\U0001F5FF"  # symbols & pictographs
-            "\U0001F680-\U0001F6FF"  # transport & map symbols
-            "\U0001F700-\U0001F77F"  # alchemical symbols
-            "\U0001F780-\U0001F7FF"  # Geometric Shapes
-            "\U0001F800-\U0001F8FF"  # Supplemental Arrows-C
-            "\U0001F900-\U0001F9FF"  # Supplemental Symbols and Pictographs
-            "\U0001FA00-\U0001FA6F"  # Chess Symbols
-            "\U0001FA70-\U0001FAFF"  # Symbols and Pictographs Extended-A
-            "\U00002702-\U000027B0"  # Dingbats
-            "\U000024C2-\U0000257F"  # Enclosed characters
-            "\U00002600-\U000026FF"  # Miscellaneous Symbols
-            "\U00002700-\U000027BF"  # Dingbats
-            "\U0000FE00-\U0000FE0F"  # Variation Selectors
-            "\U0001F1E0-\U0001F1FF"  # Flags (iOS)
+            "\U0001f600-\U0001f64f"  # emoticons
+            "\U0001f300-\U0001f5ff"  # symbols & pictographs
+            "\U0001f680-\U0001f6ff"  # transport & map symbols
+            "\U0001f700-\U0001f77f"  # alchemical symbols
+            "\U0001f780-\U0001f7ff"  # Geometric Shapes
+            "\U0001f800-\U0001f8ff"  # Supplemental Arrows-C
+            "\U0001f900-\U0001f9ff"  # Supplemental Symbols and Pictographs
+            "\U0001fa00-\U0001fa6f"  # Chess Symbols
+            "\U0001fa70-\U0001faff"  # Symbols and Pictographs Extended-A
+            "\U00002702-\U000027b0"  # Dingbats
+            "\U000024c2-\U0000257f"  # Enclosed characters
+            "\U00002600-\U000026ff"  # Miscellaneous Symbols
+            "\U00002700-\U000027bf"  # Dingbats
+            "\U0000fe00-\U0000fe0f"  # Variation Selectors
+            "\U0001f1e0-\U0001f1ff"  # Flags (iOS)
             "]+",
-            flags=re.UNICODE
+            flags=re.UNICODE,
         )
-        text = emoji_pattern.sub(r' ', text)
+        text = emoji_pattern.sub(r" ", text)
 
         # Replace special characters that can cause issues with TTS
-        text = re.sub(r'[^\w\s\.\,\;\:\?\!\'\"\-\(\)]', ' ', text)
+        text = re.sub(r"[^\w\s\.\,\;\:\?\!\'\"\-\(\)]", " ", text)
 
         # Replace multiple spaces with a single space
-        text = re.sub(r'\s+', ' ', text)
+        text = re.sub(r"\s+", " ", text)
 
         return text.strip()