Pash1986 commited on
Commit
b772595
·
verified ·
1 Parent(s): 336bda4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -41,7 +41,7 @@ except:
41
  def get_movies(message, history):
42
 
43
  try:
44
- movies = vector_store.similarity_search(message, 3)
45
  return_text = ''
46
  for movie in movies:
47
  return_text = return_text + 'Title : ' + movie.metadata['title'] + '\n------------\n' + 'Plot: ' + movie.page_content + '\n\n'
 
41
  def get_movies(message, history):
42
 
43
  try:
44
+ movies = vector_store.similarity_search(query=message, k=3, embedding_key='plot_embedding')
45
  return_text = ''
46
  for movie in movies:
47
  return_text = return_text + 'Title : ' + movie.metadata['title'] + '\n------------\n' + 'Plot: ' + movie.page_content + '\n\n'