AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-19 14:19:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: reduce model provider fetchs

Browse Source
This commit is contained in:
Yeuoly 2024-12-19 01:02:08 +08:00
parent d5c708c62b
commit 1b21d7513d
No known key found for this signature in database
GPG Key ID: A66E7E320FB19F61
2 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/contexts/__init__.py
View File

@ -15,5 +15,5 @@ workflow_variable_pool: ContextVar["VariablePool"] = ContextVar("workflow_variab
plugin_tool_providers: ContextVar[dict[str, "PluginToolProviderController"]] = ContextVar("plugin_tool_providers") plugin_tool_providers: ContextVar[dict[str, "PluginToolProviderController"]] = ContextVar("plugin_tool_providers")
plugin_tool_providers_lock: ContextVar[Lock] = ContextVar("plugin_tool_providers_lock") plugin_tool_providers_lock: ContextVar[Lock] = ContextVar("plugin_tool_providers_lock")
plugin_model_providers: ContextVar[list["PluginModelProviderEntity"]] = ContextVar("plugin_model_providers") plugin_model_providers: ContextVar[list["PluginModelProviderEntity"] | None] = ContextVar("plugin_model_providers")
plugin_model_providers_lock: ContextVar[Lock] = ContextVar("plugin_model_providers_lock") plugin_model_providers_lock: ContextVar[Lock] = ContextVar("plugin_model_providers_lock")

7
api/core/model_runtime/model_providers/model_provider_factory.py
View File

@ -77,11 +77,16 @@ class ModelProviderFactory:
try: try:
contexts.plugin_model_providers.get() contexts.plugin_model_providers.get()
except LookupError: except LookupError:
contexts.plugin_model_providers.set([]) contexts.plugin_model_providers.set(None)
contexts.plugin_model_providers_lock.set(Lock()) contexts.plugin_model_providers_lock.set(Lock())
with contexts.plugin_model_providers_lock.get(): with contexts.plugin_model_providers_lock.get():
plugin_model_providers = contexts.plugin_model_providers.get() plugin_model_providers = contexts.plugin_model_providers.get()
if plugin_model_providers is not None:
return plugin_model_providers
plugin_model_providers = []
contexts.plugin_model_providers.set(plugin_model_providers)
# Fetch plugin model providers # Fetch plugin model providers
plugin_providers = self.plugin_model_manager.fetch_model_providers(self.tenant_id) plugin_providers = self.plugin_model_manager.fetch_model_providers(self.tenant_id)