Перейти до основного вмісту

Архітектура інтеграції Pi

Цей документ описує, як OpenClaw інтегрується з pi-coding-agent та пов’язаними з ним пакетами (pi-ai, pi-agent-core, pi-tui) для реалізації можливостей свого AI-агента.

Огляд

OpenClaw використовує SDK pi, щоб вбудувати AI-агента для програмування у свою архітектуру шлюзу повідомлень. Замість запуску pi як підпроцесу або використання режиму RPC, OpenClaw безпосередньо імпортує та створює AgentSession через createAgentSession(). Цей вбудований підхід надає:
  • Повний контроль над життєвим циклом сесії та обробкою подій
  • Впровадження власних інструментів (повідомлення, sandbox, дії для конкретних каналів)
  • Налаштування системного промпту для кожного каналу/контексту
  • Збереження сесій із підтримкою розгалуження/ущільнення
  • Ротацію профілів автентифікації для кількох облікових записів із резервним перемиканням
  • Перемикання моделей незалежно від провайдера

Залежності пакетів

{
  "@mariozechner/pi-agent-core": "0.64.0",
  "@mariozechner/pi-ai": "0.64.0",
  "@mariozechner/pi-coding-agent": "0.64.0",
  "@mariozechner/pi-tui": "0.64.0"
}
ПакетПризначення
pi-aiБазові абстракції LLM: Model, streamSimple, типи повідомлень, API провайдерів
pi-agent-coreЦикл агента, виконання інструментів, типи AgentMessage
pi-coding-agentВисокорівневий SDK: createAgentSession, SessionManager, AuthStorage, ModelRegistry, вбудовані інструменти
pi-tuiКомпоненти термінального UI (використовуються в локальному режимі TUI OpenClaw)

Структура файлів

