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

Turn gpt back on

Browse Source
This commit is contained in:
Luke Robles 2024-04-08 13:44:45 -07:00
parent ee11595cb7
commit b8bdb95acb
1 changed files with 34 additions and 38 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

72
app/bot.py
View File

@ -151,47 +151,43 @@ async def fix_social_media_links(ctx):
return
# @bot.event
# async def on_message(ctx):
# if str(bot.user.id) in ctx.content:
@bot.event
async def on_message(ctx):
if str(bot.user.id) in ctx.content:
# url = "http://192.168.1.52:1337/v1/chat/completions"
# bot_prompt = (
# "You are a helpful assistant. You will answer questions conciesely"
# "and as detailed as possible. You are based out of the United states in California."
# )
url = "http://192.168.1.52:1337/v1/chat/completions"
bot_prompt = (
"You are a helpful assistant. You will answer questions conciesely"
"and as detailed as possible. You are based out of the United states in California."
)
# payload = {
# "messages": [
# {
# "content": bot_prompt,
# "role": "system",
# },
# {
# "content": ctx.content.replace(str(bot.user.id), "").replace(
# "<@> ", ""
# ),
# "role": "user",
# },
# ],
# "model": "openchat-3.5-7b",
# "stream": False,
# "max_tokens": 4096,
# "stop": ["hello"],
# "frequency_penalty": 0,
# "presence_penalty": 0,
# "temperature": 0.7,
# "top_p": 0.95,
# }
# headers = {"Content-Type": "application/json"}
payload = {
"messages": [
{
"content": bot_prompt,
"role": "system",
},
{
"content": ctx.content.replace(str(bot.user.id), "").replace(
"<@> ", ""
),
"role": "user",
},
],
"model": "openchat-3.5-7b",
"stream": False,
"max_tokens": 4096,
"stop": ["hello"],
"frequency_penalty": 0,
"presence_penalty": 0,
"temperature": 0.7,
"top_p": 0.95,
}
headers = {"Content-Type": "application/json"}
# response = requests.post(url, json=payload, headers=headers)
# await ctx.reply(
# response.json()["choices"][0]["message"]["content"].replace(
# "<|end_of_turn|>", ""
# )
# )
response = requests.post(url, json=payload, headers=headers)
answer = response.json()["choices"][0]["message"]["content"]
await ctx.reply(answer.replace("<|end_of_turn|>", ""))
bot.run(os.getenv("discord_token"))