chirag0107 commited on
Commit
e5b2909
·
verified ·
1 Parent(s): 6188af9

Update langchain_movie_search.py

Browse files
Files changed (1) hide show
  1. langchain_movie_search.py +1 -1
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": 13})
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, )