src/agents/
├── pi-embedded-runner.ts          # Повторний експорт із pi-embedded-runner/
├── pi-embedded-runner/
│   ├── run.ts                     # Основна точка входу: runEmbeddedPiAgent()
│   ├── run/
│   │   ├── attempt.ts             # Логіка однієї спроби з налаштуванням сесії
│   │   ├── params.ts              # Тип RunEmbeddedPiAgentParams
│   │   ├── payloads.ts            # Побудова payload-відповідей із результатів запуску
│   │   ├── images.ts              # Впровадження зображень для vision-моделі
│   │   └── types.ts               # EmbeddedRunAttemptResult
│   ├── abort.ts                   # Виявлення помилок переривання
│   ├── cache-ttl.ts               # Відстеження TTL кешу для обрізання контексту
│   ├── compact.ts                 # Логіка ручного/автоматичного ущільнення
│   ├── extensions.ts              # Завантаження розширень pi для вбудованих запусків
│   ├── extra-params.ts            # Додаткові параметри потоку для конкретних провайдерів
│   ├── google.ts                  # Виправлення порядку ходів для Google/Gemini
│   ├── history.ts                 # Обмеження історії (DM проти групи)
│   ├── lanes.ts                   # Командні доріжки сесії/глобальні
│   ├── logger.ts                  # Логер підсистеми
│   ├── model.ts                   # Визначення моделі через ModelRegistry
│   ├── runs.ts                    # Відстеження активних запусків, переривання, черга
│   ├── sandbox-info.ts            # Інформація про sandbox для системного промпту
│   ├── session-manager-cache.ts   # Кешування екземплярів SessionManager
│   ├── session-manager-init.ts    # Ініціалізація файлу сесії
│   ├── system-prompt.ts           # Побудова системного промпту
│   ├── tool-split.ts              # Поділ інструментів на builtIn та custom
│   ├── types.ts                   # EmbeddedPiAgentMeta, EmbeddedPiRunResult
│   └── utils.ts                   # Відображення ThinkLevel, опис помилок
├── pi-embedded-subscribe.ts       # Підписка/диспетчеризація подій сесії
├── pi-embedded-subscribe.types.ts # SubscribeEmbeddedPiSessionParams
├── pi-embedded-subscribe.handlers.ts # Фабрика обробників подій
├── pi-embedded-subscribe.handlers.lifecycle.ts
├── pi-embedded-subscribe.handlers.types.ts
├── pi-embedded-block-chunker.ts   # Поділ потокових відповідей на блоки
├── pi-embedded-messaging.ts       # Відстеження надісланих інструментом повідомлень
├── pi-embedded-helpers.ts         # Класифікація помилок, валідація ходів
├── pi-embedded-helpers/           # Допоміжні модулі
├── pi-embedded-utils.ts           # Утиліти форматування
├── pi-tools.ts                    # createOpenClawCodingTools()
├── pi-tools.abort.ts              # Обгортання AbortSignal для інструментів
├── pi-tools.policy.ts             # Політика allowlist/denylist для інструментів
├── pi-tools.read.ts               # Кастомізації інструмента read
├── pi-tools.schema.ts             # Нормалізація схем інструментів
├── pi-tools.types.ts              # Псевдонім типу AnyAgentTool
├── pi-tool-definition-adapter.ts  # Адаптер AgentTool -> ToolDefinition
├── pi-settings.ts                 # Перевизначення налаштувань
├── pi-hooks/                      # Власні hooks для pi
│   ├── compaction-safeguard.ts    # Захисне розширення
│   ├── compaction-safeguard-runtime.ts
│   ├── context-pruning.ts         # Розширення обрізання контексту на основі TTL кешу
│   └── context-pruning/
├── model-auth.ts                  # Визначення профілю автентифікації
├── auth-profiles.ts               # Сховище профілів, cooldown, failover
├── model-selection.ts             # Визначення моделі за замовчуванням
├── models-config.ts               # Генерація models.json
├── model-catalog.ts               # Кеш каталогу моделей
├── context-window-guard.ts        # Валідація вікна контексту
├── failover-error.ts              # Клас FailoverError
├── defaults.ts                    # DEFAULT_PROVIDER, DEFAULT_MODEL
├── system-prompt.ts               # buildAgentSystemPrompt()
├── system-prompt-params.ts        # Визначення параметрів системного промпту
├── system-prompt-report.ts        # Генерація звіту для налагодження
├── tool-summaries.ts              # Підсумки описів інструментів
├── tool-policy.ts                 # Визначення політики інструментів
├── transcript-policy.ts           # Політика валідації транскрипту
├── skills.ts                      # Побудова знімків/промптів Skills
├── skills/                        # Підсистема Skills
├── sandbox.ts                     # Визначення контексту sandbox
├── sandbox/                       # Підсистема sandbox
├── channel-tools.ts               # Впровадження інструментів для конкретних каналів
├── openclaw-tools.ts              # Інструменти, специфічні для OpenClaw
├── bash-tools.ts                  # Інструменти exec/process
├── apply-patch.ts                 # Інструмент apply_patch (OpenAI)
├── tools/                         # Реалізації окремих інструментів
│   ├── browser-tool.ts
│   ├── canvas-tool.ts
│   ├── cron-tool.ts
│   ├── gateway-tool.ts
│   ├── image-tool.ts
│   ├── message-tool.ts
│   ├── nodes-tool.ts
│   ├── session*.ts
│   ├── web-*.ts
│   └── ...
└── ...
Середовища виконання дій із повідомленнями для конкретних каналів тепер розміщені в каталогах розширень, якими володіють plugins, а не в src/agents/tools, наприклад:
  • файли середовища виконання дій plugin Discord
  • файл середовища виконання дій plugin Slack
  • файл середовища виконання дій plugin Telegram
  • файл середовища виконання дій plugin WhatsApp

Основний потік інтеграції

1. Запуск вбудованого агента

Основна точка входу — runEmbeddedPiAgent() у pi-embedded-runner/run.ts:
import { runEmbeddedPiAgent } from "./agents/pi-embedded-runner.js";

const result = await runEmbeddedPiAgent({
  sessionId: "user-123",
  sessionKey: "main:whatsapp:+1234567890",
  sessionFile: "/path/to/session.jsonl",
  workspaceDir: "/path/to/workspace",
  config: openclawConfig,
  prompt: "Hello, how are you?",
  provider: "anthropic",
  model: "claude-sonnet-4-6",
  timeoutMs: 120_000,
  runId: "run-abc",
  onBlockReply: async (payload) => {
    await sendToChannel(payload.text, payload.mediaUrls);
  },
});

2. Створення сесії

Усередині runEmbeddedAttempt() (яка викликається з runEmbeddedPiAgent()) використовується SDK pi:
import {
  createAgentSession,
  DefaultResourceLoader,
  SessionManager,
  SettingsManager,
} from "@mariozechner/pi-coding-agent";

