diff --git a/handlers/davinci.py b/handlers/davinci.py index adbab8e..c664f56 100644 --- a/handlers/davinci.py +++ b/handlers/davinci.py @@ -26,18 +26,18 @@ async def send(message: types.Message): ) await message.reply(response['choices'][0]['text']) - # elif message.reply_to_message.bot.id: - # response = openai.Completion.create( - # model="text-davinci-003", - # prompt=message.text, - # temperature=0.7, - # max_tokens=1000, - # top_p=1.0, - # frequency_penalty=0.0, - # presence_penalty=0.6, - # stop=["сброс"] - # ) - # await message.reply(response['choices'][0]['text']) + elif message.reply_to_message.from_user.is_bot: + response = openai.Completion.create( + model="text-davinci-003", + prompt=message.text, + temperature=0.7, + max_tokens=1000, + top_p=1.0, + frequency_penalty=0.0, + presence_penalty=0.6, + stop=["сброс"] + ) + await message.reply(response['choices'][0]['text'])