Update rag.py

rag.py

@@ -51,10 +51,11 @@ class RAG:
                 #{"role": "system", "content": sys_prompt },
                 {"role": "user", "content": query}
             ],
+            max_tokens=model_parameters['max_new_tokens'] # TODO: map other parameters
             stream=False
         )
         print(chat_completion)
-        return(chat_completion.choices[0].delta.content)
+        return(chat_completion.choices[0].message.content)
 
     
     def beautiful_context(self, docs):