Germano Cavalcante commited on
Commit
66906e4
1 Parent(s): 9a6a74b
Files changed (1) hide show
  1. 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 find_relatedness
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_relatedness
16
- from .tool_wiki_search import wiki_search
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"] = find_relatedness(
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"])