Spaces:
Running
Running
Germano Cavalcante
commited on
Commit
•
66906e4
1
Parent(s):
9a6a74b
Fixes
Browse files- routers/tool_calls.py +4 -4
routers/tool_calls.py
CHANGED
@@ -7,13 +7,13 @@ from pydantic import BaseModel
|
|
7 |
try:
|
8 |
from .tool_gpu_checker import gpu_checker_get_message
|
9 |
from .tool_bpy_doc import bpy_doc_get_documentation
|
10 |
-
from .tool_find_related import
|
11 |
from .tool_wiki_search import wiki_search
|
12 |
except:
|
13 |
from tool_gpu_checker import gpu_checker_get_message
|
14 |
from tool_bpy_doc import bpy_doc_get_documentation
|
15 |
-
from tool_find_related import
|
16 |
-
from
|
17 |
|
18 |
|
19 |
class ToolCallFunction(BaseModel):
|
@@ -43,7 +43,7 @@ def process_tool_call(tool_call: ToolCallInput) -> Dict:
|
|
43 |
output["output"] = gpu_checker_get_message(
|
44 |
function_args.get("gpu", ""))
|
45 |
elif function_name == "find_related":
|
46 |
-
output["output"] =
|
47 |
function_args["repo"], function_args["number"])
|
48 |
elif function_name == "wiki_search":
|
49 |
output["output"] = wiki_search(function_args["query"])
|
|
|
7 |
try:
|
8 |
from .tool_gpu_checker import gpu_checker_get_message
|
9 |
from .tool_bpy_doc import bpy_doc_get_documentation
|
10 |
+
from .tool_find_related import find_related
|
11 |
from .tool_wiki_search import wiki_search
|
12 |
except:
|
13 |
from tool_gpu_checker import gpu_checker_get_message
|
14 |
from tool_bpy_doc import bpy_doc_get_documentation
|
15 |
+
from tool_find_related import find_related
|
16 |
+
from tool_wiki_search import wiki_search
|
17 |
|
18 |
|
19 |
class ToolCallFunction(BaseModel):
|
|
|
43 |
output["output"] = gpu_checker_get_message(
|
44 |
function_args.get("gpu", ""))
|
45 |
elif function_name == "find_related":
|
46 |
+
output["output"] = find_related(
|
47 |
function_args["repo"], function_args["number"])
|
48 |
elif function_name == "wiki_search":
|
49 |
output["output"] = wiki_search(function_args["query"])
|