Browse Source
Revert "fix pg_vector extension requires SUPERUSER, but not availabl… (#24108)
tags/1.8.0
crazywoola
2 months ago
2 changed files with 1 additions and 7 deletions
+ 0
- 5
api/configs/middleware/vdb/pgvector_config.py
View File
| Configuration settings for PGVector (PostgreSQL with vector extension) | 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( | PGVECTOR_HOST: Optional[str] = Field( | ||||
| description="Hostname or IP address of the PostgreSQL server with PGVector extension (e.g., 'localhost')", | description="Hostname or IP address of the PostgreSQL server with PGVector extension (e.g., 'localhost')", | ||||
| default=None, | default=None, |
+ 1
- 2
api/core/rag/datasource/vdb/pgvector/pgvector.py
View File
| return | return | ||||
| with self._get_cursor() as cur: | 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)) | cur.execute(SQL_CREATE_TABLE.format(table_name=self.table_name, dimension=dimension)) | ||||
| # PG hnsw index only support 2000 dimension or less | # PG hnsw index only support 2000 dimension or less | ||||
| # ref: https://github.com/pgvector/pgvector?tab=readme-ov-file#indexing | # ref: https://github.com/pgvector/pgvector?tab=readme-ov-file#indexing |
Loading…