AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-14 11:45:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix:delete empty table bug (#15517)

Browse Source 
Co-authored-by: huangzhuo <huangzhuo1@xiaomi.com>
This commit is contained in:
huangzhuo1949 2025-03-17 10:53:26 +08:00 committed by GitHub
parent e6a8800f66
commit 695a7400a9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
api/core/rag/datasource/vdb/pgvector/pgvector.py
View File

@ -1,8 +1,10 @@
import json import json
import logging
import uuid import uuid
from contextlib import contextmanager from contextlib import contextmanager
from typing import Any from typing import Any
import psycopg2.errors
import psycopg2.extras # type: ignore import psycopg2.extras # type: ignore
import psycopg2.pool # type: ignore import psycopg2.pool # type: ignore
from pydantic import BaseModel, model_validator from pydantic import BaseModel, model_validator
@ -147,7 +149,14 @@ class PGVector(BaseVector):
if not ids: if not ids:
return return
with self._get_cursor() as cur: with self._get_cursor() as cur:
try:
cur.execute(f"DELETE FROM {self.table_name} WHERE id IN %s", (tuple(ids),)) cur.execute(f"DELETE FROM {self.table_name} WHERE id IN %s", (tuple(ids),))
except psycopg2.errors.UndefinedTable:
# table not exists
logging.warning(f"Table {self.table_name} not found, skipping delete operation.")
return
except Exception as e:
raise e
def delete_by_metadata_field(self, key: str, value: str) -> None: def delete_by_metadata_field(self, key: str, value: str) -> None:
with self._get_cursor() as cur: with self._get_cursor() as cur: