Merge pull request 'ollama local chat completion' (#11) from ollama-gtx into main

Reviewed-on: #11

garfmain.py

@@ -117,30 +117,30 @@ async def garfbot_qr(ctx, *, text):
         f"QR Code Request - User: {ctx.author.name}, Server: {ctx.guild.name}, Text: {text}"
     )
     if len(text) > 1000:
-        await ctx.send("❌ Text too long! Maximum 1000 characters.")
+        await ctx.reply("❌ Text too long! Maximum 1000 characters.")
     else:
         try:
             qr_code = await generate_qr(text)
             sendfile = discord.File(fp=qr_code, filename="qrcode.png")
-            await ctx.send(file=sendfile)
+            await ctx.reply(file=sendfile)
         except Exception as e:
             logger.error(e)
-            await ctx.send(e)
+            await ctx.reply(e)
 
 
 @garfbot.command(name="wiki")
 async def garfbot_wiki(ctx, *, query):
     summary = await garfield.wikisum(query)
-    await ctx.send(summary)
+    await ctx.reply(summary)
 
 
 @garfbot.command(name="shop")
 async def garfbot_shop(ctx, *, query):
     try:
         response = kroger.garfshop(query)
-        await ctx.send(response)
+        await ctx.reply(response)
     except Exception as e:
-        await ctx.send(f"`GarfBot Error: {str(e)}`")
+        await ctx.reply(f"`GarfBot Error: {str(e)}`")
 
 
 @garfbot.command(name="weather")
@@ -153,20 +153,21 @@ async def garfchat(ctx, *, prompt):
     if "is this true" in prompt.lower():
         messages = [msg async for msg in ctx.channel.history(limit=2)]
         prompt = messages[1].content
+        prompt = f"Is this true: {prompt}"
     answer = await garfield.generate_chat(prompt)
     logger.info(
         f"Chat Request - User: {ctx.author.name}, Server: {ctx.guild.name}, Prompt: {prompt}"
     )
-    await ctx.send(answer)
+    await ctx.reply(answer)
 
 
-@garfbot.command(name="pic")
-async def garfpic(ctx, *, prompt):
-    logger.info(
-        f"Image Request - User: {ctx.author.name}, Server: {ctx.guild.name}, Prompt: {prompt}"
-    )
-    await ctx.send(f"`Please wait... image generation queued: {prompt}`")
-    await garfield.garfpic(ctx, prompt)
+# @garfbot.command(name="pic")
+# async def garfpic(ctx, *, prompt):
+#     logger.info(
+#         f"Image Request - User: {ctx.author.name}, Server: {ctx.guild.name}, Prompt: {prompt}"
+#     )
+#     await ctx.reply(f"`Please wait... image generation queued: {prompt}`")
+#     await garfield.garfpic(ctx, prompt)
 
 
 @garfbot.command(name="help")

garfpy/garfai.py

@@ -11,7 +11,9 @@ from garfpy import logger
 
 class GarfAI:
     def __init__(self):
+        self.baseurl = config.BASE_URL
         self.openaikey = config.OPENAI_TOKEN
+        self.sysprompt = config.SYSTEM_PROMPT
         self.txtmodel = config.TXT_MODEL
         self.imgmodel = config.IMG_MODEL
         self.image_request_queue = asyncio.Queue()
@@ -80,17 +82,21 @@ class GarfAI:
 
     async def generate_chat(self, question):
         try:
-            client = AsyncOpenAI(api_key=self.openaikey)
+            client = AsyncOpenAI(
+                api_key=self.openaikey,
+                base_url=self.baseurl
+            )
             response = await client.chat.completions.create(
                 model=self.txtmodel,
                 messages=[
                     {
                         "role": "system",
-                        "content": "Pretend you are sarcastic Garfield.",
+                        "content": self.sysprompt,
                     },
-                    {"role": "user", "content": f"{question}"},
+                    {"role": "user", "content": question},
                 ],
                 max_tokens=400,
+                temperature=1.2,
             )
             answer = str(response.choices[0].message.content)
             return answer.replace("an AI language model", "a cartoon animal")

garfpy/help.py

@@ -6,9 +6,9 @@ async def help(message):
     embed.add_field(
         name="hey garfield `prompt`", value="*Responds with text.*", inline=True
     )
-    embed.add_field(
-        name="garfpic `prompt`", value="*Responds with an image.*", inline=True
-    )
+    # embed.add_field(
+    #     name="garfpic `prompt`", value="*Responds with an image.*", inline=True
+    # )
     embed.add_field(
         name="garfping `target`",
         value="*Responds with iputils-ping result from target.*",