Tool Calls: Implement Find Related

routers/tool_calls.py

@@ -1,11 +1,18 @@
 import json
 import logging
-from .tool_gpu_checker import gpu_checker_get_message
-from .tool_bpy_doc import bpy_doc_get_documentation
 from fastapi import APIRouter, Body
 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_relatedness
+except:
+    from tool_gpu_checker import gpu_checker_get_message
+    from tool_bpy_doc import bpy_doc_get_documentation
+    from tool_find_related import find_relatedness
+
 
 class ToolCallFunction(BaseModel):
     name: str
@@ -33,6 +40,9 @@ def process_tool_call(tool_call: ToolCallInput) -> Dict:
         elif function_name == "check_gpu":
             output["output"] = gpu_checker_get_message(
                 function_args.get("gpu", ""))
+        elif function_name == "find_related":
+            output["output"] = find_relatedness(
+                function_args["repo"], function_args["number"])
     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
@@ -73,6 +83,14 @@ if __name__ == "__main__":
                 "name": "check_gpu",
                 "arguments": "{\"gpu\":\"Mesa Intel(R) Iris(R) Plus Graphics 640 (Kaby Lake GT3e) (KBL GT3) Intel 4.6 (Core Profile) Mesa 22.2.5\"}"
             }
+        },
+        {
+            "id": "call_abc789",
+            "type": "function",
+            "function": {
+                "name": "find_related",
+                "arguments": "{\"repo\":\"blender\",\"number\":111434}"
+            }
         }
     ]
 

routers/tool_find_related.py

@@ -14,6 +14,7 @@ try:
 except:
     from utils_gitea import gitea_fetch_issues, gitea_json_issue_get
 
+
 def _create_issue_string(title, body):
     cleaned_body = body.replace('\r', '')
     cleaned_body = cleaned_body.replace('**System Information**\n', '')
@@ -364,7 +365,8 @@ def text_search(owner, repo, text_to_embed, limit=None):
     return result[:limit]
 
 
-def find_relatedness(gitea_issue, limit=20):
+def find_relatedness(repo, number, limit=20):
+    gitea_issue = gitea_json_issue_get('blender', repo, number)
     assert gitea_issue['repository']['owner'] == 'blender'
     repo = gitea_issue['repository']['name']
     title = gitea_issue['title']
@@ -399,9 +401,8 @@ def find_relatedness(gitea_issue, limit=20):
 
 
 @router.get("/find_related/{repo}/{number}")
-def find_related(repo: str = 'blender', number: int = 104399, limit: int = 50):
-    issue = gitea_json_issue_get('blender', repo, number)
-    related = find_relatedness(issue, limit=limit)
+def find_related(repo: str = 'blender', number: int = 104399, limit: int = 15):
+    related = find_relatedness(repo, number, limit=limit)
     return related