@keo-ai/axiom 0.1.2 → 0.1.4

package/dist/function_call_loop/types.d.ts

@@ -0,0 +1,208 @@
+/**
+ * LLM 对话消息。支持 system、user、assistant、tool 四种角色。
+ * assistant 角色可携带 tool_calls；tool 角色需携带 tool_call_id。
+ */
+export interface Message {
+    readonly role: 'system' | 'user' | 'assistant' | 'tool';
+    readonly content: string;
+    readonly tool_calls?: ReadonlyArray<ToolCall>;
+    readonly tool_call_id?: string;
+}
+/**
+ * LLM 返回的工具调用。
+ */
+export interface ToolCall {
+    readonly id: string;
+    readonly type: 'function';
+    readonly function: {
+        readonly name: string;
+        readonly arguments: string;
+    };
+}
+/**
+ * 暴露给 LLM 的工具定义。
+ */
+export interface ToolDefinition {
+    readonly type: 'function';
+    readonly function: {
+        readonly name: string;
+        readonly description: string;
+        readonly parameters: unknown;
+    };
+}
+/**
+ * Harness 记录的执行状态。
+ */
+export type HarnessRecordStatus = 'pending' | 'success' | 'error' | 'timeout' | 'cancelled';
+/**
+ * 单条执行历史记录。
+ */
+export interface HarnessRecord {
+    readonly id: string;
+    readonly turn: number;
+    readonly globalIndex: number;
+    readonly toolName: string;
+    readonly args: unknown;
+    readonly rawArgs: string;
+    readonly status: HarnessRecordStatus;
+    readonly result?: unknown;
+    readonly error?: string;
+    readonly startedAt: number;
+    readonly durationMs: number;
+}
+/**
+ * 执行前事件。
+ */
+export interface ToolExecutionStartEvent {
+    readonly type: 'execution:start';
+    readonly callId: string;
+    readonly toolName: string;
+    readonly args: unknown;
+    readonly turn: number;
+}
+/**
+ * 执行后事件（同时携带结果）。
+ */
+export interface ToolExecutionEndEvent {
+    readonly type: 'execution:end';
+    readonly callId: string;
+    readonly toolName: string;
+    readonly status: HarnessRecordStatus;
+    readonly durationMs: number;
+    readonly result?: unknown;
+    readonly error?: string;
+    readonly turn: number;
+}
+/**
+ * Loop 生命周期事件。
+ */
+export type LoopEvent = ToolExecutionStartEvent | ToolExecutionEndEvent;
+/**
+ * Tool 发现函数的返回结果。
+ */
+export interface ToolDiscoverResult {
+    readonly name: string;
+    readonly description: string;
+    readonly parameters: unknown;
+    readonly visible: boolean;
+}
+/**
+ * Tool 执行上下文。
+ */
+export interface ToolExecuteContext {
+    readonly harness: ReadonlyArray<HarnessRecord>;
+    readonly turn: number;
+    readonly metadata: unknown;
+}
+/**
+ * Tool 审批配置。
+ * 当 tool 被调用时，先执行 createTicket 生成审批单，暂停 Loop 等待外部审批。
+ * createTicket 可通过 context.metadata 访问 LoopConfig 传入的 metadata，用于携带业务上下文（如用户ID、会话ID等）。
+ * 审批通过后恢复执行时，应通过某种方式（如 createTicket 内部查库）让 Loop 直接执行，无需去掉 approval 配置。
+ */
+export interface ApprovalConfig {
+    readonly createTicket: (args: unknown, context: ToolExecuteContext) => Promise<{
+        ticketId: string;
+        approved?: boolean;
+    }>;
+}
+/**
+ * 注册到 Loop 中的 Tool。
+ */
+export interface Tool {
+    readonly name: string;
+    readonly description: string;
+    readonly parameters: unknown;
+    readonly discover?: (harness: ReadonlyArray<HarnessRecord>, metadata: unknown) => ToolDiscoverResult | Promise<ToolDiscoverResult>;
+    readonly execute: (args: unknown, context: ToolExecuteContext) => unknown | Promise<unknown>;
+    readonly approval?: ApprovalConfig;
+}
+/**
+ * Turn Policy 决策结果。
+ */
+export interface TurnPolicyResult {
+    /** 终止 Loop，此消息作为 assistant 回复写入历史 */
+    readonly message?: string;
+    /** 继续执行前作为 system 消息注入 */
+    readonly injectMessage?: string;
+}
+/**
+ * 全局轮次策略。
+ */
+export type TurnPolicy = (harness: ReadonlyArray<HarnessRecord>, turn: number, metadata: unknown) => TurnPolicyResult | Promise<TurnPolicyResult>;
+/**
+ * 上下文压缩配置。
+ */
+export interface CompressionConfig {
+    readonly keepRounds: number;
+    readonly compress: (messages: ReadonlyArray<ReadonlyArray<Message>>) => Message[] | Promise<Message[]>;
+}
+/**
+ * LLM 调用可选参数。
+ */
+export interface LLMCallOptions {
+    readonly model?: string;
+    readonly temperature?: number;
+    readonly maxTokens?: number;
+    readonly topP?: number;
+    readonly reasoningEffort?: 'low' | 'medium' | 'high';
+}
+/**
+ * LLM 调用接口。由外部注入，Loop 内部不绑定具体 Provider。
+ */
+export interface LLMCaller {
+    readonly call: (messages: ReadonlyArray<Message>, tools: ReadonlyArray<ToolDefinition>, options?: LLMCallOptions) => Promise<{
+        readonly content: string | null;
+        readonly tool_calls?: ReadonlyArray<ToolCall>;
+    }>;
+}
+/**
+ * Loop 配置。
+ */
+export interface LoopConfig {
+    /** 初始消息列表，直接作为对话起点 */
+    readonly messages: ReadonlyArray<Message>;
+    readonly tools: ReadonlyArray<Tool>;
+    readonly llmCaller: LLMCaller;
+    readonly maxTurns?: number;
+    readonly turnPolicy?: TurnPolicy;
+    readonly compression?: CompressionConfig;
+    readonly metadata?: unknown;
+    /** 告警文本：当 turnPolicy 未自定义且当前轮次等于 maxTurns - 1 时塞入 Messages */
+    readonly warningMessage?: string;
+    /** 终止文本：当 turnPolicy 未自定义且当前轮次等于 maxTurns 时塞入 Messages */
+    readonly terminateMessage?: string;
+    /** 事件监听器 */
+    readonly onEvent?: (event: LoopEvent) => void | Promise<void>;
+    /** 取消信号 */
+    readonly signal?: AbortSignal;
+    /** 模型名称。未设置时从环境变量或 Provider 默认值读取 */
+    readonly model?: string;
+    /** 采样温度，范围 0~2 */
+    readonly temperature?: number;
+    /** 单次 LLM 调用的最大输出 token 数 */
+    readonly maxTokens?: number;
+    /** 核采样概率阈值，范围 0~1 */
+    readonly topP?: number;
+    /** 推理深度。仅部分模型支持 */
+    readonly reasoningEffort?: 'low' | 'medium' | 'high';
+}
+/**
+ * 待审批信息。
+ */
+export interface PendingApprovalInfo {
+    readonly ticketId: string;
+    readonly toolName: string;
+    readonly args: unknown;
+    readonly callId: string;
+}
+/**
+ * Loop 执行结果。
+ */
+export interface LoopResult {
+    readonly messages: Message[];
+    readonly harness: ReadonlyArray<HarnessRecord>;
+    readonly finalContent: string | null;
+    readonly turns: number;
+    readonly pendingApproval?: PendingApprovalInfo;
+}

