| @@ -563,10 +563,10 @@ class DatasetRetrievalSettingApi(Resource): | |||
| case ( | |||
| VectorType.MILVUS | |||
| | VectorType.RELYT | |||
| | VectorType.PGVECTOR | |||
| | VectorType.TIDB_VECTOR | |||
| | VectorType.CHROMA | |||
| | VectorType.TENCENT | |||
| | VectorType.PGVECTO_RS | |||
| ): | |||
| return {"retrieval_method": [RetrievalMethod.SEMANTIC_SEARCH.value]} | |||
| case ( | |||
| @@ -577,6 +577,7 @@ class DatasetRetrievalSettingApi(Resource): | |||
| | VectorType.MYSCALE | |||
| | VectorType.ORACLE | |||
| | VectorType.ELASTICSEARCH | |||
| | VectorType.PGVECTOR | |||
| ): | |||
| return { | |||
| "retrieval_method": [ | |||
| @@ -56,7 +56,7 @@ class TencentVector(BaseVector): | |||
| return self._client.create_database(database_name=self._client_config.database) | |||
| def get_type(self) -> str: | |||
| return "tencent" | |||
| return VectorType.TENCENT | |||
| def to_index_struct(self) -> dict: | |||
| return {"type": self.get_type(), "vector_store": {"class_prefix": self._collection_name}} | |||