i

main.py

@@ -120,7 +120,7 @@ def get_completion(prompt, model="solar-1-mini-chat"):
         messages=messages,
         temperature=0, # this is the degree of randomness of the model's output
     )
-    return response.choices[0].message.content
+    return response.choices[-1].message.content
 
 def get_completion_from_messages(messages, model="solar-1-mini-chat", temperature=0):
     response = client.chat.completions.create(
@@ -128,7 +128,7 @@ def get_completion_from_messages(messages, model="solar-1-mini-chat", temperatur
         messages=messages,
         temperature=temperature, # this is the degree of randomness of the model's output
     )
-    return response.choices[0].message.content
+    return response.choices[-1].message.content
 
 # Chat message component (renders a chat bubble)
 def ChatMessage(msg, user):