devve1 commited on
Commit
7623b62
1 Parent(s): 6fe701f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -100,6 +100,7 @@ def main(query: str, client: QdrantClient, collection_name: str, tokenizer: Auto
100
  if (value not in seen_values and not seen_values.add(value))
101
  )
102
 
 
103
  messages = [
104
  {"role": "system", "content": 'You are a helpful assistant.'},
105
  {"role": "user", "content": st.session_state.toggle_docs['qa_prompt'].format(**args)}
 
100
  if (value not in seen_values and not seen_values.add(value))
101
  )
102
 
103
+ args = {'context': context, 'query': query}
104
  messages = [
105
  {"role": "system", "content": 'You are a helpful assistant.'},
106
  {"role": "user", "content": st.session_state.toggle_docs['qa_prompt'].format(**args)}