Quellcode durchsuchen

Revert "fix pg_vector extension requires SUPERUSER, but not availabl… (#24108)

tags/1.8.0
crazywoola vor 2 Monaten
Ursprung
Commit
8288b1dcab
Es ist kein Account mit der E-Mail-Adresse des Committers verbunden

+ 0
- 5
api/configs/middleware/vdb/pgvector_config.py Datei anzeigen

@@ -9,11 +9,6 @@ class PGVectorConfig(BaseSettings):
Configuration settings for PGVector (PostgreSQL with vector extension)
"""

PGVECTOR_IS_OWNER: Optional[bool] = Field(
description="Whether the PGVector is the owner of the database",
default=True,
)

PGVECTOR_HOST: Optional[str] = Field(
description="Hostname or IP address of the PostgreSQL server with PGVector extension (e.g., 'localhost')",
default=None,

+ 1
- 2
api/core/rag/datasource/vdb/pgvector/pgvector.py Datei anzeigen

@@ -253,8 +253,7 @@ class PGVector(BaseVector):
return

with self._get_cursor() as cur:
if dify_config.PGVECTOR_IS_OWNER:
cur.execute("CREATE EXTENSION IF NOT EXISTS vector")
cur.execute("CREATE EXTENSION IF NOT EXISTS vector")
cur.execute(SQL_CREATE_TABLE.format(table_name=self.table_name, dimension=dimension))
# PG hnsw index only support 2000 dimension or less
# ref: https://github.com/pgvector/pgvector?tab=readme-ov-file#indexing

Laden…
Abbrechen
Speichern