From 992e1eedde11e615bbc38af8723a3bf1a2046802 Mon Sep 17 00:00:00 2001 From: Yeuoly Date: Mon, 25 Nov 2024 23:36:19 +0800 Subject: [PATCH] fix: export agent dsl --- api/services/app_dsl_service.py | 13 ++++++------- 1 file changed, 6 insertions(+), 7 deletions(-) diff --git a/api/services/app_dsl_service.py b/api/services/app_dsl_service.py index 92e46c1e61..f7b74aed4e 100644 --- a/api/services/app_dsl_service.py +++ b/api/services/app_dsl_service.py @@ -627,7 +627,7 @@ class AppDslService: # reranking model dataset_configs = model_config.dataset_configs_dict 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"): dependencies.append( DependenciesAnalysisService.analyze_model_provider_dependency( @@ -640,12 +640,11 @@ class AppDslService: # tools agent_configs = model_config.agent_mode_dict if agent_configs: - for agent_config in agent_configs: - if agent_config.get("tools"): - for tool in agent_config.get("tools", []): - dependencies.append( - DependenciesAnalysisService.analyze_tool_dependency(tool.get("provider_id")) - ) + for agent_config in agent_configs.get("tools", []): + dependencies.append( + DependenciesAnalysisService.analyze_tool_dependency(agent_config.get("provider_id")) + ) + except Exception as e: logger.exception("Error extracting model config dependency", exc_info=e)