diff --git a/src/classes/ai.ts b/src/classes/ai.ts index a101464..12413ea 100644 --- a/src/classes/ai.ts +++ b/src/classes/ai.ts @@ -5,30 +5,22 @@ */ import { Anthropic } from "@anthropic-ai/sdk"; -import { - GoogleGenAI, -} from "@google/genai"; -import { AttachmentBuilder, type MessageCreateOptions } from "discord.js"; +import type { MessageCreateOptions } from "discord.js"; /** * Utility class for generating project information and images. */ export class Ai { private readonly anthropic: Anthropic; - private readonly gemini: GoogleGenAI; /** * Creates a new instance of the Ai class. * @param anthropicKey - The API key for the Anthropic API. - * @param geminiKey - The API key for the Gemini API. */ - public constructor(anthropicKey: string, geminiKey: string) { + public constructor(anthropicKey: string) { this.anthropic = new Anthropic({ apiKey: anthropicKey, }); - this.gemini = new GoogleGenAI({ - apiKey: geminiKey, - }); } /** @@ -49,19 +41,11 @@ export class Ai { `Your task is to generate a project name based on the user's description. Provide ONLY a list of 1-5 fitting names, and an explanation for why you chose them. Note that project names should be unique. Here's a list of all existing project names: ${projectResponse. map((p) => { return p.name; - }). - join(", ")}`, + }).join(", ")}`, prompt, ); - const image = await this.generateImage(prompt); - if (image === null) { - return { - content: `Project Name: ${names}\nProject Description: ${prompt}\nSorry, I was unable to generate an image for you.`, - }; - } return { content: `Project Name: ${names}\nProject Description: ${prompt}`, - files: [ new AttachmentBuilder(image, { name: "avatar.png" }) ], }; } @@ -88,24 +72,4 @@ export class Ai { join(""); return text; } - - private async generateImage(prompt: string): Promise { - const response = await this.gemini.models.generateContent({ - config: { - imageConfig: { aspectRatio: "3:4" }, - systemInstruction: - `Your task is to generate a full body anime girl mascot for this project. This means the full character should be visible. The image should have a white background. NEVER include text in the image, no text anywhere at all. The project description is provided by the user.`, - }, - contents: prompt, - model: "gemini-2.5-flash-image", - }); - const image = response.candidates?.[0]?.content?.parts?.find((p) => { - return Boolean(p.inlineData); - }); - const base64 = image?.inlineData?.data; - if (base64 === undefined) { - return null; - } - return Buffer.from(base64, "base64"); - } }