Fix tool_calls calling wiki_search without groups

routers/embedding/embeddings_issues.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:7c3c012a8f86440dacedd6f1e4e9ea9f41f096031c0ac1ed5cdf64a9a8d46e42
-size 723452942
+oid sha256:5c790700eeb373e6f80e8a52e3ac3eafd7591f2782a1fcb589a197769e25dfc7
+size 723465376

routers/tool_calls.py

@@ -5,10 +5,10 @@ from typing import List, Dict
 from pydantic import BaseModel
 
 try:
-    from .tool_gpu_checker import gpu_checker_get_message
-    from .tool_bpy_doc import bpy_doc_get_documentation
-    from .tool_find_related import find_related
-    from .tool_wiki_search import wiki_search
+    from tool_gpu_checker import gpu_checker_get_message
+    from tool_bpy_doc import bpy_doc_get_documentation
+    from tool_find_related import find_related
+    from tool_wiki_search import wiki_search
 except:
     from routers.tool_gpu_checker import gpu_checker_get_message
     from routers.tool_bpy_doc import bpy_doc_get_documentation
@@ -46,7 +46,7 @@ def process_tool_call(tool_call: ToolCallInput) -> Dict:
             output["output"] = find_related(
                 function_args["repo"], function_args["number"])
         elif function_name == "wiki_search":
-            output["output"] = wiki_search(function_args["query"])
+            output["output"] = wiki_search(**function_args)
     except json.JSONDecodeError as e:
         error_message = f"Malformed JSON encountered at position {e.pos}: {e.msg}\n {tool_call.function.arguments}"
         output["output"] = error_message
@@ -95,6 +95,14 @@ if __name__ == "__main__":
                 "name": "find_related",
                 "arguments": "{\"repo\":\"blender\",\"number\":111434}"
             }
+        },
+        {
+            "id": "call_abc101112",
+            "type": "function",
+            "function": {
+                "name": "wiki_search",
+                "arguments": "{\"query\":\"Set Snap Base\",\"groups\":[\"manual\"]}"
+            }
         }
     ]
 

routers/tool_wiki_search.py

@@ -6,7 +6,7 @@ import pickle
 import re
 import torch
 from enum import Enum
-from fastapi import APIRouter, Query
+from fastapi import APIRouter, Query, params
 from fastapi.responses import PlainTextResponse
 from heapq import nlargest
 from sentence_transformers import util
@@ -177,6 +177,9 @@ def wiki_search(
     query: str = "",
     groups: Set[Group] = Query(default={Group.dev_docs, Group.manual})
 ) -> str:
+    if isinstance(groups, params.Query):
+        groups = groups.default
+
     texts = G_data._sort_similarity(query, groups)
     result: str = ''
     for text in texts:
@@ -187,5 +190,5 @@ def wiki_search(
 if __name__ == '__main__':
     tests = ["Set Snap Base", "Building the Manual",
              "Bisect Object", "Who are the Triagers", "4.3 Release Notes Motion Paths"]
-    result = wiki_search(tests[0], {Group.dev_docs, Group.manual})
+    result = wiki_search(tests[0])
     print(result)