Daniel Foley commited on
Commit
83b3fa4
·
1 Parent(s): fa714bc

fixed input errors... hopefully

Browse files
Files changed (2) hide show
  1. RAG.py +1 -2
  2. streamlit_app.py +0 -2
RAG.py CHANGED
@@ -16,8 +16,7 @@ import json
16
  import logging
17
 
18
  def retrieve(query: str,vectorstore:PineconeVectorStore, k: int = 1000) -> Tuple[List[Document], List[float]]:
19
- start = time.time()
20
- load_dotenv()
21
  # pinecone_api_key = os.getenv("PINECONE_API_KEY")
22
  # pc = Pinecone(api_key=pinecone_api_key)
23
 
 
16
  import logging
17
 
18
  def retrieve(query: str,vectorstore:PineconeVectorStore, k: int = 1000) -> Tuple[List[Document], List[float]]:
19
+ start = time.time()
 
20
  # pinecone_api_key = os.getenv("PINECONE_API_KEY")
21
  # pc = Pinecone(api_key=pinecone_api_key)
22
 
streamlit_app.py CHANGED
@@ -129,8 +129,6 @@ def main():
129
  response, sources = process_message(
130
  query=user_input,
131
  llm=st.session_state.llm,
132
- index_name=INDEX_NAME,
133
- embeddings=st.session_state.embeddings,
134
  vectorstore=st.session_state.pinecone
135
  )
136
 
 
129
  response, sources = process_message(
130
  query=user_input,
131
  llm=st.session_state.llm,
 
 
132
  vectorstore=st.session_state.pinecone
133
  )
134