From 7b010bfd0f74d88146702007af4e9da4db334a7c Mon Sep 17 00:00:00 2001
From: "Edgar P. Burkhart" <git@edgarpierre.fr>
Date: Sat, 22 Mar 2025 17:07:12 +0100
Subject: [PATCH] Fix async response handling in indu function and update chat
completion method
---
botbotbot/__main__.py | 6 +++---
botbotbot/ai.py | 2 +-
2 files changed, 4 insertions(+), 4 deletions(-)
diff --git a/botbotbot/__main__.py b/botbotbot/__main__.py
index 237296f..9537f9f 100644
--- a/botbotbot/__main__.py
+++ b/botbotbot/__main__.py
@@ -264,7 +264,7 @@ async def voice_random_nicks(member, before, after):
async def indu(ctx, prompt):
logger.info(f"INDU {ctx.author} {prompt}")
await ctx.defer()
- res_stream = aibot.get_response_stream(prompt)
+ res_stream = await aibot.get_response_stream(prompt)
embed = discord.Embed(
title=prompt,
@@ -275,8 +275,8 @@ async def indu(ctx, prompt):
message = await ctx.respond(embed=embed)
async for chunk in res_stream:
- if chunk.choices[0].delta.content is not None:
- embed.description += chunk.choices[0].delta.content
+ if chunk.data.choices[0].delta.content is not None:
+ embed.description += chunk.data.choices[0].delta.content
await message.edit(embed=embed)
embed.color = None
diff --git a/botbotbot/ai.py b/botbotbot/ai.py
index 31c5538..eeb9ab2 100644
--- a/botbotbot/ai.py
+++ b/botbotbot/ai.py
@@ -11,7 +11,7 @@ class AIBot:
self.system_message = system_message
def get_responses(self, message):
- return self.client.chat(
+ return self.client.chat.complete(
model=self.model,
messages=self.base_message + [{"role": "user", "content": message}],
max_tokens=self.max_tokens,