Update rag.py
Browse files
rag.py
CHANGED
@@ -115,7 +115,7 @@ class RAG:
|
|
115 |
chat_completion = client.chat.completions.create(
|
116 |
model="tgi",
|
117 |
messages=[
|
118 |
-
{"role": "user", "content":
|
119 |
],
|
120 |
temperature=model_parameters["temperature"],
|
121 |
max_tokens=model_parameters["max_new_tokens"],
|
|
|
115 |
chat_completion = client.chat.completions.create(
|
116 |
model="tgi",
|
117 |
messages=[
|
118 |
+
{"role": "user", "content": query}
|
119 |
],
|
120 |
temperature=model_parameters["temperature"],
|
121 |
max_tokens=model_parameters["max_new_tokens"],
|