Update README.md

README.md

@@ -107,11 +107,11 @@ contexts = [
 
 ## convert query into a format as follows:
 ## user: {user}\nagent: {agent}\nuser: {user}
-formatted_query = '\n'.join([turn['role'] + ": " + turn['content'] for turn in messages]).strip()
+formatted_query = '\n'.join([turn['role'] + ": " + turn['content'] for turn in query]).strip()
 
 ## get query and context embeddings
 query_input = tokenizer(formatted_query, return_tensors='pt')
-ctx_input = tokenizer(contexts, padding=True, return_tensors='pt')
+ctx_input = tokenizer(contexts, padding=True, truncation=True, max_length=512, return_tensors='pt')
 query_emb = query_encoder(**query_input).last_hidden_state[:, 0, :] # (1, emb_dim)
 ctx_emb = context_encoder(**ctx_input).last_hidden_state[:, 0, :] # (num_ctx, emb_dim)