Procházet zdrojové kódy

Fix the vector retrieval sorting issue (#6431)

Co-authored-by: weifj <“weifj@tuyuansu.com.cn”>
tags/0.6.15
leoterry před 1 rokem
rodič
revize
dc847ba145
Žádný účet není propojen s e-mailovou adresou tvůrce revize

+ 2
- 1
api/core/rag/datasource/vdb/chroma/chroma_vector.py Zobrazit soubor

@@ -111,7 +111,8 @@ class ChromaVector(BaseVector):
metadata=metadata,
)
docs.append(doc)

# Sort the documents by score in descending order
docs = sorted(docs, key=lambda x: x.metadata['score'], reverse=True)
return docs

def search_by_full_text(self, query: str, **kwargs: Any) -> list[Document]:

+ 2
- 0
api/core/rag/datasource/vdb/qdrant/qdrant_vector.py Zobrazit soubor

@@ -362,6 +362,8 @@ class QdrantVector(BaseVector):
metadata=metadata,
)
docs.append(doc)
# Sort the documents by score in descending order
docs = sorted(docs, key=lambda x: x.metadata['score'], reverse=True)
return docs

def search_by_full_text(self, query: str, **kwargs: Any) -> list[Document]:

+ 2
- 1
api/core/rag/datasource/vdb/weaviate/weaviate_vector.py Zobrazit soubor

@@ -216,7 +216,8 @@ class WeaviateVector(BaseVector):
if score > score_threshold:
doc.metadata['score'] = score
docs.append(doc)

# Sort the documents by score in descending order
docs = sorted(docs, key=lambda x: x.metadata['score'], reverse=True)
return docs

def search_by_full_text(self, query: str, **kwargs: Any) -> list[Document]:

Načítá se…
Zrušit
Uložit