benjika commited on
Commit
0f7ccc9
·
verified ·
1 Parent(s): 7cb0a3b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -102,7 +102,7 @@ def clean_context(context_list, max_tokens=350, min_length=50):
102
 
103
 
104
  def get_k8s_answer(query):
105
- retrieved_context = retriever.get_relevant_documents(query)
106
  cleaned_context = clean_context(retrieved_context, max_tokens=350)
107
 
108
  input_text = prompt_template.format(context=cleaned_context, input=query)
 
102
 
103
 
104
  def get_k8s_answer(query):
105
+ retrieved_context = retriever.invoke(query)
106
  cleaned_context = clean_context(retrieved_context, max_tokens=350)
107
 
108
  input_text = prompt_template.format(context=cleaned_context, input=query)