Bladeren bron

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

tags/1.8.0
crazywoola 2 maanden geleden
bovenliggende
commit
8288b1dcab
No account linked to committer's email address

+ 0
- 5
api/configs/middleware/vdb/pgvector_config.py Bestand weergeven

@@ -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 Bestand weergeven

@@ -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…
Annuleren
Opslaan