const resourceLoader = new DefaultResourceLoader({
  cwd: resolvedWorkspace,
  agentDir,
  settingsManager,
  additionalExtensionPaths,
});
await resourceLoader.reload();

const { session } = await createAgentSession({
  cwd: resolvedWorkspace,
  agentDir,
  authStorage: params.authStorage,
  modelRegistry: params.modelRegistry,
  model: params.model,
  thinkingLevel: mapThinkingLevel(params.thinkLevel),
  tools: builtInTools,
  customTools: allCustomTools,
  sessionManager,
  settingsManager,
  resourceLoader,
});

applySystemPromptOverrideToSession(session, systemPromptOverride);

3. Підписка на події

subscribeEmbeddedPiSession() підписується на події AgentSession у pi:
const subscription = subscribeEmbeddedPiSession({
  session: activeSession,
  runId: params.runId,
  verboseLevel: params.verboseLevel,
  reasoningMode: params.reasoningLevel,
  toolResultFormat: params.toolResultFormat,
  onToolResult: params.onToolResult,
  onReasoningStream: params.onReasoningStream,
  onBlockReply: params.onBlockReply,
  onPartialReply: params.onPartialReply,
  onAgentEvent: params.onAgentEvent,
});
Серед оброблюваних подій:
  • message_start / message_end / message_update (потоковий текст/мислення)
  • tool_execution_start / tool_execution_update / tool_execution_end
  • turn_start / turn_end
  • agent_start / agent_end
  • auto_compaction_start / auto_compaction_end

4. Надсилання промпту

Після налаштування сесії надсилається промпт:
await session.prompt(effectivePrompt, { images: imageResult.images });
SDK обробляє весь цикл роботи агента: надсилання до LLM, виконання викликів інструментів, потокове передавання відповідей. Впровадження зображень є локальним для промпту: OpenClaw завантажує посилання на зображення з поточного промпту та передає їх через images лише для цього ходу. Він не сканує повторно попередні ходи історії, щоб повторно впроваджувати payload-и зображень.

Архітектура інструментів

Конвеєр інструментів

  1. Базові інструменти: codingTools із pi (read, bash, edit, write)
  2. Власні заміни: OpenClaw замінює bash на exec/process, налаштовує read/edit/write для sandbox
  3. Інструменти OpenClaw: повідомлення, browser, canvas, sessions, cron, gateway тощо
  4. Інструменти каналів: інструменти дій для Discord/Telegram/Slack/WhatsApp
  5. Фільтрація політиками: інструменти фільтруються за профілем, провайдером, агентом, групою, політиками sandbox
  6. Нормалізація схем: схеми очищуються з урахуванням особливостей Gemini/OpenAI
  7. Обгортання AbortSignal: інструменти обгортаються для підтримки сигналів переривання

Адаптер визначень інструментів

AgentTool із pi-agent-core має іншу сигнатуру execute, ніж ToolDefinition у pi-coding-agent. Адаптер у pi-tool-definition-adapter.ts з’єднує їх:
export function toToolDefinitions(tools: AnyAgentTool[]): ToolDefinition[] {
  return tools.map((tool) => ({
    name: tool.name,
    label: tool.label ?? name,
    description: tool.description ?? "",
    parameters: tool.parameters,
    execute: async (toolCallId, params, onUpdate, _ctx, signal) => {
      // pi-coding-agent signature differs from pi-agent-core
      return await tool.execute(toolCallId, params, signal, onUpdate);
    },
  }));
}

Стратегія поділу інструментів

splitSdkTools() передає всі інструменти через customTools:
export function splitSdkTools(options: { tools: AnyAgentTool[]; sandboxEnabled: boolean }) {
  return {
    builtInTools: [], // Empty. We override everything
    customTools: toToolDefinitions(options.tools),
  };
}
Це забезпечує узгодженість фільтрації політиками, інтеграції sandbox і розширеного набору інструментів OpenClaw для всіх провайдерів.

Побудова системного промпту

Системний промпт будується в buildAgentSystemPrompt() (system-prompt.ts). Він формує повний промпт із розділами, зокрема Tooling, Tool Call Style, Safety guardrails, довідка CLI OpenClaw, Skills, Docs, Workspace, Sandbox, Messaging, Reply Tags, Voice, Silent Replies, Heartbeats, Runtime metadata, а також Memory і Reactions, якщо вони ввімкнені, плюс необов’язкові файли контексту й додатковий вміст системного промпту. Для мінімального режиму промпту, який використовується субагентами, розділи обрізаються. Промпт застосовується після створення сесії через applySystemPromptOverrideToSession():
const systemPromptOverride = createSystemPromptOverride(appendPrompt);
applySystemPromptOverrideToSession(session, systemPromptOverride);

