Spaces:
Runtime error
Runtime error
Upload 6 files
Browse files- 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"},
|
|
|
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"]
|