feat: integrate psycogreen for gevent compatibility in PostgreSQL (#1…

…2253) Signed-off-by: -LAN- <laipz8200@outlook.com>
langgenius · Dec 31, 2024 · 63a0b8b · 63a0b8b
1 parent 634b382
commit 63a0b8b
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 44 deletions.
diff --git a/api/app.py b/api/app.py
@@ -25,6 +25,10 @@ def is_db_command():
         # grpc gevent
         grpc_gevent.init_gevent()
 
+        import psycogreen.gevent  # type: ignore
+
+        psycogreen.gevent.patch_psycopg()
+
     from app_factory import create_app
 
     app = create_app()

diff --git a/api/poetry.lock b/api/poetry.lock
diff --git a/api/pyproject.toml b/api/pyproject.toml
@@ -61,6 +61,7 @@ openai = "~1.52.0"
 openpyxl = "~3.1.5"
 pandas = { version = "~2.2.2", extras = ["performance", "excel"] }
 pandas-stubs = "~2.2.3.241009"
+psycogreen = "~1.0.2"
 psycopg2-binary = "~2.9.6"
 pycryptodome = "3.19.1"
 pydantic = "~2.9.2"