From 6103aa78307fea2c4cc1d067462bce3a635c1c6f Mon Sep 17 00:00:00 2001 From: jyong <718720800@qq.com> Date: Wed, 10 Sep 2025 15:02:03 +0800 Subject: [PATCH] fix user_id missed --- api/services/rag_pipeline/rag_pipeline.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/api/services/rag_pipeline/rag_pipeline.py b/api/services/rag_pipeline/rag_pipeline.py index cc3a19a2a5..6bfe239964 100644 --- a/api/services/rag_pipeline/rag_pipeline.py +++ b/api/services/rag_pipeline/rag_pipeline.py @@ -1312,11 +1312,16 @@ class RagPipelineService: "uninstalled_recommended_plugins": uninstalled_plugin_list, } - def get_datasource_plugins(self, pipeline_id: str, is_published: bool) -> list[dict]: + def get_datasource_plugins(self, dataset_id: str, is_published: bool) -> list[dict]: """ Get datasource plugins """ - pipeline: Pipeline | None = db.session.query(Pipeline).filter(Pipeline.id == pipeline_id).first() + dataset: Dataset | None = db.session.query(Dataset).filter(Dataset.id == dataset_id).first() + if not dataset: + raise ValueError("Dataset not found") + pipeline: Pipeline | None = db.session.query(Pipeline).filter(Pipeline.id == dataset.pipeline_id).first() + if not pipeline: + raise ValueError("Pipeline not found") workflow: Workflow | None = None if is_published: