viber-channel 0.6.0 → 0.7.0

package/viber-gemma-bridge.ts

@@ -0,0 +1,169 @@
+#!/usr/bin/env bun
+/**
+ * viber-gemma-bridge — Gemma 4 runtime bridge (#318 step-02).
+ *
+ * Gemma is a "bare LLM" runtime: unlike Codex it hosts no MCP tools and runs no
+ * app-server. The bridge joins a Viber conversation as an instance, forwards each
+ * inbound message to ollama (`/api/chat`, model `gemma4:e4b`), and posts the
+ * assistant's reply back. ALL the lifecycle — await-invite join, prime-on-join,
+ * lock, token refresh, SSE loop, post — is the SHARED `bridge_core`
+ * (runAwaitInviteBridge); this file is only the Gemma adapter: a per-conversation
+ * chat history + the ollama call. No tool host, no anti-double-post, no
+ * elicitation (a non-tool runtime never posts out-of-band, so outboundHandled is
+ * always false).
+ *
+ * The ollama endpoint is configurable (VIBER_OLLAMA_URL): server-hosted Gemma by
+ * default, or a client's own local ollama.
+ */
+import { join } from "node:path";
+import { loadAuth } from "./lib/auth.ts";
+import { defaultLabel } from "./lib/conversation.ts";
+import { clientFingerprint } from "./lib/fingerprint.ts";
+import {
+  agentIdentityFromEnv,
+  BridgeShutdownError,
+  type ConversationRuntime,
+  type MakeRunTurn,
+  type RunTurn,
+  runAwaitInviteBridge,
+} from "./lib/bridge_core.ts";
+import {
+  DEFAULT_GEMMA_MODEL,
+  type ChatMessage,
+  ollamaChat,
+  OllamaResponseError,
+  OllamaUnavailableError,
+} from "./lib/ollama_chat.ts";
+
+const LOG_PREFIX = "[viber-gemma-bridge]";
+const BASE_URL = process.env.VIBER_BASE_URL ?? "https://viber.dgypx.dev";
+const LOCK_DIR =
+  process.env.APPDATA
+    ? join(process.env.APPDATA, "viber")
+    : join(process.env.HOME ?? "/tmp", ".config", "viber");
+
+/** Concise voice-friendly system prompt (ported from viber/standalone_ai.py). */
+const SYSTEM_PROMPT =
+  "You are a friendly, concise voice assistant. Reply naturally, as in a spoken " +
+  "conversation — short sentences, no markdown. Answer in the user's language. " +
+  "Keep replies to 2-3 sentences unless more detail is explicitly requested.";
+// Append the agent identity (#309 step-17) after the base prompt so a spawned
+// gemma agent knows its own name/role. Empty when neither was provided.
+const EFFECTIVE_SYSTEM_PROMPT = [SYSTEM_PROMPT, agentIdentityFromEnv()]
+  .filter((line) => line.length > 0)
+  .join("\n");
+/** Keep the last N turns (user+assistant) of per-conversation history. */
+const MAX_HISTORY_MESSAGES = 40;
+
+interface GemmaBridgeOptions {
+  once: boolean;
+  includeAgentMessages: boolean;
+  awaitInvite: boolean;
+  reuseAuthInstance: boolean;
+}
+
+export function parseArgs(argv: string[]): GemmaBridgeOptions {
+  const opts: GemmaBridgeOptions = {
+    once: false,
+    includeAgentMessages: false,
+    awaitInvite: false,
+    reuseAuthInstance: false,
+  };
+  for (const arg of argv) {
+    if (arg === "--once") opts.once = true;
+    else if (arg === "--include-agent-messages") opts.includeAgentMessages = true;
+    else if (arg === "--await-invite") opts.awaitInvite = true;
+    else if (arg === "--reuse-auth-instance") opts.reuseAuthInstance = true;
+    else if (arg === "-h" || arg === "--help") {
+      process.stdout.write(
+        "viber-gemma-bridge — join a Viber conversation as a Gemma (ollama) agent\n\n" +
+          "USAGE\n  viber-gemma-bridge --await-invite\n\n" +
+          "OPTIONS\n  --await-invite              Wait for server-pushed conversation joins.\n" +
+          "  --once                      Exit after the first handled message.\n" +
+          "  --include-agent-messages    React to other agents' messages too.\n" +
+          "  --reuse-auth-instance       Reuse auth.json's persisted instance.\n\n" +
+          "ENV\n  VIBER_OLLAMA_URL   ollama base URL (default http://127.0.0.1:11434).\n" +
+          "  VIBER_GEMMA_MODEL  ollama model (default gemma4:e4b).\n",
+      );
+      process.exit(0);
+    } else throw new Error(`Unknown argument: ${arg}`);
+  }
+  if (!opts.awaitInvite) {
+    throw new Error("viber-gemma-bridge requires --await-invite (server-pushed joins).");
+  }
+  return opts;
+}
+
+/** Build the Gemma per-conversation adapter: a local chat history + ollama call. */
+export function makeGemmaRunTurn(): MakeRunTurn {
+  const ollamaUrl = process.env.VIBER_OLLAMA_URL;
+  const model = process.env.VIBER_GEMMA_MODEL ?? DEFAULT_GEMMA_MODEL;
+  // Per-conversation history, keyed by conversation id (each DM is independent).
+  const histories = new Map<string, ChatMessage[]>();
+
+  return async (convId: string, _runtime: ConversationRuntime, _signal: AbortSignal): Promise<RunTurn> => {
+    if (!histories.has(convId)) histories.set(convId, []);
+    return async (content, _msg, signal) => {
+      const history = histories.get(convId)!;
+      const messages: ChatMessage[] = [
+        { role: "system", content: EFFECTIVE_SYSTEM_PROMPT },
+        ...history,
+        { role: "user", content },
+      ];
+      let reply: string;
+      try {
+        reply = await ollamaChat(messages, { baseUrl: ollamaUrl, model, signal });
+      } catch (err) {
+        if (err instanceof Error && err.name === "AbortError") throw err; // stream torn down
+        // Surface a clean spoken error instead of crashing the turn.
+        if (err instanceof OllamaUnavailableError) {
+          process.stderr.write(`${LOG_PREFIX} ollama unavailable: ${err.message}\n`);
+          return { reply: "Sorry, my local model isn't reachable right now.", outboundHandled: false };
+        }
+        if (err instanceof OllamaResponseError) {
+          process.stderr.write(`${LOG_PREFIX} ollama error: ${err.message}\n`);
+          return { reply: "Sorry, I hit an error generating a reply.", outboundHandled: false };
+        }
+        throw err;
+      }
+      // Record the turn + trim history (only on success).
+      history.push({ role: "user", content }, { role: "assistant", content: reply });
+      if (history.length > MAX_HISTORY_MESSAGES) {
+        histories.set(convId, history.slice(-MAX_HISTORY_MESSAGES));
+      }
+      return { reply, outboundHandled: false };
+    };
+  };
+}
+
+export async function main(argv: string[] = process.argv.slice(2)): Promise<void> {
+  const options = parseArgs(argv);
+  const auth = loadAuth();
+  const fingerprint = clientFingerprint(process.cwd());
+  if (fingerprint !== auth.client_fingerprint) {
+    throw new Error("Folder fingerprint mismatch — auth file was issued for a different machine/folder.");
+  }
+  const label = process.env.VIBER_CODEX_BRIDGE_LABEL ?? `Gemma bridge • ${defaultLabel(process.cwd())}`;
+
+  await runAwaitInviteBridge({
+    baseUrl: BASE_URL,
+    lockDir: LOCK_DIR,
+    logPrefix: LOG_PREFIX,
+    sessionTag: "gemma",
+    label,
+    kind: "gemma",
+    options,
+    auth,
+    fingerprint,
+    makeRunTurn: makeGemmaRunTurn(),
+  });
+}
+
+if (import.meta.main) {
+  try {
+    await main();
+  } catch (err) {
+    process.stderr.write(`${LOG_PREFIX} ${err instanceof Error ? err.message : String(err)}\n`);
+    process.exit(err instanceof BridgeShutdownError ? err.exitCode : 1);
+  }
+}