From ab1730bbaacb2248dec3d30af0933c1b52b3bb67 Mon Sep 17 00:00:00 2001
From: jyong <718720800@qq.com>
Date: Tue, 3 Jun 2025 16:51:21 +0800
Subject: [PATCH] r2

---
 api/app.py                                    | 22 +++++++++----------
 .../app/apps/pipeline/pipeline_generator.py   |  1 +
 2 files changed, 12 insertions(+), 11 deletions(-)

diff --git a/api/app.py b/api/app.py
index 4f393f6c20..34889cff67 100644
--- a/api/app.py
+++ b/api/app.py
@@ -19,18 +19,18 @@ else:
     # If you are using debugpy and set GEVENT_SUPPORT=True, you can debug with gevent.
     if (flask_debug := os.environ.get("FLASK_DEBUG", "0")) and flask_debug.lower() in {"false", "0", "no"}:
         from gevent import monkey
+        #
+        # # gevent
+        # monkey.patch_all()
+        #
+        # from grpc.experimental import gevent as grpc_gevent  # type: ignore
+        #
+        # # grpc gevent
+        # grpc_gevent.init_gevent()
 
-        # gevent
-        monkey.patch_all()
-
-        from grpc.experimental import gevent as grpc_gevent  # type: ignore
-
-        # grpc gevent
-        grpc_gevent.init_gevent()
-
-        import psycogreen.gevent  # type: ignore
-
-        psycogreen.gevent.patch_psycopg()
+        # import psycogreen.gevent  # type: ignore
+        #
+        # psycogreen.gevent.patch_psycopg()
 
     from app_factory import create_app
 
diff --git a/api/core/app/apps/pipeline/pipeline_generator.py b/api/core/app/apps/pipeline/pipeline_generator.py
index d56e5243b3..61d4b723e1 100644
--- a/api/core/app/apps/pipeline/pipeline_generator.py
+++ b/api/core/app/apps/pipeline/pipeline_generator.py
@@ -236,6 +236,7 @@ class PipelineGenerator(BaseAppGenerator):
             "documents": [PipelineDocument(
                 id=document.id,
                 position=document.position,
+                data_source_type=document.data_source_type,
                 data_source_info=json.loads(document.data_source_info) if document.data_source_info else None,
                 name=document.name,
                 indexing_status=document.indexing_status,