awinml commited on
Commit
69427c3
·
1 Parent(s): f130384

Upload 6 files

Browse files
Files changed (1) hide show
  1. utils/retriever.py +4 -2
utils/retriever.py CHANGED
@@ -1,6 +1,7 @@
1
  import numpy as np
2
 
3
 
 
4
  def query_pinecone(
5
  dense_vec,
6
  top_k,
@@ -9,8 +10,9 @@ def query_pinecone(
9
  ):
10
  xc = index.query(
11
  vector=dense_vec,
12
- top_k=top_k,
13
- filter={"QA_Flag": {"$eq": "Answer"}, "index": {"$in": indices}},
 
14
  include_metadata=True,
15
  )
16
  return xc["matches"]
 
1
  import numpy as np
2
 
3
 
4
+
5
  def query_pinecone(
6
  dense_vec,
7
  top_k,
 
10
  ):
11
  xc = index.query(
12
  vector=dense_vec,
13
+ top_k=top_k,
14
+ filter={"QA_Flag": {"$eq": "Answer"},
15
+ "index": {"$in": indices}},
16
  include_metadata=True,
17
  )
18
  return xc["matches"]