91 lines
3.4 KiB
TypeScript
91 lines
3.4 KiB
TypeScript
|
/**
|
|||
|
|
* 平台能力桥（任务 A3.11）：把平台的能力目录注册成 Pi 的工具。
|
|||
|
|
*
|
|||
|
|
* 这个文件由平台以 `--extension <path>` 加载，运行在 Pi 子进程里（Pi 用 jiti 直接读 TS）。
|
|||
|
|
* 它自己不判断权限、不碰凭据，只做一件事：把 Pi 的工具调用转成对平台内部回调端点的 HTTP 请求，
|
|||
|
|
* 平台侧收口到 Core.invoke（架构设计 5.1「每次都过 Core」）。
|
|||
|
|
*
|
|||
|
|
* 配置来自环境变量 APS_BRIDGE_CONFIG 指向的 JSON 文件，由平台在建立会话时写入：
|
|||
|
|
* { baseUrl, internalToken, sessionId, capabilities: [{ toolName, capabilityId, description, parameters }] }
|
|||
|
|
*/
|
|||
|
|
import { readFileSync } from 'node:fs';
|
|||
|
|
|
|||
|
|
interface BridgeCapability {
|
|||
|
|
toolName: string;
|
|||
|
|
capabilityId: string;
|
|||
|
|
description: string;
|
|||
|
|
parameters: Record<string, unknown>;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
interface BridgeConfig {
|
|||
|
|
baseUrl: string;
|
|||
|
|
internalToken: string;
|
|||
|
|
sessionId: string;
|
|||
|
|
capabilities: BridgeCapability[];
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
interface PlatformReceipt {
|
|||
|
|
executionId: string;
|
|||
|
|
status: 'accepted' | 'pending-confirmation' | 'succeeded' | 'failed';
|
|||
|
|
result?: unknown;
|
|||
|
|
errorKind?: string;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/** Pi 扩展的注册接口（只声明用到的部分，避免扩展依赖 Pi 的类型包） */
|
|||
|
|
interface PiExtensionApi {
|
|||
|
|
registerTool(tool: {
|
|||
|
|
name: string;
|
|||
|
|
label: string;
|
|||
|
|
description: string;
|
|||
|
|
parameters: Record<string, unknown>;
|
|||
|
|
execute: (toolCallId: string, params: unknown, signal?: AbortSignal) => Promise<{ content: Array<{ type: 'text'; text: string }>; details: unknown }>;
|
|||
|
|
}): void;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
export default function apsBridge(pi: PiExtensionApi): void {
|
|||
|
|
const configPath = process.env.APS_BRIDGE_CONFIG ?? '';
|
|||
|
|
if (configPath === '') return;
|
|||
|
|
|
|||
|
|
let config: BridgeConfig;
|
|||
|
|
try {
|
|||
|
|
config = JSON.parse(readFileSync(configPath, 'utf8')) as BridgeConfig;
|
|||
|
|
} catch {
|
|||
|
|
// 配置读不到就不注册任何工具：宁可让模型说做不到，也不要注册一个坏工具
|
|||
|
|
return;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
for (const capability of config.capabilities) {
|
|||
|
|
pi.registerTool({
|
|||
|
|
name: capability.toolName,
|
|||
|
|
label: capability.capabilityId,
|
|||
|
|
description: capability.description,
|
|||
|
|
parameters: capability.parameters,
|
|||
|
|
async execute(toolCallId: string, params: unknown, signal?: AbortSignal) {
|
|||
|
|
const response = await fetch(`${config.baseUrl}/internal/tool-call`, {
|
|||
|
|
method: 'POST',
|
|||
|
|
headers: { 'content-type': 'application/json', 'x-aps-internal-token': config.internalToken },
|
|||
|
|
body: JSON.stringify({
|
|||
|
|
sessionId: config.sessionId,
|
|||
|
|
capabilityId: capability.capabilityId,
|
|||
|
|
params,
|
|||
|
|
idempotencyKey: `call-${toolCallId}`,
|
|||
|
|
}),
|
|||
|
|
...(signal === undefined ? {} : { signal }),
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
if (!response.ok) {
|
|||
|
|
const detail = await response.text().catch(() => '');
|
|||
|
|
throw new Error(`平台拒绝调用（HTTP ${response.status}）：${detail.slice(0, 200)}`);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
const receipt = (await response.json()) as PlatformReceipt;
|
|||
|
|
if (receipt.status === 'failed') {
|
|||
|
|
const detail = (receipt.result as { message?: string } | undefined)?.message ?? '未提供原因';
|
|||
|
|
throw new Error(`能力 ${capability.capabilityId} 执行失败（${receipt.errorKind ?? 'unknown'}）：${detail}`);
|
|||
|
|
}
|
|||
|
|
// 成功：把回执原样交给模型（含 executionId，便于它引用证据）
|
|||
|
|
return { content: [{ type: 'text', text: JSON.stringify(receipt) }], details: receipt };
|
|||
|
|
},
|
|||
|
|
});
|
|||
|
|
}
|
|||
|
|
}
|