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,