Spaces:
Sleeping
Sleeping
Update langchain_movie_search.py
Browse files
langchain_movie_search.py
CHANGED
|
@@ -112,7 +112,7 @@ class MoviesSearch:
|
|
| 112 |
)
|
| 113 |
|
| 114 |
# retriever = self.retrieve_vector_store.as_retriever()
|
| 115 |
-
retriever = self.retrieve_vector_store.as_retriever(search_kwargs={"k":
|
| 116 |
|
| 117 |
prompt = PromptTemplate.from_template(template="{context}", template_format="f-string")
|
| 118 |
combine_docs = create_stuff_documents_chain(llm=hf_llm, prompt=prompt, )
|
|
|
|
| 112 |
)
|
| 113 |
|
| 114 |
# retriever = self.retrieve_vector_store.as_retriever()
|
| 115 |
+
retriever = self.retrieve_vector_store.as_retriever(search_kwargs={"k": 1})
|
| 116 |
|
| 117 |
prompt = PromptTemplate.from_template(template="{context}", template_format="f-string")
|
| 118 |
combine_docs = create_stuff_documents_chain(llm=hf_llm, prompt=prompt, )
|