From 166b286d42c7f7bb94e6fa71701c4a58e7e8df0f Mon Sep 17 00:00:00 2001
From: milo <miguelloy97@gmail.com>
Date: Wed, 7 May 2025 17:20:34 -0400
Subject: [PATCH] Bot can how reply when using the "!chat" command then
 prompting it. The .env file points to an ollama instance, the ai.py file has
 the model being used.

---
 .env                           |   2 +-
 __pycache__/ai.cpython-310.pyc | Bin 0 -> 738 bytes
 ai.py                          |  25 +++++++++++++++++++++++++
 bot.py                         |   9 +++++++++
 4 files changed, 35 insertions(+), 1 deletion(-)
 create mode 100644 __pycache__/ai.cpython-310.pyc
 create mode 100644 ai.py

diff --git a/.env b/.env
index 93137f0..ceff269 100644
--- a/.env
+++ b/.env
@@ -1,2 +1,2 @@
 DISCORD_TOKEN=MTM2OTc3NDY4OTYzNDg4MTU4Ng.G9Nrgz.akHoOO9SrXCDwiOCI3BUXfdR4bpSNb9zrVx9UI
-OLLAMA_API_URL=https://your-ollama-api-url.com/v1/models/your-model-name/generate
+OLLAMA_API_URL=http://192.168.1.100:11434/api/generate
diff --git a/__pycache__/ai.cpython-310.pyc b/__pycache__/ai.cpython-310.pyc
new file mode 100644
index 0000000000000000000000000000000000000000..25c35616ddc9a62adc2593631dca83622e3445c3
GIT binary patch
literal 738
zcmYjPL2DC16rMLbyV)d7M9>~oWbx!)yeg%L!B!MvD=kG#w}fqGtXp?8>&&EWmPCa1
z>d{jb(xd;xi+^CQp86LA(Ko4u4$SxX-n=*OdoxeAS`C8h{?FU*zEDDbIdXM*NS@%<
zj}SQFkdxkpg&rWMIW4@N2Sj+s1QBM8duPz|Iph8b=>;4%NgSMGHd%|oHglQsK9@!e
z4|?tOZnyKY)9<``(cjwW4!}kIUp`tV?I~`3g1}*d<+%6Nn?iNZkbN+80-SxOQ#zsh
z9{1l<b$3E%?2Jski8u3kz-wppHQ6LQbP=b7*U<*2<QsEdMoewo0WxJksupjUaKerO
zPcbJTZ?MfN`^YeRZV%y*#Lcy;QDoYvG+%nS`z-dXUq~);8<a{GrLh5?iL}5@*}j&;
z^B<_(hDzvC4z;LS+iNnqWII)DdsQi^qKfWB)%EQS@nIyiiQcAJF8EFq*Inw|r!6a`
z#``<-r?O3LQZv&10ajOmE8^`5W0?_0#@1Jl2BI{X99j?c#J*xKqE@D3_xvh!EI=8f
z+*lCbiMoNF-^+ugWJ?RBlOo$2iDZ2!R#bK%k`*}^6=G;~@>~`o>ZZF1&-6el9_>n#
zq}hXVY_DPcewy_!yK|qX?;uc0fkN<y25o3S2yIxTZMX&DC;C?{VnAF|bu^YrZG(%^
sP&kVy5wFBKQB5TC$hpodaMMMF%lHMOTs(H;qZbfD8o~kuv;hnBACI-ZH~;_u

literal 0
HcmV?d00001

diff --git a/ai.py b/ai.py
new file mode 100644
index 0000000..dc9c632
--- /dev/null
+++ b/ai.py
@@ -0,0 +1,25 @@
+# ai.py
+
+import requests
+import os
+from dotenv import load_dotenv
+
+load_dotenv()
+AI_URL = os.getenv("OLLAMA_API_URL")
+
+def get_ai_response(prompt):
+    payload = {
+        "model": "mistral:7b",  # Adjust to match your model
+        "prompt": prompt,
+        "stream": False
+    }
+
+    try:
+        response = requests.post(AI_URL, json=payload)
+        if response.status_code == 200:
+            result = response.json()
+            return result.get("response", "[No response]")
+        else:
+            return f"[Error {response.status_code}] {response.text}"
+    except Exception as e:
+        return f"[Request failed] {str(e)}"
diff --git a/bot.py b/bot.py
index f120523..dd12767 100644
--- a/bot.py
+++ b/bot.py
@@ -2,6 +2,8 @@ import os
 import discord
 from discord.ext import commands
 from dotenv import load_dotenv
+from ai import get_ai_response
+
 
 load_dotenv()
 TOKEN = os.getenv("DISCORD_TOKEN")
@@ -19,4 +21,11 @@ async def on_ready():
 async def ping(ctx):
     await ctx.send("🏓 Pong!")
 
+@bot.command()
+async def chat(ctx, *, message):
+    await ctx.send("🤖 Thinking...")
+    reply = get_ai_response(message)
+    await ctx.send(reply)
+
+
 bot.run(TOKEN)