fix(telegram): pass rich text prompts to cli backends

obviyus · obviyus · commit 45e36a241a79 · 2026-06-13T21:45:22.000+05:30
diff --git a/src/agents/cli-runner/helpers.system-prompt.test.ts b/src/agents/cli-runner/helpers.system-prompt.test.ts
@@ -106,4 +106,21 @@ describe("buildCliAgentSystemPrompt", () => {
     expect(prompt).toContain("session=agent:main:telegram:direct:peer");
     expect(prompt).toContain("sessionId=session-123");
   });
+
+  it("includes Telegram rich text guidance for CLI final replies", () => {
+    const prompt = buildCliAgentSystemPrompt({
+      workspaceDir: "/tmp/openclaw",
+      tools: [],
+      modelDisplay: "anthropic/claude-opus-4-8",
+      runtimeChannel: "telegram",
+      runtimeChatType: "direct",
+      runtimeCapabilities: ["richText"],
+    });
+
+    expect(prompt).toContain("Telegram rich text is available");
+    expect(prompt).toContain("headings, tables");
+    expect(prompt).toContain("This is not legacy MarkdownV2/parse_mode");
+    expect(prompt).toContain("channel=telegram");
+    expect(prompt).not.toContain("### message tool");
+  });
 });
diff --git a/src/agents/cli-runner/helpers.ts b/src/agents/cli-runner/helpers.ts
@@ -16,6 +16,7 @@ import { KeyedAsyncQueue } from "openclaw/plugin-sdk/keyed-async-queue";
 import { isAcpRuntimeSpawnAvailable } from "../../acp/runtime/availability.js";
 import type { SourceReplyDeliveryMode } from "../../auto-reply/get-reply-options.types.js";
 import type { ThinkLevel } from "../../auto-reply/thinking.js";
+import type { ChatType } from "../../channels/chat-type.js";
 import type { CliBackendConfig } from "../../config/types.js";
 import type { OpenClawConfig } from "../../config/types.openclaw.js";
 import { privateFileStore } from "../../infra/private-file-store.js";
