Rename vectors

This commit is contained in:
Derrick Mwiti
2025-06-10 11:40:10 +03:00
committed by GitHub
parent 32e29e77a7
commit d35bc3b28c
@@ -103,17 +103,17 @@ documents = [
] ]
query_text = "How does AI help in medicine?" query_text = "How does AI help in medicine?"
dense_doc_vectors = [ dense_documents = [
models.Document(text=doc, model="BAAI/bge-small-en") models.Document(text=doc, model="BAAI/bge-small-en")
for doc in documents for doc in documents
] ]
dense_query_vector = models.Document(text=query_text, model="BAAI/bge-small-en") dense_query = models.Document(text=query_text, model="BAAI/bge-small-en")
colbert_doc_vectors = [ colbert_documents = [
models.Document(text=doc, model="colbert-ir/colbertv2.0") models.Document(text=doc, model="colbert-ir/colbertv2.0")
for doc in documents for doc in documents
] ]
colbert_query_vector = models.Document(text=query_text, model="colbert-ir/colbertv2.0") colbert_query = models.Document(text=query_text, model="colbert-ir/colbertv2.0")
``` ```
@@ -149,8 +149,8 @@ points = [
models.PointStruct( models.PointStruct(
id=i, id=i,
vector={ vector={
"dense": dense_vectors[i], "dense": dense_documents[i],
"colbert": colbert_vectors[i] "colbert": colbert_documents[i]
}, },
payload={"text": documents[i]} payload={"text": documents[i]}
) for i in range(len(documents)) ) for i in range(len(documents))
@@ -166,10 +166,10 @@ Now let’s run a search:
results = client.query_points( results = client.query_points(
collection_name="dense_multivector_demo", collection_name="dense_multivector_demo",
prefetch=models.Prefetch( prefetch=models.Prefetch(
query=dense_query_vector, query=dense_query,
using="dense", using="dense",
), ),
query=colbert_query_vector, query=colbert_query,
using="colbert", using="colbert",
limit=3, limit=3,
with_payload=True with_payload=True