deno.json

@@ -1,6 +1,6 @@
 {
   "imports": {
-    "@openai/openai": "jsr:@openai/openai@^4.79.1",
+    "@slack/bolt": "npm:@slack/bolt@^3.22.0",
-    "@slack/bolt": "npm:@slack/bolt@^4.2.0"
+    "groq-sdk": "npm:groq-sdk@^0.7.0"
   }
 }

main.ts

@@ -1,8 +1,8 @@
 // SPDX-License-Identifier: WTFPL
 
 const {
-  /** https://openrouter.ai/settings/keys */
+  /** https://console.groq.com/docs/api-keys */
-  OPENROUTER_API_KEY = "",
+  GROQ_API_KEY,
   /** https://api.slack.com/apps → Basic Information ページにある Signing Secret */
   SLACK_SIGNING_SECRET = "",
   /** https://api.slack.com/apps → Permissions ページにある `xoxb-` から始まるボットトークン */
@@ -18,8 +18,8 @@ const system = `\
 句読点の無いフレンドリーな感じです。`;
 
 const n = 10;
-const model = "google/gemini-2.0-flash-exp:free";
+const textModel = "llama-3.3-70b-versatile";
-const baseURL = "https://openrouter.ai/api/v1";
+const visionModel = "llama-3.2-90b-vision-preview";
 const urlPattern = new URLPattern({ pathname: "*.*" });
 const separatorRegex = /[<>]|[^\p{L}\p{N}\p{P}\p{S}]+/u;
 const imageExtensionRegex = /[.](?:gif|png|jpe?g|webp)$/i;
@@ -32,30 +32,28 @@ https://dash.deno.com/playground/darazllm
     @darazllm /help     このテキストを表示
 
 使用する会話: 最新${n}件
-モデル: ${model}
+テキスト用モデル: ${textModel}
-ベースURL: ${baseURL}
+画像用モデル: ${visionModel}
 区切り文字: ${separatorRegex}
 画像拡張子: ${imageExtensionRegex}
 システムプロンプト:
 
 ${system}`;
 
-import OpenAI from "jsr:@openai/openai";
 import bolt from "npm:@slack/bolt";
+import { Groq } from "npm:groq-sdk";
 
-type Messages = Array<OpenAI.Chat.ChatCompletionMessageParam>;
+type Messages = Array<Groq.Chat.ChatCompletionMessageParam>;
 
-const openai = new OpenAI({
+const groq = new Groq({
-  baseURL,
+  apiKey: GROQ_API_KEY,
-  apiKey: OPENROUTER_API_KEY,
   timeout: 10_000,
 });
-
 const kv = await Deno.openKv();
 
 async function chat(messages: Messages): Promise<string | null> {
-  const res = await openai.chat.completions.create({
+  const res = await groq.chat.completions.create({
-    model,
+    model: textModel,
     messages: [{ role: "system", content: system }, ...messages],
   });
 
@@ -66,6 +64,25 @@ function isImageUrl(message: string): boolean {
   return urlPattern.test(message) && imageExtensionRegex.test(message);
 }
 
+/** llama-3.2-90b-vision-preview はシステムプロンプトに対応してない、かつ、画像は1枚まで */
+async function visionChat(
+  content: Array<Groq.Chat.ChatCompletionContentPart>,
+): Promise<string | null> {
+  const res = await groq.chat.completions.create({
+    model: visionModel,
+    messages: [
+      {
+        role: "user",
+        content: content.map((c) =>
+          c.type === "text" ? { ...c, text: `${system}\n\n${c.text}` } : c,
+        ),
+      },
+    ],
+  });
+
+  return res.choices[0].message.content;
+}
+
 const app = new bolt.App({
   token: SLACK_BOT_TOKEN,
   signingSecret: SLACK_SIGNING_SECRET,
@@ -93,7 +110,10 @@ app.message(async (c) => {
   const kve = await kv.get<Messages>(["channel", c.message.channel]);
   const messages = kve.value ?? [];
 
-  const content: Array<OpenAI.Chat.ChatCompletionContentPart> = prompt
+  if (prompt) messages.push({ role: "user", content: prompt });
+
+  if (isMention) {
+    const content: Array<Groq.Chat.ChatCompletionContentPart> = prompt
       .split(separatorRegex)
       .filter(Boolean)
       .map((text) => {
@@ -104,11 +124,12 @@ app.message(async (c) => {
         }
       });
 
-  if (content.length > 0) messages.push({ role: "user", content });
+    const visionMode = content.some((c) => c.type === "image_url");
 
-  if (isMention) {
     try {
-      const res = await chat(messages.slice(-n));
+      const res = visionMode
+        ? await visionChat(content)
+        : await chat(messages.slice(-n));
 
       if (!res) throw new Error("Empty response");