updating k value
Browse files
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("
|
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":
|
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 = (
|