diff --git a/api/services/rag_pipeline/pipeline_template/remote/remote_retrieval.py b/api/services/rag_pipeline/pipeline_template/remote/remote_retrieval.py index 7acffea360..4ed0f6d0a4 100644 --- a/api/services/rag_pipeline/pipeline_template/remote/remote_retrieval.py +++ b/api/services/rag_pipeline/pipeline_template/remote/remote_retrieval.py @@ -4,6 +4,7 @@ from typing import Optional import requests from configs import dify_config +from services.rag_pipeline.pipeline_template.database.database_retrieval import DatabasePipelineTemplateRetrieval from services.rag_pipeline.pipeline_template.pipeline_template_base import PipelineTemplateRetrievalBase from services.rag_pipeline.pipeline_template.pipeline_template_type import PipelineTemplateType from services.recommend_app.buildin.buildin_retrieval import BuildInRecommendAppRetrieval @@ -20,16 +21,16 @@ class RemotePipelineTemplateRetrieval(PipelineTemplateRetrievalBase): try: result = self.fetch_pipeline_template_detail_from_dify_official(template_id) except Exception as e: - logger.warning("fetch recommended app detail from dify official failed: %r, switch to built-in.", e) - result = BuildInRecommendAppRetrieval.fetch_recommended_app_detail_from_builtin(template_id) + logger.warning("fetch recommended app detail from dify official failed: %r, switch to database.", e) + result = [DatabasePipelineTemplateRetrieval.fetch_pipeline_template_detail_from_db(template_id)] return result def get_pipeline_templates(self, language: str) -> dict: try: result = self.fetch_pipeline_templates_from_dify_official(language) except Exception as e: - logger.warning("fetch pipeline templates from dify official failed: %r, switch to built-in.", e) - result = BuildInRecommendAppRetrieval.fetch_recommended_apps_from_builtin(language) + logger.warning("fetch pipeline templates from dify official failed: %r, switch to database.", e) + result = DatabasePipelineTemplateRetrieval.fetch_pipeline_templates_from_db(language) return result def get_type(self) -> str: