xtex/akari-bot
Archived
1
0
Fork 0
Code Issues Pull requests Activity

Update summary 2 openai 1.0.0 api interface

Browse source
This commit is contained in:
多羅狼 2024-01-03 23:58:35 +08:00 committed by GitHub
parent 896a78087d
commit 91ea824350
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
modules/summary/__init__.py
View file

@ -1,4 +1,4 @@
import openai from openai import AsyncOpenAI
import re import re
from config import Config from config import Config
@ -10,7 +10,9 @@ from core.logger import Logger
from core.petal import count_petal from core.petal import count_petal
from core.utils.cooldown import CoolDown from core.utils.cooldown import CoolDown
openai.api_key = Config('openai_api_key') client = AsyncOpenAI(
api_key=Config('openai_api_key'),
)
s = module('summary', s = module('summary',
developers=['Dianliang233', 'OasisAkari'], developers=['Dianliang233', 'OasisAkari'],
@ -61,7 +63,7 @@ async def _(msg: Bot.MessageSession):
nth += 1 nth += 1
else: else:
break break
completion = openai.ChatCompletion.create( completion = await client.chat.completions.create(
model='gpt-3.5-turbo', model='gpt-3.5-turbo',
messages=[ messages=[
{'role': 'system', "content": "You are a helpful assistants who summarizes chat logs."}, {'role': 'system', "content": "You are a helpful assistants who summarizes chat logs."},
@ -70,8 +72,8 @@ async def _(msg: Bot.MessageSession):
{post_texts}'''}, {post_texts}'''},
] ]
) )
output = completion['choices'][0]['message']['content'] output = completion.choices[0].message.content
tokens = completion['usage']['total_tokens'] tokens = completion.usage.total_tokens
if not is_superuser: if not is_superuser:
petal = await count_petal(tokens) petal = await count_petal(tokens)
msg.data.modify_petal(-petal) msg.data.modify_petal(-petal)