Керування сесіями

Файли сесій

Сесії — це JSONL-файли з деревоподібною структурою (зв’язки id/parentId). За збереження відповідає SessionManager із Pi:
const sessionManager = SessionManager.open(params.sessionFile);
OpenClaw обгортає його за допомогою guardSessionManager() для безпечної обробки результатів інструментів.

Кешування сесій

session-manager-cache.ts кешує екземпляри SessionManager, щоб уникнути повторного розбору файлу:
await prewarmSessionFile(params.sessionFile);
sessionManager = SessionManager.open(params.sessionFile);
trackSessionManagerAccess(params.sessionFile);

Обмеження історії

limitHistoryTurns() обрізає історію розмови залежно від типу каналу (DM чи група).

Ущільнення

Автоматичне ущільнення запускається при переповненні контексту. Типові сигнатури переповнення включають request_too_large, context length exceeded, input exceeds the maximum number of tokens, input token count exceeds the maximum number of input tokens, input is too long for the model і ollama error: context length exceeded. compactEmbeddedPiSessionDirect() обробляє ручне ущільнення:
const compactResult = await compactEmbeddedPiSessionDirect({
  sessionId, sessionFile, provider, model, ...
});

Автентифікація та визначення моделі

Профілі автентифікації

OpenClaw підтримує сховище профілів автентифікації з кількома API-ключами на провайдера:
const authStore = ensureAuthProfileStore(agentDir, { allowKeychainPrompt: false });
const profileOrder = resolveAuthProfileOrder({ cfg, store: authStore, provider, preferredProfile });
Профілі обертаються при збоях із відстеженням cooldown:
await markAuthProfileFailure({ store, profileId, reason, cfg, agentDir });
const rotated = await advanceAuthProfile();

Визначення моделі

import { resolveModel } from "./pi-embedded-runner/model.js";

const { model, error, authStorage, modelRegistry } = resolveModel(
  provider,
  modelId,
  agentDir,
  config,
);

// Uses pi's ModelRegistry and AuthStorage
authStorage.setRuntimeApiKey(model.provider, apiKeyInfo.apiKey);

Failover

FailoverError запускає резервне перемикання моделі, якщо це налаштовано:
if (fallbackConfigured && isFailoverErrorMessage(errorText)) {
  throw new FailoverError(errorText, {
    reason: promptFailoverReason ?? "unknown",
    provider,
    model: modelId,
    profileId,
    status: resolveFailoverStatus(promptFailoverReason),
  });
}

Розширення Pi

OpenClaw завантажує власні розширення pi для спеціалізованої поведінки:

Захист ущільнення

src/agents/pi-hooks/compaction-safeguard.ts додає захисні механізми для ущільнення, зокрема адаптивне бюджетування токенів, а також зведення про збої інструментів і файлові операції:
if (resolveCompactionMode(params.cfg) === "safeguard") {
  setCompactionSafeguardRuntime(params.sessionManager, { maxHistoryShare });
  paths.push(resolvePiExtensionPath("compaction-safeguard"));
}

Обрізання контексту

src/agents/pi-hooks/context-pruning.ts реалізує обрізання контексту на основі TTL кешу:
if (cfg?.agents?.defaults?.contextPruning?.mode === "cache-ttl") {
  setContextPruningRuntime(params.sessionManager, {
    settings,
    contextWindowTokens,
    isToolPrunable,
    lastCacheTouchAt,
  });
  paths.push(resolvePiExtensionPath("context-pruning"));
}

Потокова передача та блокові відповіді

Поділ на блоки

EmbeddedBlockChunker керує потоковим перетворенням тексту на окремі блоки відповіді:
const blockChunker = blockChunking ? new EmbeddedBlockChunker(blockChunking) : null;

Видалення тегів thinking/final

Потоковий вивід обробляється для видалення блоків <think>/<thinking> і виділення вмісту <final>:
const stripBlockTags = (text: string, state: { thinking: boolean; final: boolean }) => {
  // Strip <think>...</think> content
  // If enforceFinalTag, only return <final>...</final> content
};

Директиви відповіді

