70 lines
2.0 KiB
Python
70 lines
2.0 KiB
Python
import os
|
|
|
|
import discord
|
|
from dotenv import load_dotenv
|
|
|
|
load_dotenv()
|
|
|
|
from honcho import Client as HonchoClient
|
|
from honcho import LRUCache
|
|
|
|
intents = discord.Intents.default()
|
|
intents.messages = True
|
|
intents.message_content = True
|
|
|
|
CACHE = LRUCache(50) # Support 50 concurrent active conversations cached in memory
|
|
|
|
honcho = HonchoClient("http://localhost:8000")
|
|
|
|
bot = discord.Bot(intents=intents)
|
|
|
|
def get_or_create(user_id, location_id) -> int:
|
|
key = f"{user_id}+{location_id}"
|
|
session_id = CACHE.get(key)
|
|
if session_id is None:
|
|
session = honcho.create_session(user_id, location_id)
|
|
print(session)
|
|
session_id = session["id"]
|
|
CACHE.put(key, session_id)
|
|
return session_id
|
|
|
|
@bot.event
|
|
async def on_ready():
|
|
print(f'We have logged in as {bot.user}')
|
|
|
|
@bot.event
|
|
async def on_message(message):
|
|
if message.author == bot.user:
|
|
return
|
|
|
|
user_id = f"discord_{str(message.author.id)}"
|
|
location_id=str(message.channel.id)
|
|
# key = f"{user_id}+{location_id}"
|
|
|
|
session_id = get_or_create(user_id, location_id)
|
|
|
|
inp = message.content
|
|
honcho.create_message_for_session(user_id, session_id, True, inp)
|
|
async with message.channel.typing():
|
|
await message.channel.send("Fake LLM Message")
|
|
|
|
honcho.create_message_for_session(user_id, session_id, False, "Fake LLM Message")
|
|
|
|
@bot.slash_command(name = "restart", description = "Restart the Conversation")
|
|
async def restart(ctx):
|
|
user_id=f"discord_{str(ctx.author.id)}"
|
|
location_id=str(ctx.channel_id)
|
|
key = f"{user_id}+{location_id}"
|
|
session_id = CACHE.get(key)
|
|
if session_id is not None:
|
|
honcho.delete_session(user_id, session_id)
|
|
session = honcho.create_session(user_id, location_id)
|
|
session_id = session["id"]
|
|
CACHE.put(key, session_id)
|
|
|
|
msg = "Great! The conversation has been restarted. What would you like to talk about?"
|
|
honcho.create_message_for_session(user_id, session_id, False, msg)
|
|
await ctx.respond(msg)
|
|
|
|
bot.run(os.environ["BOT_TOKEN"])
|