Update tools/orchestrator.py

tools/orchestrator.py

@@ -7,6 +7,7 @@ import time
 from model_logic import call_model_stream, MODELS_BY_PROVIDER, get_default_model_display_name_for_provider
 from memory_logic import retrieve_memories_semantic, retrieve_rules_semantic
 from tools.websearch import search_and_scrape_duckduckgo, scrape_url
+from tools.space_builder import build_huggingface_space
 import prompts
 from utils import format_insights_for_prompt
 
@@ -27,7 +28,7 @@ def decide_on_tool(user_input: str, chat_history_for_prompt: list, initial_insig
     if len(user_input.split()) <= 3 and any(kw in user_input_lower for kw in ["hello", "hi", "thanks", "ok", "bye"]) and not "?" in user_input:
         return "quick_respond", {}
     
-    if len(user_input.split()) > 3 or "?" in user_input or any(w in user_input_lower for w in ["what is", "how to", "explain", "search for"]):
+    if len(user_input.split()) > 3 or "?" in user_input or any(w in user_input_lower for w in ["what is", "how to", "explain", "search for", "build", "create", "make"]):
         history_snippet = "\n".join([f"{msg['role']}: {msg['content'][:100]}" for msg in chat_history_for_prompt[-2:]])
         guideline_snippet = initial_insights_ctx_str[:200].replace('\n', ' ')
         tool_user_prompt = prompts.get_tool_user_prompt(user_input, history_snippet, guideline_snippet)
@@ -37,7 +38,7 @@ def decide_on_tool(user_input: str, chat_history_for_prompt: list, initial_insig
         
         if tool_model_display:
             try:
-                tool_resp_raw = "".join(list(call_model_stream(provider=TOOL_DECISION_PROVIDER, model_display_name=tool_model_display, messages=tool_decision_messages, temperature=0.0, max_tokens=150)))
+                tool_resp_raw = "".join(list(call_model_stream(provider=TOOL_DECISION_PROVIDER, model_display_name=tool_model_display, messages=tool_decision_messages, temperature=0.0, max_tokens=200)))
                 json_match_tool = re.search(r"\{.*\}", tool_resp_raw, re.DOTALL)
                 if json_match_tool:
                     action_data = json.loads(json_match_tool.group(0))
@@ -86,6 +87,22 @@ def orchestrate_and_respond(user_input: str, provider_name: str, model_display_n
             context_str = "Web Content:\n" + "\n".join([f"Source {i+1}:\nURL:{r.get('url','N/A')}\nTitle:{r.get('title','N/A')}\nContent:\n{(r.get('content') or r.get('error') or 'N/A')[:3500]}\n---" for i,r in enumerate(web_results)]) if web_results else f"No results from {action_type} for '{query_or_url}'."
             yield "status", "[Synthesizing web report...]"
             final_system_prompt_str += " Generate report/answer from web content, history, & guidelines. Cite URLs as [Source X]."
+    elif action_type == "build_huggingface_space":
+        build_prompt = action_input_dict.get("build_prompt")
+        if not build_prompt:
+             context_str = "Tool Action Failed: The model decided to build a space but did not provide the necessary instructions (build_prompt)."
+             final_system_prompt_str += " Report the tool failure to the user."
+        else:
+            yield "status", f"[Tool: Building Space '{build_prompt[:50]}'...]"
+            build_result = build_huggingface_space(build_prompt)
+            if "error" in build_result:
+                context_str = f"Hugging Face Space Builder Tool Result:\n- Status: FAILED\n- Error: {build_result['error']}"
+                final_system_prompt_str += " The space building tool failed. Report the error to the user and ask if they want to try again."
+                yield "status", "[Tool: Space building failed.]"
+            else:
+                context_str = f"Hugging Face Space Builder Tool Result:\n- Status: SUCCESS\n- Details: {build_result['result']}"
+                final_system_prompt_str += " The space building tool has completed. Inform the user about the result, providing any links or key information from the tool's output."
+                yield "status", "[Tool: Space building complete.]"
     else: # quick_respond or fallback
         final_system_prompt_str += " Respond directly using guidelines & history."