Директиви відповіді, такі як [[media:url]], [[voice]], [[reply:id]], аналізуються та виділяються:
const { text: cleanedText, mediaUrls, audioAsVoice, replyToId } = consumeReplyDirectives(chunk);

Обробка помилок

Класифікація помилок

pi-embedded-helpers.ts класифікує помилки для відповідної обробки:
isContextOverflowError(errorText)     // Context too large
isCompactionFailureError(errorText)   // Compaction failed
isAuthAssistantError(lastAssistant)   // Auth failure
isRateLimitAssistantError(...)        // Rate limited
isFailoverAssistantError(...)         // Should failover
classifyFailoverReason(errorText)     // "auth" | "rate_limit" | "quota" | "timeout" | ...

Резервний рівень thinking

Якщо рівень thinking не підтримується, використовується резервний варіант:
const fallbackThinking = pickFallbackThinkingLevel({
  message: errorText,
  attempted: attemptedThinking,
});
if (fallbackThinking) {
  thinkLevel = fallbackThinking;
  continue;
}

Інтеграція sandbox

Коли режим sandbox увімкнено, інструменти й шляхи обмежуються:
const sandbox = await resolveSandboxContext({
  config: params.config,
  sessionKey: sandboxSessionKey,
  workspaceDir: resolvedWorkspace,
});

if (sandboxRoot) {
  // Use sandboxed read/edit/write tools
  // Exec runs in container
  // Browser uses bridge URL
}

Обробка для конкретних провайдерів

Anthropic

  • Очищення магічного рядка відмови
  • Валідація ходів для послідовних ролей
  • Сувора валідація параметрів інструментів Pi на upstream-рівні

Google/Gemini

  • Санітизація схем інструментів, якою володіє plugin

OpenAI

  • Інструмент apply_patch для моделей Codex
  • Обробка зниження рівня thinking

Інтеграція TUI

OpenClaw також має локальний режим TUI, який безпосередньо використовує компоненти pi-tui:
// src/tui/tui.ts
import { ... } from "@mariozechner/pi-tui";
Це забезпечує інтерактивний досвід роботи в терміналі, подібний до нативного режиму pi.

Ключові відмінності від Pi CLI

АспектPi CLIВбудований OpenClaw
ВикликКоманда pi / RPCSDK через createAgentSession()
ІнструментиСтандартні інструменти для програмуванняВласний набір інструментів OpenClaw
Системний промптAGENTS.md + промптиДинамічний для кожного каналу/контексту
Зберігання сесій~/.pi/agent/sessions/~/.openclaw/agents/<agentId>/sessions/ (або $OPENCLAW_STATE_DIR/agents/<agentId>/sessions/)
AuthОкремі облікові даніКілька профілів із ротацією
РозширенняЗавантажуються з дискаПрограмно + шляхи на диску
Обробка подійРендеринг TUIНа основі callback-ів (onBlockReply тощо)

Майбутні міркування

Напрями для потенційного доопрацювання:
  1. Узгодження сигнатур інструментів: наразі виконується адаптація між сигнатурами pi-agent-core і pi-coding-agent
  2. Обгортання менеджера сесій: guardSessionManager додає безпеку, але підвищує складність
  3. Завантаження розширень: можна безпосередніше використовувати ResourceLoader із pi
  4. Складність обробника потоків: subscribeEmbeddedPiSession значно розрісся
  5. Особливості провайдерів: багато кодових шляхів для конкретних провайдерів, які pi потенційно міг би обробляти сам

Тести

Покриття інтеграції Pi охоплює такі набори тестів:
  • src/agents/pi-*.test.ts
  • src/agents/pi-auth-json.test.ts
  • src/agents/pi-embedded-*.test.ts
  • src/agents/pi-embedded-helpers*.test.ts
  • src/agents/pi-embedded-runner*.test.ts
  • src/agents/pi-embedded-runner/**/*.test.ts
  • src/agents/pi-embedded-subscribe*.test.ts
  • src/agents/pi-tools*.test.ts
  • src/agents/pi-tool-definition-adapter*.test.ts
  • src/agents/pi-settings.test.ts
  • src/agents/pi-hooks/**/*.test.ts
Живі/опціональні:
  • src/agents/pi-embedded-runner-extraparams.live.test.ts (увімкніть OPENCLAW_LIVE_TEST=1)
Актуальні команди запуску див. у Pi Development Workflow.