estel/dify
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

add qdrant metadata.doc_id index when create qdrant collection (#4570)

Browse Source
This commit is contained in:
Jyong
2024-05-23 18:11:01 +08:00
committed by GitHub
parent 233c4150d1
commit 24624491cd
2 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
api/core/rag/datasource/vdb/field.py
View File

@@ -8,3 +8,4 @@ class Field(Enum):
VECTOR = "vector" VECTOR = "vector"
TEXT_KEY = "text" TEXT_KEY = "text"
PRIMARY_KEY = "id" PRIMARY_KEY = "id"
DOC_ID = "metadata.doc_id"

5
api/core/rag/datasource/vdb/qdrant/qdrant_vector.py
View File

@@ -115,9 +115,12 @@ class QdrantVector(BaseVector):
timeout=int(self._client_config.timeout), timeout=int(self._client_config.timeout),
) )
# create payload index # create group_id payload index
self._client.create_payload_index(collection_name, Field.GROUP_KEY.value, self._client.create_payload_index(collection_name, Field.GROUP_KEY.value,
field_schema=PayloadSchemaType.KEYWORD) field_schema=PayloadSchemaType.KEYWORD)
# create doc_id payload index
self._client.create_payload_index(collection_name, Field.DOC_ID.value,
field_schema=PayloadSchemaType.KEYWORD)
# creat full text index # creat full text index
text_index_params = TextIndexParams( text_index_params = TextIndexParams(
type=TextIndexType.TEXT, type=TextIndexType.TEXT,