소스 검색

fix pg_vector extension requires SUPERUSER, but not available on Huawei Cloud RDS (#24093)

Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
tags/1.8.0
Elvis_LEE 2 달 전
부모
커밋
16d1289a0a
No account linked to committer's email address
2개의 변경된 파일7개의 추가작업 그리고 1개의 파일을 삭제
  1. 5
    0
      api/configs/middleware/vdb/pgvector_config.py
  2. 2
    1
      api/core/rag/datasource/vdb/pgvector/pgvector.py

+ 5
- 0
api/configs/middleware/vdb/pgvector_config.py 파일 보기

@@ -9,6 +9,11 @@ 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,

+ 2
- 1
api/core/rag/datasource/vdb/pgvector/pgvector.py 파일 보기

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

with self._get_cursor() as cur:
cur.execute("CREATE EXTENSION IF NOT EXISTS vector")
if dify_config.PGVECTOR_IS_OWNER:
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

Loading…
취소
저장