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