devve1 commited on
Commit
836111b
1 Parent(s): a9efbc8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +22 -20
app.py CHANGED
@@ -86,26 +86,28 @@ def query_hybrid_search(query: str, client: QdrantClient, collection_name: str,
86
 
87
  return client.query_batch_points(
88
  collection_name=collection_name,
89
- requests=QueryRequest(
90
- prefetch=Prefetch(query=sparse_embeddings.as_object(), using="text-sparse", limit=10),
91
- with_vector=False,
92
- with_payload=True,
93
- query=FusionQuery(fusion=Fusion.DSBF),
94
- limit=3
95
- #filter=Filter(must_not=[
96
- # HasIdCondition(has_id=[])
97
- #])
98
- ),
99
- QueryRequest(
100
- prefetch=Prefetch(query=dense_embeddings, using="text-dense", limit=10),
101
- with_vector=False,
102
- with_payload=True,
103
- query=FusionQuery(fusion=Fusion.DSBF),
104
- limit=3
105
- #filter=Filter(must_not=[
106
- # HasIdCondition(has_id=[])
107
- #])
108
- )
 
 
109
  )
110
 
111
  def main(query: str, client: QdrantClient, collection_name: str, tokenizer: AutoTokenizer, llm: vllm.LLM, dense_model: OptimumEncoder, sparse_model: SparseTextEmbedding):
 
86
 
87
  return client.query_batch_points(
88
  collection_name=collection_name,
89
+ requests=[
90
+ QueryRequest(
91
+ prefetch=Prefetch(query=sparse_embeddings.as_object(), using="text-sparse", limit=10),
92
+ with_vector=False,
93
+ with_payload=True,
94
+ query=FusionQuery(fusion=Fusion.DSBF),
95
+ limit=3
96
+ #filter=Filter(must_not=[
97
+ # HasIdCondition(has_id=[])
98
+ #])
99
+ ),
100
+ QueryRequest(
101
+ prefetch=Prefetch(query=dense_embeddings, using="text-dense", limit=10),
102
+ with_vector=False,
103
+ with_payload=True,
104
+ query=FusionQuery(fusion=Fusion.DSBF),
105
+ limit=3
106
+ #filter=Filter(must_not=[
107
+ # HasIdCondition(has_id=[])
108
+ #])
109
+ )
110
+ ]
111
  )
112
 
113
  def main(query: str, client: QdrantClient, collection_name: str, tokenizer: AutoTokenizer, llm: vllm.LLM, dense_model: OptimumEncoder, sparse_model: SparseTextEmbedding):