feat(datasource): add datasource content preview api

10 months ago · 6aba39a2dd
parent 49bb15fae1
commit 6aba39a2dd
1 changed files with 2 additions and 1 deletions
--- a/api/controllers/console/datasets/rag_pipeline/datasource_content_preview.py
+++ b/api/controllers/console/datasets/rag_pipeline/datasource_content_preview.py
@ -38,7 +38,7 @@ class DataSourceContentPreviewApi(Resource):
            raise ValueError("missing datasource_type")

        rag_pipeline_service = RagPipelineService()
-        return rag_pipeline_service.run_datasource_node_preview(
+        preview_content = rag_pipeline_service.run_datasource_node_preview(
            pipeline=pipeline,
            node_id=node_id,
            user_inputs=inputs,
@ -46,6 +46,7 @@ class DataSourceContentPreviewApi(Resource):
            datasource_type=datasource_type,
            is_published=True,
        )
+        return preview_content, 200

 api.add_resource(
    DataSourceContentPreviewApi,