Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>tags/1.8.0
| @@ -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, | |||
| @@ -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 | |||