Update app.py
Browse files
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)}
|