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

Detect failure

Browse source
This commit is contained in:
Dianliang233 2024-01-03 14:49:34 +08:00
parent 576e7d9de0
commit 6856b23904
1 changed files with 9 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
modules/ask/__init__.py
View file

@ -39,12 +39,12 @@ assistant = sync_client.beta.assistants.create(
model="gpt-3.5-turbo-1106" model="gpt-3.5-turbo-1106"
) )
assistant_gpt4 = sync_client.beta.assistants.create( # assistant_gpt4 = sync_client.beta.assistants.create(
name="AkariBot", # name="AkariBot",
instructions=INSTRUCTIONS, # instructions=INSTRUCTIONS,
tools=[{"type": "code_interpreter"}], # tools=[{"type": "code_interpreter"}],
model="gpt-4-1106-preview" # model="gpt-4-1106-preview"
) # )
a = module('ask', developers=['Dianliang233'], desc='{ask.help.desc}') a = module('ask', developers=['Dianliang233'], desc='{ask.help.desc}')
@ -78,7 +78,7 @@ async def _(msg: Bot.MessageSession):
]) ])
run = await client.beta.threads.runs.create( run = await client.beta.threads.runs.create(
thread_id=thread.id, thread_id=thread.id,
assistant_id=assistant_gpt4.id if gpt4 else assistant.id, assistant_id=assistant.id,
) )
while True: while True:
run = await client.beta.threads.runs.retrieve( run = await client.beta.threads.runs.retrieve(
@ -87,6 +87,8 @@ async def _(msg: Bot.MessageSession):
) )
if run.status == 'completed': if run.status == 'completed':
break break
elif run.status == 'failed':
raise RuntimeError(run.json())
await asyncio.sleep(1) await asyncio.sleep(1)
messages = await client.beta.threads.messages.list( messages = await client.beta.threads.messages.list(