We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent faaa477 commit 2f545d1Copy full SHA for 2f545d1
llm-complete-guide/steps/eval_retrieval.py
@@ -27,6 +27,7 @@
27
get_db_conn,
28
get_embeddings,
29
get_es_client,
30
+ get_pinecone_client,
31
get_topn_similar_docs,
32
rerank_documents,
33
)
@@ -94,6 +95,8 @@ def query_similar_docs(
94
95
vector_store_name = find_vectorstore_name()
96
if vector_store_name == "pgvector":
97
conn = get_db_conn()
98
+ elif vector_store_name == "pinecone":
99
+ pinecone_index = get_pinecone_client()
100
else:
101
es_client = get_es_client()
102
@@ -103,6 +106,7 @@ def query_similar_docs(
103
106
embedded_question,
104
107
conn=conn,
105
108
es_client=es_client,
109
+ pinecone_index=pinecone_index,
110
n=num_docs,
111
include_metadata=True,
112
0 commit comments