dc endpoint and slash commands support

discord_bot/discord_bot/bot.py

@@ -3,6 +3,8 @@
 import os
 from collections import defaultdict
 from typing import Self
+import requests
+import json
 
 import discord
 from discord.ext import commands
@@ -40,8 +42,29 @@ async def bad_button(
         )
 
 
+@bot.command()
+@commands.has_any_role("Admins", "Moderators")
+async def sync(ctx) -> None:
+    await ctx.send("Synchronizing commands...")
+    await bot.tree.sync()
+
+
+def query_llm(prompt, stop_signs):
+    """Returns llm response"""
+    url = "http://llm:9000/v1/completions"
+    headers = {"Content-Type": "application/json"}
+    data = {"prompt": prompt, "stop": stop_signs}
+
+    response = requests.post(url, headers=headers, data=json.dumps(data))
+
+    if response.status_code == 200:
+        return response.json()
+    else:
+        return response.text
+
+
 async def get_chats_history():
-    """Taking chat conversation from all chanells."""
+    """Taking chat conversation from all channels."""
     chats_history = defaultdict(list)
     for guild in bot.guilds:
         readable_channels = filter(
@@ -58,7 +81,7 @@ async def get_chats_history():
 
 @bot.command()
 async def show(ctx: commands.Context, limit: int = 100):
-    """Shows what get_chats_history gets."""
+    """Shows the results of get_chats_history"""
     last_messages = await get_chats_history()
     channel_id = ctx.channel.id
     if last_messages[channel_id]:
@@ -68,6 +91,19 @@ async def show(ctx: commands.Context, limit: int = 100):
         await ctx.send("Brak ostatnich wiadomości.")
 
 
+@bot.tree.command(name="chatknml", description="Porozmawiaj z chatbotem")
+async def chatknml(interaction: discord.Interaction, *, prompt: str):
+    """Passes the prompt to the llm and returns the answer."""
+    await interaction.response.defer()
+
+    query = "\n\n### Instructions:\n" + prompt + "\n\n### Response:\n"
+    stop_signs = ["\n", "###"]
+
+    result = query_llm(query, stop_signs)
+
+    await interaction.followup.send(result["choices"][0]["text"])
+
+
 def main():
     """Entrypoint."""
     bot.run(os.environ["TOKEN"])