Skip to content

Commit 2f545d1

Browse files
committed
add pinecone to retrieval step
1 parent faaa477 commit 2f545d1

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

llm-complete-guide/steps/eval_retrieval.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@
2727
get_db_conn,
2828
get_embeddings,
2929
get_es_client,
30+
get_pinecone_client,
3031
get_topn_similar_docs,
3132
rerank_documents,
3233
)
@@ -94,6 +95,8 @@ def query_similar_docs(
9495
vector_store_name = find_vectorstore_name()
9596
if vector_store_name == "pgvector":
9697
conn = get_db_conn()
98+
elif vector_store_name == "pinecone":
99+
pinecone_index = get_pinecone_client()
97100
else:
98101
es_client = get_es_client()
99102

@@ -103,6 +106,7 @@ def query_similar_docs(
103106
embedded_question,
104107
conn=conn,
105108
es_client=es_client,
109+
pinecone_index=pinecone_index,
106110
n=num_docs,
107111
include_metadata=True,
108112
)

0 commit comments

Comments
 (0)