broadfield-dev commited on
Commit
bca02f9
·
verified ·
1 Parent(s): cde37a1

Update tools/orchestrator.py

Browse files
Files changed (1) hide show
  1. tools/orchestrator.py +8 -2
tools/orchestrator.py CHANGED
@@ -73,7 +73,12 @@ def orchestrate_and_respond(user_input: str, provider_name: str, model_display_n
73
  ]
74
  markdown_content = "".join(list(call_model_stream(provider_name, model_display_name, space_gen_messages, ui_api_key_override, 0.1, 4096)))
75
  yield "status", "[Tool: Creating Space...]"
76
- result = create_huggingface_space(markdown_content=markdown_content.strip(), **action_input)
 
 
 
 
 
77
  context_str = f"Tool Result (Create Space): {result.get('result') or result.get('error', 'Unknown outcome')}"
78
  else:
79
  context_str = "Tool Failed: Missing parameters for create_huggingface_space. Required: " + ", ".join(params)
@@ -85,7 +90,8 @@ def orchestrate_and_respond(user_input: str, provider_name: str, model_display_n
85
  if "error" in result:
86
  context_str = f"Tool Result (List Files): Error - {result['error']}"
87
  else:
88
- files_str = "\n- ".join(result.get("files", []))
 
89
  context_str = f"Tool Result (List Files):\nStatus: {result.get('status', 'OK')}\nFiles:\n- {files_str}"
90
  else:
91
  context_str = "Tool Failed: Missing parameters for list_huggingface_space_files. Required: " + ", ".join(params)
 
73
  ]
74
  markdown_content = "".join(list(call_model_stream(provider_name, model_display_name, space_gen_messages, ui_api_key_override, 0.1, 4096)))
75
  yield "status", "[Tool: Creating Space...]"
76
+ result = create_huggingface_space(
77
+ owner=action_input["owner"],
78
+ space_name=action_input["space_name"],
79
+ sdk=action_input["sdk"],
80
+ markdown_content=markdown_content.strip()
81
+ )
82
  context_str = f"Tool Result (Create Space): {result.get('result') or result.get('error', 'Unknown outcome')}"
83
  else:
84
  context_str = "Tool Failed: Missing parameters for create_huggingface_space. Required: " + ", ".join(params)
 
90
  if "error" in result:
91
  context_str = f"Tool Result (List Files): Error - {result['error']}"
92
  else:
93
+ files_list = result.get("files", [])
94
+ files_str = "\n- ".join(files_list) if files_list else "No files found."
95
  context_str = f"Tool Result (List Files):\nStatus: {result.get('status', 'OK')}\nFiles:\n- {files_str}"
96
  else:
97
  context_str = "Tool Failed: Missing parameters for list_huggingface_space_files. Required: " + ", ".join(params)