s

app.py

@@ -34,17 +34,18 @@ async def on_message(message):
         async for message in channel.history(limit=10):
             messages.append(message.content)
         previous_messages = ("\n".join(messages))
+
         response = ai.text_generation(previous_messages, temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0)
-        splited_response = split_string(response)
-        for part in splited_response:
+        splitted_response = split_string(response)
+        for part in splitted_response:
             await channel.send(part)
 
         message_counts[message.channel.id] = 0 # Reset the counter
 
 
-    response = ai.text_generation(message, temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0)
-    splited_response = split_string(str(response))
-    for part in splited_response:
+    response = str(ai.text_generation(message, details=False, stream=False, temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0))
+    splitted_response = split_string(response)
+    for part in splitted_response:
         await channel.send(part)
 
     await bot.process_commands(message) # Ensure commands still work