Update langchain_movie_search.py

langchain_movie_search.py

@@ -111,7 +111,9 @@ class MoviesSearch:
             return_full_text=True,
         )
 
-        retriever = self.retrieve_vector_store.as_retriever()
+        # retriever = self.retrieve_vector_store.as_retriever()
+        retriever = self.retrieve_vector_store.as_retriever(search_kwargs={"k": 3})
+        
         prompt = PromptTemplate.from_template(template="{context}", template_format="f-string")
         combine_docs = create_stuff_documents_chain(llm=hf_llm, prompt=prompt, )