From 3d6bcb99d04a4fc6f8dc64f685ea315691994fe3 Mon Sep 17 00:00:00 2001
From: shaohuzhang1 <80892890+shaohuzhang1@users.noreply.github.com>
Date: Mon, 18 Aug 2025 10:39:21 +0800
Subject: [PATCH] fix: Upgrading psycopg (#3869)

---
 apps/application/migrations/0015_re_database_index.py | 7 +++----
 pyproject.toml                                        | 2 +-
 2 files changed, 4 insertions(+), 5 deletions(-)

diff --git a/apps/application/migrations/0015_re_database_index.py b/apps/application/migrations/0015_re_database_index.py
index 740a2a2d2..cafe14e20 100644
--- a/apps/application/migrations/0015_re_database_index.py
+++ b/apps/application/migrations/0015_re_database_index.py
@@ -1,9 +1,8 @@
 # Generated by Django 4.2.15 on 2024-09-18 16:14
 import logging
 
-import psycopg2
+import psycopg
 from django.db import migrations
-from psycopg2 import extensions
 
 from smartdoc.const import CONFIG
 
@@ -17,7 +16,7 @@ def get_connect(db_name):
         "port": CONFIG.get('DB_PORT')
     }
     # 建立连接
-    connect = psycopg2.connect(**conn_params)
+    connect = psycopg.connect(**conn_params)
     return connect
 
 
@@ -28,7 +27,7 @@ def sql_execute(conn, reindex_sql: str, alter_database_sql: str):
     @param conn:
     @param alter_database_sql:
     """
-    conn.set_isolation_level(extensions.ISOLATION_LEVEL_AUTOCOMMIT)
+    conn.autocommit = True
     with conn.cursor() as cursor:
         cursor.execute(reindex_sql, [])
         cursor.execute(alter_database_sql, [])
diff --git a/pyproject.toml b/pyproject.toml
index ea87b807d..52ea56180 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -23,7 +23,7 @@ langchain-huggingface = "0.1.2"
 langchain-ollama = "0.3.2"
 langgraph = "0.3.27"
 mcp = "1.8.0"
-psycopg2-binary = "2.9.10"
+psycopg = { extras = ["binary"], version = "3.2.9" }
 jieba = "0.42.1"
 diskcache = "5.6.3"
 pillow = "10.4.0"