Update prompts.py

prompts.py

@@ -7,7 +7,13 @@ DEFAULT_SYSTEM_PROMPT = os.getenv(
 )
 
 METRICS_SYSTEM_PROMPT = "You are a precise JSON output agent. Output a single JSON object containing interaction metrics as requested by the user. Do not include any explanatory text before or after the JSON object."
-TOOL_SYSTEM_PROMPT = "You are a precise routing agent... Output JSON only. Example: {\"action\": \"search_duckduckgo_and_report\", \"action_input\": {\"search_engine_query\": \"query\"}}"
+
+TOOL_SYSTEM_PROMPT = """You are a precise routing agent. Your task is to select the most appropriate action to respond to a user's query.
+You must output a single JSON object. Do not include any explanatory text.
+Example for web search: {"action": "search_duckduckgo_and_report", "action_input": {"search_engine_query": "latest AI news"}}
+Example for building a space: {"action": "build_huggingface_space", "action_input": {"build_prompt": "Create a Gradio app that translates English to French."}}
+"""
+
 INSIGHT_SYSTEM_PROMPT = """You are an expert AI knowledge base curator. Your primary function is to meticulously analyze an interaction and update the AI's guiding principles (insights/rules) to improve its future performance and self-understanding.
 **CRITICAL OUTPUT REQUIREMENT: You MUST output a single, valid XML structure representing a list of operation objects.**
 The root element should be `<operations_list>`. Each operation should be an `<operation>` element.
@@ -27,7 +33,19 @@ def get_metrics_user_prompt(user_input: str, bot_response: str) -> str:
     return f"User: \"{user_input}\"\nAI: \"{bot_response}\"\nMetrics: \"takeaway\" (3-7 words), \"response_success_score\" (0.0-1.0), \"future_confidence_score\" (0.0-1.0). Output JSON ONLY, ensure it's a single, valid JSON object."
 
 def get_tool_user_prompt(user_input: str, history_snippet: str, guideline_snippet: str) -> str:
-    return f"User Query: \"{user_input}\nRecent History:\n{history_snippet}\nGuidelines: {guideline_snippet}...\nAvailable Actions: quick_respond, answer_using_conversation_memory, search_duckduckgo_and_report, scrape_url_and_report.\nSelect one action and input. Output JSON."
+    available_actions = [
+        "quick_respond",
+        "answer_using_conversation_memory",
+        "search_duckduckgo_and_report",
+        "scrape_url_and_report",
+        "build_huggingface_space"
+    ]
+    return f"""User Query: "{user_input}"
+Recent History:
+{history_snippet}
+Guidelines: {guideline_snippet}...
+Available Actions: {', '.join(available_actions)}.
+Select one action and the necessary input. If the user wants to build, create, or make a Hugging Face Space, use the 'build_huggingface_space' action. Output JSON ONLY."""
 
 def get_final_response_prompt(history_str: str, insights_str: str, user_input: str, context_str: str = None) -> str:
     base = f"History:\n{history_str}\n\nGuidelines:\n{insights_str}"