package/dist/function_call_loop/types.js

@@ -0,0 +1,2 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });

package/dist/index.d.ts

@@ -32,3 +32,5 @@ export type { Model, PredictConfig, PredictWithMessagesConfig } from './predict'
 /** 向量检索（Embedding + pgvector + Rerank） */
 export { EmbeddingSearch, embed } from './embedding_search';
 export type { EmbeddingSearchConfig, SearchResult } from './embedding_search';
+/** Function Call Loop */
+export * as FunctionCallLoop from './function_call_loop';

package/dist/index.js

@@ -19,8 +19,41 @@
  *
  * @packageDocumentation
  */
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.embed = exports.EmbeddingSearch = exports.MODEL_REGISTRY = exports.BailianProvider = exports.Predictor = exports.LLM = void 0;
+exports.FunctionCallLoop = exports.embed = exports.EmbeddingSearch = exports.MODEL_REGISTRY = exports.BailianProvider = exports.Predictor = exports.LLM = void 0;
 /** LLM 静态入口类，封装 Provider 连接、故障转移和返回解析 */
 var predict_1 = require("./predict");
 Object.defineProperty(exports, "LLM", { enumerable: true, get: function () { return predict_1.LLM; } });
@@ -37,3 +70,5 @@ Object.defineProperty(exports, "MODEL_REGISTRY", { enumerable: true, get: functi
 var embedding_search_1 = require("./embedding_search");
 Object.defineProperty(exports, "EmbeddingSearch", { enumerable: true, get: function () { return embedding_search_1.EmbeddingSearch; } });
 Object.defineProperty(exports, "embed", { enumerable: true, get: function () { return embedding_search_1.embed; } });
+/** Function Call Loop */
+exports.FunctionCallLoop = __importStar(require("./function_call_loop"));

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@keo-ai/axiom",
-  "version": "0.1.2",
+  "version": "0.1.4",
   "description": "基于 LLM 的预测与推理库，支持多 Provider 切换",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",