Selaa lähdekoodia

fix:delete empty table bug (#15517)

Co-authored-by: huangzhuo <huangzhuo1@xiaomi.com>
tags/1.1.0
huangzhuo1949 7 kuukautta sitten
vanhempi
commit
695a7400a9
No account linked to committer's email address
1 muutettua tiedostoa jossa 10 lisäystä ja 1 poistoa
  1. 10
    1
      api/core/rag/datasource/vdb/pgvector/pgvector.py

+ 10
- 1
api/core/rag/datasource/vdb/pgvector/pgvector.py Näytä tiedosto

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
if not ids: if not ids:
return return
with self._get_cursor() as cur: with self._get_cursor() as cur:
cur.execute(f"DELETE FROM {self.table_name} WHERE id IN %s", (tuple(ids),))
try:
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:

Loading…
Peruuta
Tallenna