Преглед на файлове

fix: Lookup errors for contextvars used in ToolManager

Yeuoly преди 5 месеца
родител
ревизия
77a62f33b3
променени са 1 файла, в които са добавени 7 реда и са изтрити 0 реда
  1. 7 0
      api/core/tools/tool_manager.py

+ 7 - 0
api/core/tools/tool_manager.py

@@ -100,6 +100,13 @@ class ToolManager:
         """
         get the plugin provider
         """
+        # check if context is set
+        try:
+            contexts.plugin_tool_providers.get()
+        except LookupError:
+            contexts.plugin_tool_providers.set({})
+            contexts.plugin_tool_providers_lock.set(Lock())
+
         with contexts.plugin_tool_providers_lock.get():
             plugin_tool_providers = contexts.plugin_tool_providers.get()
             if provider in plugin_tool_providers: