edpibu/botbotbot
1
Fork 0
Code Issues Releases Activity

Fix async response handling in indu function and update chat completion method

Browse source
This commit is contained in:
Edgar P. Burkhart 2025-03-22 17:07:12 +01:00
parent 93077e7d49
commit 7b010bfd0f
Signed by: edpibu
GPG key ID: 9833D3C5A25BD227
2 changed files with 4 additions and 4 deletions

6
botbotbot/__main__.py
View file

@ -264,7 +264,7 @@ async def voice_random_nicks(member, before, after):
async def indu(ctx, prompt): async def indu(ctx, prompt):
logger.info(f"INDU {ctx.author} {prompt}") logger.info(f"INDU {ctx.author} {prompt}")
await ctx.defer() await ctx.defer()
res_stream = aibot.get_response_stream(prompt) res_stream = await aibot.get_response_stream(prompt)
embed = discord.Embed( embed = discord.Embed(
title=prompt, title=prompt,
@ -275,8 +275,8 @@ async def indu(ctx, prompt):
message = await ctx.respond(embed=embed) message = await ctx.respond(embed=embed)
async for chunk in res_stream: async for chunk in res_stream:
if chunk.choices[0].delta.content is not None: if chunk.data.choices[0].delta.content is not None:
embed.description += chunk.choices[0].delta.content embed.description += chunk.data.choices[0].delta.content
await message.edit(embed=embed) await message.edit(embed=embed)
embed.color = None embed.color = None

2
botbotbot/ai.py
View file

@ -11,7 +11,7 @@ class AIBot:
self.system_message = system_message self.system_message = system_message
def get_responses(self, message): def get_responses(self, message):
return self.client.chat( return self.client.chat.complete(
model=self.model, model=self.model,
messages=self.base_message + [{"role": "user", "content": message}], messages=self.base_message + [{"role": "user", "content": message}],
max_tokens=self.max_tokens, max_tokens=self.max_tokens,