ldooks/dragon-bot
ldooks/dragon-bot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add a try/except to the llm
All checks were successful
Build and push / changes (push) Successful in 33s
Details
Build and push / Lint-Python (push) Successful in 6s
Details
Build and push / Build-and-Push-Docker (push) Successful in 1m12s
Details
Build and push / sync-argocd-app (push) Successful in 3s
Details

Browse Source
This commit is contained in:
Luke R 2024-06-06 08:03:07 -07:00
parent c17c5d6dd9
commit 7e4b561a6f
1 changed files with 11 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
app/bot.py
View File

@ -158,6 +158,7 @@ async def on_message(ctx):
} }
headers = {"Content-Type": "application/json"} headers = {"Content-Type": "application/json"}
try:
await ctx.channel.trigger_typing() await ctx.channel.trigger_typing()
response = requests.post(url, json=payload, headers=headers) response = requests.post(url, json=payload, headers=headers)
answer = response.json()["choices"][0]["message"]["content"] answer = response.json()["choices"][0]["message"]["content"]
@ -167,6 +168,8 @@ async def on_message(ctx):
await ctx.reply(answer[2000:].replace("<|end_of_turn|>", "")) await ctx.reply(answer[2000:].replace("<|end_of_turn|>", ""))
else: else:
await ctx.reply(answer.replace("<|end_of_turn|>", "")) await ctx.reply(answer.replace("<|end_of_turn|>", ""))
except KeyError:
await ctx.reply("Somethings wrong, maybe the LLM crashed")
bot.run(os.getenv("discord_token")) bot.run(os.getenv("discord_token"))