fix: export agent dsl

This commit is contained in:
Yeuoly 2024-11-25 23:36:19 +08:00
parent c2ce8e638e
commit 992e1eedde
No known key found for this signature in database
GPG Key ID: A66E7E320FB19F61

View File

@ -627,7 +627,7 @@ class AppDslService:
# reranking model # reranking model
dataset_configs = model_config.dataset_configs_dict dataset_configs = model_config.dataset_configs_dict
if dataset_configs: if dataset_configs:
for dataset_config in dataset_configs: for dataset_config in dataset_configs.get("datasets", {}).get("datasets", []):
if dataset_config.get("reranking_model"): if dataset_config.get("reranking_model"):
dependencies.append( dependencies.append(
DependenciesAnalysisService.analyze_model_provider_dependency( DependenciesAnalysisService.analyze_model_provider_dependency(
@ -640,12 +640,11 @@ class AppDslService:
# tools # tools
agent_configs = model_config.agent_mode_dict agent_configs = model_config.agent_mode_dict
if agent_configs: if agent_configs:
for agent_config in agent_configs: for agent_config in agent_configs.get("tools", []):
if agent_config.get("tools"): dependencies.append(
for tool in agent_config.get("tools", []): DependenciesAnalysisService.analyze_tool_dependency(agent_config.get("provider_id"))
dependencies.append( )
DependenciesAnalysisService.analyze_tool_dependency(tool.get("provider_id"))
)
except Exception as e: except Exception as e:
logger.exception("Error extracting model config dependency", exc_info=e) logger.exception("Error extracting model config dependency", exc_info=e)