Update rag.py

rag.py

@@ -51,7 +51,7 @@ class RAG:
                 #{"role": "system", "content": sys_prompt },
                 {"role": "user", "content": query}
             ],
-            max_tokens=model_parameters['max_new_tokens'] # TODO: map other parameters
+            max_tokens=model_parameters['max_new_tokens'], # TODO: map other parameters
             stream=False
         )
         print(chat_completion)