Sbnos commited on
Commit
8a2c4ea
·
verified ·
1 Parent(s): d938a07

updating k value

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -6,7 +6,7 @@ from langchain.embeddings import HuggingFaceBgeEmbeddings
6
  from langchain.chains import ConversationalRetrievalChain
7
 
8
  # --- Configuration ---
9
- TOGETHER_API_KEY = os.environ.get("TOGETHER_API_KEY")
10
  if not TOGETHER_API_KEY:
11
  st.error("Missing TOGETHER_API_KEY environment variable.")
12
  st.stop()
@@ -52,7 +52,7 @@ vectorstore = Chroma(
52
  persist_directory=persist_directory,
53
  embedding_function=embeddings
54
  )
55
- retriever = vectorstore.as_retriever(search_kwargs={"k":5})
56
 
57
  # System prompt template
58
  SYSTEM_PROMPT = (
 
6
  from langchain.chains import ConversationalRetrievalChain
7
 
8
  # --- Configuration ---
9
+ TOGETHER_API_KEY = os.environ.get("pilotikval")
10
  if not TOGETHER_API_KEY:
11
  st.error("Missing TOGETHER_API_KEY environment variable.")
12
  st.stop()
 
52
  persist_directory=persist_directory,
53
  embedding_function=embeddings
54
  )
55
+ retriever = vectorstore.as_retriever(search_kwargs={"k":20})
56
 
57
  # System prompt template
58
  SYSTEM_PROMPT = (