Fix Telegram bot to call Zen API directly

bot.js

@@ -1,87 +1,159 @@
 const { Bot } = require("grammy");
 
-// Get environment variables
+// Environment variables
 const BOT_TOKEN = process.env.TELEGRAM_BOT_TOKEN;
 const CHAT_ID = process.env.TELEGRAM_CHAT_ID;
-const OPENCODE_URL = process.env.OPENCODE_URL || "http://localhost:7860";
+const ZEN_API_KEY = process.env.OPENCODE_ZEN_KEY;
 
 if (!BOT_TOKEN) {
   console.error("[ERROR] TELEGRAM_BOT_TOKEN not set");
   process.exit(1);
 }
+if (!ZEN_API_KEY) {
+  console.error("[ERROR] OPENCODE_ZEN_KEY not set");
+  process.exit(1);
+}
 
-// Create bot
+const ZEN_BASE = "https://opencode.ai/zen/v1";
 const bot = new Bot(BOT_TOKEN);
 
-// Store sessions
-const sessions = new Map();
+// User sessions: chatId -> { model, history }
+const userSessions = new Map();
 
-// Helper: Call OpenCode API
-async function callOpenCode(message, sessionId) {
-  try {
-    // Create a new session or use existing
-    const sessionResponse = await fetch(`${OPENCODE_URL}/session`, {
-      method: "POST",
-      headers: { "Content-Type": "application/json" },
-      body: JSON.stringify({}),
-    });
-
-    if (!sessionResponse.ok) {
-      throw new Error("Failed to create session");
-    }
+function getEndpoint(modelId) {
+  const openaiPrefixes = ["gpt-", "deepseek-", "grok-", "minimax-", "glm-", "kimi-", "big-pickle", "mimo-", "laguna-", "ling-", "north-", "nemotron-"];
+  if (openaiPrefixes.some((p) => modelId.startsWith(p) || modelId.includes(p))) {
+    return `${ZEN_BASE}/chat/completions`;
+  }
+  if (modelId.startsWith("claude-") || modelId.startsWith("qwen3")) {
+    return `${ZEN_BASE}/messages`;
+  }
+  return `${ZEN_BASE}/chat/completions`;
+}
 
-    const session = await sessionResponse.json();
-    const sid = session.id || sessionId;
+async function callZenAPI(messages, modelId) {
+  const endpoint = getEndpoint(modelId);
+  const headers = {
+    Authorization: `Bearer ${ZEN_API_KEY}`,
+    "Content-Type": "application/json",
+  };
+
+  let body;
+  if (endpoint.includes("/messages")) {
+    body = { model: modelId, messages, max_tokens: 4096 };
+  } else {
+    body = { model: modelId, messages, max_tokens: 4096 };
+  }
 
-    // Send message to OpenCode
-    const response = await fetch(`${OPENCODE_URL}/session/${sid}/message`, {
-      method: "POST",
-      headers: { "Content-Type": "application/json" },
-      body: JSON.stringify({ content: message }),
-    });
+  const response = await fetch(endpoint, {
+    method: "POST",
+    headers,
+    body: JSON.stringify(body),
+  });
 
-    if (!response.ok) {
-      throw new Error("Failed to send message");
-    }
+  if (!response.ok) {
+    const errText = await response.text();
+    throw new Error(`API ${response.status}: ${errText.substring(0, 300)}`);
+  }
 
-    const result = await response.json();
-    return result.content || result.message || "No response from OpenCode";
-  } catch (error) {
-    console.error("[ERROR] OpenCode API call failed:", error.message);
-    return `Error communicating with OpenCode: ${error.message}`;
+  const data = await response.json();
+
+  if (data.content) {
+    return data.content.map((c) => c.text || "").join("");
+  }
+  if (data.choices && data.choices[0]) {
+    return data.choices[0].message.content;
   }
+  return JSON.stringify(data).substring(0, 2000);
 }
 
-// Handle /start command
+// /start
 bot.command("start", async (ctx) => {
+  const chatId = ctx.chat.id;
+  if (CHAT_ID && chatId.toString() !== CHAT_ID.toString()) return;
+
+  userSessions.set(chatId, { model: "deepseek-v4-flash-free", history: [] });
   await ctx.reply(
-    "Welcome to OpenCode AI Agent!\n\n" +
-      "Send me any message and I'll respond using AI.\n\n" +
+    "🤖 Welcome to OpenCode AI Agent!\n\n" +
+      "I'm connected to OpenCode Zen with 40+ AI models.\n\n" +
       "Commands:\n" +
-      "/start - Show this message\n" +
-      "/help - Show help\n" +
-      "/new - Start a new session"
+      "/model <name> - Switch model\n" +
+      "/models - List available models\n" +
+      "/new - New conversation\n" +
+      "/help - Show help\n\n" +
+      "Just send me a message to start chatting!\n" +
+      "Default model: DeepSeek V4 Flash (FREE)"
   );
 });
 
-// Handle /help command
+// /help
 bot.command("help", async (ctx) => {
   await ctx.reply(
-    "OpenCode AI Agent Help\n\n" +
-      "Simply send me a message and I'll process it using OpenCode AI.\n\n" +
-      "Features:\n" +
-      "- AI-powered responses\n" +
-      "- Code assistance\n" +
-      "- Multiple AI models available\n\n" +
-      "Use /new to start a fresh conversation."
+    "📖 Help\n\n" +
+      "Send any message and I'll respond using AI.\n\n" +
+      "Commands:\n" +
+      "/start - Initialize bot\n" +
+      "/models - List all models\n" +
+      "/model <name> - Switch model\n" +
+      "/new - Clear history\n" +
+      "/help - This message\n\n" +
+      "Free models: deepseek-v4-flash-free, mimo-v2.5-free, big-pickle, laguna-s-2.1-free"
+  );
+});
+
+// /models
+bot.command("models", async (ctx) => {
+  await ctx.reply(
+    "📋 Available Models:\n\n" +
+      "🆓 FREE:\n" +
+      "• deepseek-v4-flash-free\n" +
+      "• mimo-v2.5-free\n" +
+      "• big-pickle\n" +
+      "• laguna-s-2.1-free\n" +
+      "• ling-3.0-flash-free\n" +
+      "• north-mini-code-free\n" +
+      "• nemotron-3-ultra-free\n\n" +
+      "💰 Paid:\n" +
+      "• gpt-5.5, gpt-5.4, gpt-5.4-mini, gpt-5-nano\n" +
+      "• claude-sonnet-5, claude-opus-5, claude-haiku-4-5\n" +
+      "• gemini-3.6-flash, gemini-3.1-pro\n" +
+      "• deepseek-v4-pro, qwen3.7-max, grok-4.5\n\n" +
+      "Use: /model <model-id>"
   );
 });
 
-// Handle /new command
+// /model
+bot.command("model", async (ctx) => {
+  const chatId = ctx.chat.id;
+  if (CHAT_ID && chatId.toString() !== CHAT_ID.toString()) return;
+
+  const args = ctx.message.text.split(" ").slice(1);
+  if (args.length === 0) {
+    const session = userSessions.get(chatId);
+    await ctx.reply(`Current model: ${session?.model || "deepseek-v4-flash-free"}`);
+    return;
+  }
+
+  const modelId = args[0].toLowerCase();
+  if (!userSessions.has(chatId)) {
+    userSessions.set(chatId, { model: modelId, history: [] });
+  } else {
+    userSessions.get(chatId).model = modelId;
+    userSessions.get(chatId).history = [];
+  }
+
+  await ctx.reply(`✅ Model switched to: ${modelId}\nHistory cleared. Send me a message!`);
+});
+
+// /new
 bot.command("new", async (ctx) => {
   const chatId = ctx.chat.id;
-  sessions.delete(chatId);
-  await ctx.reply("New session started. Send me a message!");
+  if (CHAT_ID && chatId.toString() !== CHAT_ID.toString()) return;
+
+  const session = userSessions.get(chatId);
+  const model = session?.model || "deepseek-v4-flash-free";
+  userSessions.set(chatId, { model, history: [] });
+  await ctx.reply("🔄 New conversation started!");
 });
 
 // Handle text messages
@@ -89,31 +161,38 @@ bot.on("message:text", async (ctx) => {
   const chatId = ctx.chat.id;
   const message = ctx.message.text;
 
-  // Check if it's the allowed chat
+  // Check allowed chat
   if (CHAT_ID && chatId.toString() !== CHAT_ID.toString()) {
     await ctx.reply("Sorry, I'm only available in the designated group.");
     return;
   }
 
-  // Show typing indicator
   await ctx.replyWithChatAction("typing");
 
   // Get or create session
-  let sessionId = sessions.get(chatId);
-  if (!sessionId) {
-    sessionId = `telegram_${chatId}`;
-    sessions.set(chatId, sessionId);
+  if (!userSessions.has(chatId)) {
+    userSessions.set(chatId, { model: "deepseek-v4-flash-free", history: [] });
   }
+  const session = userSessions.get(chatId);
 
-  // Call OpenCode
-  const response = await callOpenCode(message, sessionId);
+  // Build messages
+  const messages = [...session.history, { role: "user", content: message }];
 
-  // Send response
   try {
-    // Telegram has a 4096 character limit
+    const response = await callZenAPI(messages, session.model);
+
+    // Save to history
+    session.history.push({ role: "user", content: message });
+    session.history.push({ role: "assistant", content: response });
+
+    // Keep history under 20 messages
+    if (session.history.length > 20) {
+      session.history = session.history.slice(-20);
+    }
+
+    // Send (split if too long for Telegram's 4096 char limit)
     if (response.length > 4000) {
-      // Split into multiple messages
-      const chunks = response.match(/.{1,4000}/gs);
+      const chunks = response.match(/.{1,4000}/gs) || [response];
       for (const chunk of chunks) {
         await ctx.reply(chunk);
       }
@@ -121,20 +200,21 @@ bot.on("message:text", async (ctx) => {
       await ctx.reply(response);
     }
   } catch (error) {
-    console.error("[ERROR] Failed to send message:", error.message);
-    await ctx.reply("Failed to send response. Please try again.");
+    console.error("[ERROR]", error.message);
+    await ctx.reply(`❌ Error: ${error.message}`);
   }
 });
 
 // Error handler
 bot.catch((err) => {
-  console.error("[ERROR] Bot error:", err);
+  console.error("[BOT ERROR]", err);
 });
 
-// Start bot
+// Start
 console.log("[OK] Starting Telegram bot...");
 bot.start({
   onStart: () => {
     console.log("[OK] Telegram bot is running!");
+    console.log("[OK] Chat ID filter:", CHAT_ID || "none (accepting all)");
   },
 });