@@ -119,6 +120,9 @@ export function buildCliAgentSystemPrompt(params: {
   extraSystemPrompt?: string;
   sourceReplyDeliveryMode?: SourceReplyDeliveryMode;
   silentReplyPromptMode?: SilentReplyPromptMode;
+  runtimeChannel?: string;
+  runtimeChatType?: ChatType;
+  runtimeCapabilities?: string[];
   ownerNumbers?: string[];
   heartbeatPrompt?: string;
   docsPath?: string;
@@ -152,6 +156,9 @@ export function buildCliAgentSystemPrompt(params: {
       model: params.modelDisplay,
       defaultModel: defaultModelLabel,
       shell: detectRuntimeShell(),
+      channel: params.runtimeChannel,
+      chatType: params.runtimeChatType,
+      capabilities: params.runtimeCapabilities,
     },
   });
   return buildConfiguredAgentSystemPrompt({
diff --git a/src/agents/cli-runner/prepare.test.ts b/src/agents/cli-runner/prepare.test.ts
@@ -5,6 +5,7 @@ import os from "node:os";
 import path from "node:path";
 import { CURRENT_SESSION_VERSION } from "openclaw/plugin-sdk/agent-sessions";
 import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
+import type { ChannelPlugin } from "../../channels/plugins/types.plugin.js";
 import type { OpenClawConfig } from "../../config/types.openclaw.js";
 import { registerLegacyContextEngine } from "../../context-engine/legacy.registration.js";
 import {
@@ -14,6 +15,11 @@ import {
 import type { ContextEngine } from "../../context-engine/types.js";
 import { getGlobalHookRunner } from "../../plugins/hook-runner-global.js";
 import { clearMemoryPluginState, registerMemoryPromptSection } from "../../plugins/memory-state.js";
+import { setActivePluginRegistry } from "../../plugins/runtime.js";
+import {
+  createChannelTestPluginBase,
+  createTestRegistry,
+} from "../../test-utils/channel-plugins.js";
 import { captureEnv, setTestEnvValue } from "../../test-utils/env.js";
 import { testing as cliBackendsTesting } from "../cli-backends.js";
 import { hashCliSessionText } from "../cli-session.js";
@@ -276,6 +282,7 @@ describe("shouldSkipLocalCliCredentialEpoch", () => {
     ensureSandboxWorkspaceForSessionMock.mockReset();
     resetContextWindowCacheForTest();
     clearMemoryPluginState();
+    setActivePluginRegistry(createTestRegistry());
     vi.unstubAllEnvs();
     sessionFileEnvSnapshot?.restore();
     sessionFileEnvSnapshot = undefined;
@@ -1012,6 +1019,45 @@ describe("shouldSkipLocalCliCredentialEpoch", () => {
     }
   });
 
+  it("passes Telegram rich text capabilities into CLI system prompts", async () => {
+    const { dir, sessionFile } = createSessionFile();
+    setActivePluginRegistry(
+      createTestRegistry([
+        {
+          pluginId: "telegram",
+          source: "test",
+          plugin: {
+            ...createChannelTestPluginBase({ id: "telegram", label: "Telegram" }),
+            agentPrompt: {
+              messageToolCapabilities: () => ["richText"],
+            },
+          } satisfies ChannelPlugin,
+        },
+      ]),
+    );
+
+    try {
+      const context = await prepareCliRunContext({
+        sessionId: "session-test",
+        sessionFile,
+        workspaceDir: dir,
+        prompt: "latest ask",
+        provider: "test-cli",
+        model: "test-model",
+        timeoutMs: 1_000,
+        runId: "run-test-telegram-rich-text",
+        messageChannel: "telegram",
+        config: createCliBackendConfig(),
+      });
+
+      expect(context.systemPrompt).toContain("channel=telegram");
+      expect(context.systemPrompt).toContain("Telegram rich text is available");
+      expect(context.systemPrompt).toContain("This is not legacy MarkdownV2/parse_mode");
+    } finally {
+      fs.rmSync(dir, { recursive: true, force: true });
+    }
+  });
+
   it("ignores volatile prompt text when static prompt text matches", async () => {
     const { dir, sessionFile } = createSessionFile();
     try {
diff --git a/src/agents/cli-runner/prepare.ts b/src/agents/cli-runner/prepare.ts
@@ -28,6 +28,7 @@ import { annotateInterSessionPromptText } from "../../sessions/input-provenance.
 import { resolveSkillsPromptForRun } from "../../skills/loading/workspace.js";
 import { resolveEmbeddedRunSkillEntries } from "../../skills/runtime/embedded-run-entries.js";
 import { resolveUserPath } from "../../utils.js";
+import { normalizeMessageChannel } from "../../utils/message-channel.js";
 import { resolveAgentDir, resolveSessionAgentIds } from "../agent-scope.js";
 import { externalCliDiscoveryForProviderAuth } from "../auth-profiles/external-cli-discovery.js";
 import { loadAuthProfileStoreForRuntime } from "../auth-profiles/store.js";
@@ -70,6 +71,7 @@ import {
 } from "../embedded-agent-runner/sandbox-skills.js";
 import { resolveHeartbeatPromptForSystemPrompt } from "../heartbeat-system-prompt.js";
 import { applyPluginTextReplacements } from "../plugin-text-transforms.js";
+import { collectRuntimeChannelCapabilities } from "../runtime-capabilities.js";
 import { ensureSandboxWorkspaceForSession } from "../sandbox.js";
 import { ensureSystemPromptCacheBoundary } from "../system-prompt-cache-boundary.js";
 import { buildSystemPromptReport } from "../system-prompt-report.js";
@@ -535,6 +537,12 @@ export async function prepareCliRunContext(
           agentId: sessionAgentId,
           sessionKey: params.sessionKey?.trim() || params.sessionId,
         });
+  const runtimeChannel = normalizeMessageChannel(params.messageChannel ?? params.messageProvider);
+  const runtimeCapabilities = collectRuntimeChannelCapabilities({
+    cfg: params.config,
+    channel: runtimeChannel,
+    accountId: params.agentAccountId,
+  });
   const builtSystemPrompt = buildCliAgentSystemPrompt({
     workspaceDir,
     cwd,
@@ -543,6 +551,9 @@ export async function prepareCliRunContext(
     extraSystemPrompt,
     sourceReplyDeliveryMode: params.sourceReplyDeliveryMode,
     silentReplyPromptMode: params.silentReplyPromptMode,
+    runtimeChannel,
+    runtimeChatType: params.sessionEntry?.chatType,
+    runtimeCapabilities,
     ownerNumbers: params.ownerNumbers,
     heartbeatPrompt,
     docsPath: openClawReferences.docsPath ?? undefined,
