Add main.py
This commit is contained in:
commit
996e23dbf6
|
@ -0,0 +1,32 @@
|
|||
import asyncio
|
||||
import os
|
||||
import discord
|
||||
from characterai import aiocai
|
||||
|
||||
intents = discord.Intents(messages=True, message_content=True, guilds=True)
|
||||
client = aiocai.Client(CHAR_AI_TOKEN)
|
||||
char = CHAT_ID
|
||||
|
||||
|
||||
class MyClient(discord.Client):
|
||||
channel = discord.channel
|
||||
|
||||
async def on_message(self, message) -> None:
|
||||
if message.channel.id == CHANNEL and message.author.id != BOT_ID:
|
||||
text = await self.push_to_char_ai(message)
|
||||
print(text)
|
||||
await self.channel.send(text.text)
|
||||
|
||||
async def push_to_char_ai(self, message) -> str:
|
||||
async with await client.connect() as chat:
|
||||
data = await chat.send_message(char=char,
|
||||
text=f"{message.author.name} says: {message.content}",
|
||||
chat_id=CHAT_ID)
|
||||
return data
|
||||
|
||||
async def on_ready(self) -> None:
|
||||
self.channel = self.get_channel(CHANNEL)
|
||||
|
||||
|
||||
bot = MyClient(intents=intents)
|
||||
bot.run(BOT_TOKEN)
|
Loading…
Reference in New Issue