youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-04 01:18:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

r2

Browse Source
This commit is contained in:
jyong
2025-05-15 17:19:14 +08:00
parent 93ac6d37e9
commit 4ff971c8a3
11 changed files with 71 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
api/services/rag_pipeline/pipeline_template/database/database_retrieval.py
View File

@ -2,12 +2,12 @@ from typing import Optional
from extensions.ext_database import db
from models.dataset import Pipeline, PipelineBuiltInTemplate
from services.app_dsl_service import AppDslService
from services.recommend_app.recommend_app_base import RecommendAppRetrievalBase
from services.recommend_app.recommend_app_type import RecommendAppType
from services.rag_pipeline.pipeline_template.pipeline_template_base import PipelineTemplateRetrievalBase
from services.rag_pipeline.pipeline_template.pipeline_template_type import PipelineTemplateType
#from services.rag_pipeline.rag_pipeline_dsl_service import RagPipelineDslService
class DatabasePipelineTemplateRetrieval(RecommendAppRetrievalBase):
class DatabasePipelineTemplateRetrieval(PipelineTemplateRetrievalBase):
"""
Retrieval pipeline template from database
"""
@ -21,7 +21,7 @@ class DatabasePipelineTemplateRetrieval(RecommendAppRetrievalBase):
return result
def get_type(self) -> str:
return RecommendAppType.DATABASE
return PipelineTemplateType.DATABASE
@classmethod
def fetch_pipeline_templates_from_db(cls, language: str) -> dict:
@ -61,5 +61,5 @@ class DatabasePipelineTemplateRetrieval(RecommendAppRetrievalBase):
"name": pipeline.name,
"icon": pipeline.icon,
"mode": pipeline.mode,
"export_data": AppDslService.export_dsl(app_model=pipeline),
"export_data": RagPipelineDslService.export_rag_pipeline_dsl(pipeline=pipeline),
}