refactor: Enhance message processing by extracting code from attachments and updating GIF usage guidelines

Author: 0-don

src/events/ai/ai.ts

@@ -14,6 +14,7 @@ import {
   makeImageParts,
   MAX_MESSAGES_PER_CHANNEL,
   TOOLS,
+  extractCodeFromAttachments,
 } from "./utils";
 
 @Discord()
@@ -50,6 +51,15 @@ export class AiChat {
 
     const { replyContext, repliedImages } = await this.getReplyContext(message);
 
+    // Extract code from current message attachments
+    let attachmentContext = "";
+    if (message.attachments.size > 0) {
+      const codeContent = await extractCodeFromAttachments(message);
+      if (codeContent) {
+        attachmentContext = `\n\n[Code from attachment]:\n${codeContent}`;
+      }
+    }
+
     if (
       !userMsg &&
       message.attachments.size === 0 &&
@@ -64,7 +74,7 @@ export class AiChat {
 
     // Get user context
     const userContext = await this.getUserContext(message.author.id, message);
-    const fullMessage = `${userMsg}${replyContext}${userContext}`;
+    const fullMessage = `${userMsg}${attachmentContext}${replyContext}${userContext}`;
 
     // Retry logic with exponential backoff
     let lastError: Error | null = null;
@@ -229,7 +239,6 @@ export class AiChat {
         // Get context for the replied user too
         const repliedUserContext = await this.getUserContext(
           repliedUser.id,
-
           message
         );
 
@@ -280,4 +289,4 @@ export class AiChat {
     }
     return null;
   }
-}
+}

src/events/ai/prompt.ts

@@ -1,15 +1,13 @@
-export const AI_SYSTEM_PROMPT = `You are Coding Global, the official Discord bot for the coding.global programming server (discord.gg/coding). Be sarcastic yet helpful and concise - dry humor but still useful.
+export const AI_SYSTEM_PROMPT = `You are Coding Global, the official Discord bot for the coding.global programming server (discord.gg/coding). Be sarcastic yet helpful and concise few senteces max no matter the question - dry humor but still useful.
 
 PERSONALITY:
 - Never start with "Oh" - use varied openings
 - Sarcastic but not mean: "Sure, whatever" or "Here's your code..."
 - Use ellipses (...) for indifference
 - Professional programmers, trainees, students, and apprentices hang out here
-- FREQUENTLY use GIFs to enhance responses - search for relevant reaction GIFs when appropriate
-- Use GIFs for: reactions to code problems, celebrations, frustrations, programming humor, etc.
 
 GIF USAGE GUIDELINES:
-- Use GIFs sparingly - only when they genuinely enhance the conversation
+- Use GIFs sometimes - only when they genuinely enhance the conversation
 - Good for: major celebrations, epic fails, or when specifically asked
 - Avoid using GIFs for routine responses or simple questions
 - Never use GIFs just to fill space - substance over entertainment

src/events/ai/utils.ts

@@ -51,15 +51,24 @@ export async function gatherMessageContext(
 
     const contextParts = await Promise.all(
       allMessages.map(async (msg) => {
+        let content = msg.content?.trim() || "";
+
         if (msg.attachments.size > 0) {
           try {
             const msgImages = await makeImageParts(msg);
             images.push(...msgImages);
+
+            // Extract code from attachments
+            const codeContent = await extractCodeFromAttachments(msg);
+            if (codeContent) {
+              content += `\n\n[Code from attachment]:\n${codeContent}`;
+            }
           } catch (error) {
-            console.error("Error processing images:", error);
+            console.error("Error processing attachments:", error);
           }
         }
-        return msg.content?.trim() || "";
+
+        return content;
       })
     );
 
@@ -71,6 +80,55 @@ export async function gatherMessageContext(
   }
 }
 
+export async function extractCodeFromAttachments(
+  message: Message
+): Promise<string | null> {
+  let extractedCode = "";
+
+  for (const attachment of message.attachments.values()) {
+    const { name, url, contentType, size } = attachment;
+
+    // Skip if file is too large (>1MB)
+    if (size && size > 1024 * 1024) {
+      console.log(`Skipping large file: ${name} (${size} bytes)`);
+      continue;
+    }
+
+    // Check if it's text-based content
+    if (contentType?.startsWith("text/") || !contentType) {
+      try {
+        console.log(`Fetching code from attachment: ${name}`);
+        const response = await fetch(url);
+
+        if (!response.ok) {
+          console.error(`Failed to fetch ${name}: ${response.status}`);
+          continue;
+        }
+
+        const content = await response.text();
+
+        // Limit content length to prevent token overflow
+        const maxLength = 8000;
+        const truncatedContent =
+          content.length > maxLength
+            ? content.substring(0, maxLength) + "\n... (content truncated)"
+            : content;
+
+        if (extractedCode) {
+          extractedCode += `\n\n--- File: ${name} ---\n`;
+        } else {
+          extractedCode += `--- File: ${name} ---\n`;
+        }
+        extractedCode += truncatedContent;
+      } catch (error) {
+        console.error(`Error fetching attachment ${name}:`, error);
+      }
+    }
+  }
+
+  return extractedCode || null;
+}
+
 export async function makeImageParts(message: Message): Promise<string[]> {
   const images: string[] = [];