Update tools/orchestrator.py

tools/orchestrator.py

@@ -3,6 +3,7 @@ import json
 import re
 import logging
 import time
+from huggingface_hub import whoami
 
 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
@@ -64,22 +65,34 @@ def orchestrate_and_respond(user_input: str, provider_name: str, model_display_n
     context_str = None
 
     if action_type == "create_huggingface_space":
-        params = ["owner", "space_name", "sdk", "description"]
+        params = ["space_name", "sdk", "description"]
         if all(p in action_input for p in params):
-            yield "status", "[Tool: Generating space content...]"
-            space_gen_messages = [
-                {"role": "system", "content": prompts.SPACE_GENERATION_SYSTEM_PROMPT},
-                {"role": "user", "content": prompts.get_space_generation_user_prompt(**action_input)}
-            ]
-            markdown_content = "".join(list(call_model_stream(provider_name, model_display_name, space_gen_messages, ui_api_key_override, 0.1, 4096)))
-            yield "status", "[Tool: Creating Space...]"
-            result = create_huggingface_space(
-                owner=action_input["owner"],
-                space_name=action_input["space_name"],
-                sdk=action_input["sdk"],
-                markdown_content=markdown_content.strip()
-            )
-            context_str = f"Tool Result (Create Space): {result.get('result') or result.get('error', 'Unknown outcome')}"
+            try:
+                hf_token = os.getenv("HF_TOKEN")
+                if not hf_token: raise ValueError("Hugging Face token (HF_TOKEN) not found.")
+                
+                yield "status", "[Tool: Verifying user identity...]"
+                user_info = whoami(token=hf_token)
+                owner = user_info.get("name")
+                if not owner: raise ValueError("Could not determine owner from HF token.")
+                action_input["owner"] = owner
+                
+                yield "status", "[Tool: Generating space content...]"
+                space_gen_messages = [
+                    {"role": "system", "content": prompts.SPACE_GENERATION_SYSTEM_PROMPT},
+                    {"role": "user", "content": prompts.get_space_generation_user_prompt(**action_input)}
+                ]
+                markdown_content = "".join(list(call_model_stream(provider_name, model_display_name, space_gen_messages, ui_api_key_override, 0.1, 4096)))
+                yield "status", "[Tool: Creating Space...]"
+                result = create_huggingface_space(
+                    owner=owner,
+                    space_name=action_input["space_name"],
+                    sdk=action_input["sdk"],
+                    markdown_content=markdown_content.strip()
+                )
+                context_str = f"Tool Result (Create Space): {result.get('result') or result.get('error', 'Unknown outcome')}"
+            except Exception as e:
+                context_str = f"Tool Failed: An error occurred during space creation process - {e}"
         else:
             context_str = "Tool Failed: Missing parameters for create_huggingface_space. Required: " + ", ".join(params)
     elif action_type == "list_huggingface_space_files":