devve1 commited on
Commit
9dd9d27
1 Parent(s): 8336c69

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -68,10 +68,10 @@ def query_hybrid_search(query: str, client: QdrantClient, collection_name: str,
68
  )
69
 
70
  def main(query: str, client: QdrantClient, collection_name: str, llm: Llama, dense_model: OptimumEncoder, sparse_model: SparseTextEmbedding):
71
- result = query_hybrid_search(query, client, collection_name, dense_model, sparse_model)
72
- print(result)
73
 
74
- docs = [(record.payload['text'], record.payload['metadata']) for record in records_list]
75
  contents, metadatas = [list(t) for t in zip(*docs)]
76
 
77
  context = "\n".join(contents)
 
68
  )
69
 
70
  def main(query: str, client: QdrantClient, collection_name: str, llm: Llama, dense_model: OptimumEncoder, sparse_model: SparseTextEmbedding):
71
+ scored_points = query_hybrid_search(query, client, collection_name, dense_model, sparse_model)
72
+ print(scored_points)
73
 
74
+ docs = [(scored_point.payload['text'], scored_point.payload['metadata']) for scored_point in scored_points]
75
  contents, metadatas = [list(t) for t in zip(*docs)]
76
 
77
  context = "\n".join(contents)