From 04ef5b27836dffc9d7bee99b2053359a0a419a16 Mon Sep 17 00:00:00 2001 From: cucusenok <31804608+cucusenok@users.noreply.github.com> Date: Wed, 17 Sep 2025 06:30:45 +0400 Subject: [PATCH] Fix: usage of postgresql -> postgres for db_type (#10120) ### What problem does this PR solve? This PR fixes incorrect naming for PostgreSQL usage by replacing all instances of `postgresql` with the correct `postgres` in the `db_type` field. This resolves potential configuration errors and ensures consistency when specifying the database type. Also fixed handling of None for `get_queue_length` ### Type of change - [x] Bug Fix (non-breaking change which fixes an issue) Co-authored-by: cucusenok --- agent/tools/exesql.py | 4 ++-- api/apps/canvas_app.py | 2 +- api/db/services/document_service.py | 2 ++ web/src/pages/agent/options.ts | 2 +- web/src/pages/data-flow/options.ts | 2 +- web/src/pages/flow/constant.tsx | 2 +- 6 files changed, 8 insertions(+), 6 deletions(-) diff --git a/agent/tools/exesql.py b/agent/tools/exesql.py index 317941713..c4bc4fdb4 100644 --- a/agent/tools/exesql.py +++ b/agent/tools/exesql.py @@ -53,7 +53,7 @@ class ExeSQLParam(ToolParamBase): self.max_records = 1024 def check(self): - self.check_valid_value(self.db_type, "Choose DB type", ['mysql', 'postgresql', 'mariadb', 'mssql']) + self.check_valid_value(self.db_type, "Choose DB type", ['mysql', 'postgres', 'mariadb', 'mssql']) self.check_empty(self.database, "Database name") self.check_empty(self.username, "database username") self.check_empty(self.host, "IP Address") @@ -111,7 +111,7 @@ class ExeSQL(ToolBase, ABC): if self._param.db_type in ["mysql", "mariadb"]: db = pymysql.connect(db=self._param.database, user=self._param.username, host=self._param.host, port=self._param.port, password=self._param.password) - elif self._param.db_type == 'postgresql': + elif self._param.db_type == 'postgres': db = psycopg2.connect(dbname=self._param.database, user=self._param.username, host=self._param.host, port=self._param.port, password=self._param.password) elif self._param.db_type == 'mssql': diff --git a/api/apps/canvas_app.py b/api/apps/canvas_app.py index 4ced90a3d..81f67b098 100644 --- a/api/apps/canvas_app.py +++ b/api/apps/canvas_app.py @@ -332,7 +332,7 @@ def test_db_connect(): if req["db_type"] in ["mysql", "mariadb"]: db = MySQLDatabase(req["database"], user=req["username"], host=req["host"], port=req["port"], password=req["password"]) - elif req["db_type"] == 'postgresql': + elif req["db_type"] == 'postgres': db = PostgresqlDatabase(req["database"], user=req["username"], host=req["host"], port=req["port"], password=req["password"]) elif req["db_type"] == 'mssql': diff --git a/api/db/services/document_service.py b/api/db/services/document_service.py index d58bfb322..3ebee2b7a 100644 --- a/api/db/services/document_service.py +++ b/api/db/services/document_service.py @@ -702,6 +702,8 @@ def queue_raptor_o_graphrag_tasks(doc, ty, priority): def get_queue_length(priority): group_info = REDIS_CONN.queue_info(get_svr_queue_name(priority), SVR_CONSUMER_GROUP_NAME) + if not group_info: + return 0 return int(group_info.get("lag", 0) or 0) diff --git a/web/src/pages/agent/options.ts b/web/src/pages/agent/options.ts index 4ad5a6457..63dd07a65 100644 --- a/web/src/pages/agent/options.ts +++ b/web/src/pages/agent/options.ts @@ -2133,7 +2133,7 @@ export const QWeatherTimePeriodOptions = [ '30d', ]; -export const ExeSQLOptions = ['mysql', 'postgresql', 'mariadb', 'mssql'].map( +export const ExeSQLOptions = ['mysql', 'postgres', 'mariadb', 'mssql'].map( (x) => ({ label: upperFirst(x), value: x, diff --git a/web/src/pages/data-flow/options.ts b/web/src/pages/data-flow/options.ts index 4ad5a6457..63dd07a65 100644 --- a/web/src/pages/data-flow/options.ts +++ b/web/src/pages/data-flow/options.ts @@ -2133,7 +2133,7 @@ export const QWeatherTimePeriodOptions = [ '30d', ]; -export const ExeSQLOptions = ['mysql', 'postgresql', 'mariadb', 'mssql'].map( +export const ExeSQLOptions = ['mysql', 'postgres', 'mariadb', 'mssql'].map( (x) => ({ label: upperFirst(x), value: x, diff --git a/web/src/pages/flow/constant.tsx b/web/src/pages/flow/constant.tsx index 79a58baa0..b4ec32902 100644 --- a/web/src/pages/flow/constant.tsx +++ b/web/src/pages/flow/constant.tsx @@ -2911,7 +2911,7 @@ export const QWeatherTimePeriodOptions = [ '30d', ]; -export const ExeSQLOptions = ['mysql', 'postgresql', 'mariadb', 'mssql'].map( +export const ExeSQLOptions = ['mysql', 'postgres', 'mariadb', 'mssql'].map( (x) => ({ label: upperFirst(x), value: x,