Removed problematic import 'NotRegisteredError' from tool_parse library.

For some reason we can't import NotRegisteredError -> The API doesn't run. it got replaced with a generic Exception inside �gent.py.
antoninoLorenzo · Aug 21, 2024 · 3857c57 · 3857c57
1 parent c41710c
commit 3857c57
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/agent/agent.py b/src/agent/agent.py
@@ -3,7 +3,7 @@
 import json
 from json import JSONDecodeError
 
-from tool_parse import ToolRegistry, NotRegisteredError
+from tool_parse import ToolRegistry
 
 from src.agent.knowledge import Store
 from src.agent.llm import LLM, AVAILABLE_PROVIDERS
@@ -121,7 +121,7 @@ def invoke_tools(self, tool_response):
                 )
                 call_stack.append(tool_meta)
                 results.append({'role': 'tool', 'content': str(res)})
-            except NotRegisteredError:
+            except Exception:
                 pass
 
         return results

diff --git a/src/api.py b/src/api.py
@@ -28,7 +28,7 @@
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.responses import StreamingResponse
 from pydantic_settings import BaseSettings
-from tool_parse import ToolRegistry, NotRegisteredError
+from tool_parse import ToolRegistry
 
 from src import initialize_knowledge
 from src.agent import Agent