Update app.py
Browse files
app.py
CHANGED
@@ -46,7 +46,7 @@ def create_retriever_from_chroma(vectorstore_path="docs/chroma/", search_type='m
|
|
46 |
llm = llamacpp.LlamaCpp(
|
47 |
model_path='qwen2-0_5b-instruct-q4_0.gguf',
|
48 |
n_gpu_layers=0,
|
49 |
-
temperature=0.
|
50 |
top_p=0.9,
|
51 |
n_ctx=22000,
|
52 |
n_batch=2000,
|
@@ -83,7 +83,7 @@ def create_retriever_from_chroma(vectorstore_path="docs/chroma/", search_type='m
|
|
83 |
)
|
84 |
|
85 |
|
86 |
-
retriever=vectorstore.as_retriever(search_type=
|
87 |
|
88 |
rephrase_prompt = hub.pull("langchain-ai/chat-langchain-rephrase")
|
89 |
|
|
|
46 |
llm = llamacpp.LlamaCpp(
|
47 |
model_path='qwen2-0_5b-instruct-q4_0.gguf',
|
48 |
n_gpu_layers=0,
|
49 |
+
temperature=0.1,
|
50 |
top_p=0.9,
|
51 |
n_ctx=22000,
|
52 |
n_batch=2000,
|
|
|
83 |
)
|
84 |
|
85 |
|
86 |
+
retriever=vectorstore.as_retriever(search_type = search_type, search_kwargs={"k": k})
|
87 |
|
88 |
rephrase_prompt = hub.pull("langchain-ai/chat-langchain-rephrase")
|
89 |
|