74 lines
2.0 KiB
Python
74 lines
2.0 KiB
Python
import os
|
|
import nextcord
|
|
from nextcord.ext import commands
|
|
import requests
|
|
import openai
|
|
import json
|
|
|
|
SquogAILink = "https://api.webraft.in/"
|
|
SquogAIToken = "wr-tmkosxfStCjKSKsUJZueAS"
|
|
|
|
SquogAI = openai.OpenAI(api_key=SquogAIToken, base_url=SquogAILink)
|
|
|
|
SquogDataFile = open("./userinfo/userinfo.json", "r")
|
|
SquogData = json.loads(SquogDataFile.read())
|
|
SquogDataFile.close()
|
|
|
|
SquogUserPreset = {"id": 999, "messages": []}
|
|
|
|
async def getResponse(SquogMessageList):
|
|
return SquogAI.chat.completions.create(messages=SquogMessageList, model="gpt-4o", temperature=0.6)
|
|
|
|
async def find(list: list, param: str, value: any):
|
|
found = None
|
|
for g in list:
|
|
print(g)
|
|
if g[param] == value:
|
|
found = g
|
|
return found
|
|
|
|
async def flush(data):
|
|
SquogDataFile = open("./userdata/userdata.json", "w")
|
|
SquogDataFile.write(json.dumps(SquogData))
|
|
SquogDataFile.flush()
|
|
SquogDataFile.close()
|
|
|
|
class AI(commands.Cog):
|
|
def __init__(self, bot: commands.Bot):
|
|
self.client = bot
|
|
|
|
@commands.Cog.listener()
|
|
async def on_message(self, message: nextcord.Message):
|
|
print(message.content)
|
|
await self.client.process_commands(message)
|
|
|
|
SquogMention = False
|
|
|
|
for Mention in message.mentions:
|
|
if Mention.id == self.client.user.id:
|
|
SquogMention = True
|
|
|
|
if not SquogMention:
|
|
return
|
|
|
|
SquogUser = await find(SquogData, "id", message.author.id)
|
|
|
|
if not SquogUser:
|
|
SquogUser = SquogUserPreset.copy()
|
|
SquogUser["id"] = message.author.id
|
|
SquogUser["messages"] = []
|
|
SquogData.append(SquogUser)
|
|
|
|
SquogUser["messages"].append({"role": "user", "content": message.content})
|
|
|
|
SquogResponse = await getResponse(SquogUser["messages"])
|
|
print(SquogResponse)
|
|
|
|
SquogUser["messages"].append(SquogResponse["choices"][0]["message"])
|
|
|
|
await message.reply(SquogResponse["choices"][0]["message"]["content"])
|
|
|
|
await flush(SquogData)
|
|
|
|
def setup(bot):
|
|
bot.add_cog(AI(bot)) |