query

.gitignore

@@ -0,0 +1 @@
+.streamlit

app.py

@@ -25,7 +25,7 @@ OPENAI_CHAT_MODELS = [
 ]
 
 
-def load_pinecone_vectorstore():
+def load_bge_embeddings():
     model_name = "BAAI/bge-small-en-v1.5"
     model_kwargs = {"device": "cpu"}
     encode_kwargs = {"normalize_embeddings": True}
@@ -35,6 +35,11 @@ def load_pinecone_vectorstore():
         encode_kwargs=encode_kwargs,
         query_instruction="Represent this question for searching relevant passages: ",
     )
+    return emb_fn
+
+
+def load_pinecone_vectorstore():
+    emb_fn = load_bge_embeddings()
     pc = PineconeClient(api_key=st.secrets["pinecone_api_key"])
     index = pc.Index(st.secrets["pinecone_index_name"])
     vectorstore = Pinecone(
@@ -48,5 +53,6 @@ def load_pinecone_vectorstore():
 
 
 vectorstore = load_pinecone_vectorstore()
-docs = vectorstore.similarity_search_with_score("artificial intelligence")
+query = st.text_area("Enter query")
+docs = vectorstore.similarity_search_with_score(query)
 st.write(docs)