@axiom-lattice/core 1.0.1

package/dist/index.d.ts

@@ -0,0 +1,356 @@
+import { BaseMessage } from '@langchain/core/messages';
+import { ZodType } from 'zod/v3';
+import { $ZodType } from 'zod/v4/core';
+import { BaseChatModel, BaseChatModelCallOptions } from '@langchain/core/language_models/chat_models';
+import { BaseLanguageModelInput } from '@langchain/core/language_models/base';
+import { CallbackManagerForLLMRun } from '@langchain/core/callbacks/manager';
+import { ChatResult } from '@langchain/core/outputs';
+import { LLMConfig, AgentConfig, GraphBuildOptions } from '@axiom-lattice/protocols';
+import * as protocols from '@axiom-lattice/protocols';
+export { protocols as Protocols };
+export { AgentConfig, AgentType, GraphBuildOptions, MemoryType } from '@axiom-lattice/protocols';
+import { CompiledStateGraph } from '@langchain/langgraph';
+import { BaseCheckpointSaver } from '@langchain/langgraph-checkpoint';
+
+/**
+ * BaseLatticeManager - 抽象基类，为各种Lattice管理器提供通用功能
+ * 使用统一的Lattice注册表，不同类型的Lattice通过前缀区分
+ * @template TItem - 管理的项目类型
+ */
+declare abstract class BaseLatticeManager<TItem = any> {
+    protected static registry: Map<string, any>;
+    /**
+     * 受保护的构造函数，防止外部直接实例化
+     */
+    protected constructor();
+    /**
+     * 获取管理器实例（由子类实现）
+     */
+    static getInstance(): BaseLatticeManager<any>;
+    /**
+     * 获取Lattice类型，用于构造键前缀
+     * 子类必须重写此方法以提供唯一的类型标识符
+     */
+    protected abstract getLatticeType(): string;
+    /**
+     * 构造完整的键名，包含类型前缀
+     * @param key 原始键名
+     */
+    protected getFullKey(key: string): string;
+    /**
+     * 注册项目
+     * @param key 项目键名（不含前缀）
+     * @param item 项目实例
+     */
+    register(key: string, item: TItem): void;
+    /**
+     * 获取指定项目
+     * @param key 项目键名（不含前缀）
+     */
+    get(key: string): TItem | undefined;
+    /**
+     * 获取所有当前类型的项目
+     */
+    getAll(): TItem[];
+    /**
+     * 检查项目是否存在
+     * @param key 项目键名（不含前缀）
+     */
+    has(key: string): boolean;
+    /**
+     * 移除项目
+     * @param key 项目键名（不含前缀）
+     */
+    remove(key: string): boolean;
+    /**
+     * 清空当前类型的所有项目
+     */
+    clear(): void;
+    /**
+     * 获取当前类型的项目数量
+     */
+    count(): number;
+    /**
+     * 获取当前类型的项目键名列表（不含前缀）
+     */
+    keys(): string[];
+}
+
+/**
+ * ModelLattice类，继承自BaseChatModel
+ * 简化版本，只保留通过LLMConfig创建LLM实例的功能
+ */
+declare class ModelLattice extends BaseChatModel {
+    private config;
+    private llm;
+    lc_namespace: string[];
+    /**
+     * 构造函数
+     * @param config LLM配置
+     */
+    constructor(config: LLMConfig);
+    /**
+     * 返回模型类型
+     */
+    _llmType(): string;
+    /**
+     * 返回模型类型
+     */
+    _modelType(): string;
+    /**
+     * 实现BaseChatModel的_generate方法
+     * @param messages 消息数组
+     * @param options 调用选项
+     * @param runManager 回调管理器
+     * @returns 聊天结果
+     */
+    _generate(messages: BaseMessage[], options: this["ParsedCallOptions"], runManager?: CallbackManagerForLLMRun): Promise<ChatResult>;
+    /**
+     * 将工具绑定到模型
+     * @param tools 工具列表
+     * @param tool_choice 工具选择选项
+     * @returns 绑定工具后的模型
+     */
+    bindTools(tools: any[], { tool_choice, ...kwargs }?: {
+        tool_choice?: "auto" | "none" | "required" | any;
+        [key: string]: any;
+    }): any;
+    /**
+     * 使用结构化输出调用LLM
+     * @param input 输入
+     * @param schema 结构化输出的schema
+     * @param options 调用选项
+     * @returns 结构化输出
+     */
+    invokeWithStructuredOutput<RunOutput extends Record<string, any> = Record<string, any>>(input: BaseLanguageModelInput, schema: ZodType<RunOutput> | $ZodType<RunOutput> | Record<string, any>, options?: BaseChatModelCallOptions & {
+        includeRaw?: boolean;
+        name?: string;
+        method?: "functionCalling" | "jsonMode";
+    }): Promise<RunOutput | {
+        raw: BaseMessage;
+        parsed: RunOutput;
+    }>;
+    /**
+     * 创建LLM实例
+     * @param config LLM配置
+     * @returns LLM实例
+     */
+    private initChatModel;
+}
+
+type ModelConfig = LLMConfig;
+type ModelClient = ModelLattice;
+interface ModelLatticeInterface {
+    key: string;
+    client: ModelClient;
+}
+/**
+ * ModelLatticeManager - 单例模型Lattice管理器
+ * 负责注册、管理各种语言模型Lattice
+ */
+declare class ModelLatticeManager extends BaseLatticeManager<ModelLatticeInterface> {
+    private static _instance;
+    /**
+     * 获取ModelLatticeManager单例实例
+     */
+    static getInstance(): ModelLatticeManager;
+    /**
+     * 获取Lattice类型前缀
+     */
+    protected getLatticeType(): string;
+    /**
+     * 注册模型Lattice
+     * @param key Lattice键名
+     * @param config 模型配置
+     */
+    registerLattice(key: string, config: ModelConfig): void;
+    /**
+     * 获取ModelLattice
+     * @param key Lattice键名
+     */
+    getModelLattice(key: string): ModelLatticeInterface;
+    /**
+     * 获取所有Lattice
+     */
+    getAllLattices(): ModelLatticeInterface[];
+    /**
+     * 检查Lattice是否存在
+     * @param key Lattice键名
+     */
+    hasLattice(key: string): boolean;
+    /**
+     * 移除Lattice
+     * @param key Lattice键名
+     */
+    removeLattice(key: string): boolean;
+    /**
+     * 清空所有Lattice
+     */
+    clearLattices(): void;
+    /**
+     * 获取Lattice数量
+     */
+    getLatticeCount(): number;
+    /**
+     * 获取Lattice键名列表
+     */
+    getLatticeKeys(): string[];
+}
+declare const modelLatticeManager: ModelLatticeManager;
+declare const registerModelLattice: (key: string, config: ModelConfig) => void;
+declare const getModelLattice: (key: string) => ModelLatticeInterface;
+
+/**
+ * Agent Lattice 类型定义
+ *
+ * 包含Agent Lattice相关的所有类型定义，供其他模块共享使用
+ */
+
+type AgentClient = CompiledStateGraph<any, any, any, any, any>;
+interface AgentLattice {
+    config: AgentConfig;
+    client: AgentClient | undefined;
+}
+
+/**
+ * AgentLatticeManager - 单例Agent Lattice管理器
+ * 负责注册、管理各种Agent Lattice
+ */
+declare class AgentLatticeManager extends BaseLatticeManager<AgentLattice> {
+    private static _instance;
+    /**
+     * 获取AgentLatticeManager单例实例
+     */
+    static getInstance(): AgentLatticeManager;
+    /**
+     * 获取Lattice类型前缀
+     */
+    protected getLatticeType(): string;
+    /**
+     * 注册Agent Lattice
+     * @param config Agent配置
+     */
+    registerLattice(config: AgentConfig): void;
+    /**
+     * 获取AgentLattice
+     * @param key Lattice键名
+     */
+    getAgentLattice(key: string): AgentLattice | undefined;
+    /**
+     * 获取所有Lattice
+     */
+    getAllLattices(): AgentLattice[];
+    /**
+     * 检查Lattice是否存在
+     * @param key Lattice键名
+     */
+    hasLattice(key: string): boolean;
+    /**
+     * 移除Lattice
+     * @param key Lattice键名
+     */
+    removeLattice(key: string): boolean;
+    /**
+     * 获取Agent配置
+     * @param key Lattice键名
+     */
+    getAgentConfig(key: string): AgentConfig | undefined;
+    /**
+     * 获取所有Agent配置
+     */
+    getAllAgentConfigs(): AgentConfig[];
+    /**
+     * 验证Agent输入参数
+     * @param key Lattice键名
+     * @param input 输入参数
+     */
+    validateAgentInput(key: string, input: any): boolean;
+    /**
+     * 构建Agent参数
+     *
+     * @param agentLattice Agent Lattice对象
+     * @param options 构建选项
+     * @returns 返回Agent构建参数
+     */
+    private buildAgentParams;
+    /**
+     * 初始化Agent客户端
+     *
+     * 使用AgentGraphBuilderFactory构建Graph并设置为客户端
+     *
+     * @param key Lattice键名
+     * @param options 构建选项
+     * @returns 返回CompiledGraph对象
+     */
+    initializeClient(key: string, options?: GraphBuildOptions): AgentClient;
+}
+declare const agentLatticeManager: AgentLatticeManager;
+declare const registerAgentLattice: (config: AgentConfig) => void;
+declare const registerAgentLattices: (configs: AgentConfig[]) => void;
+declare const getAgentLattice: (key: string) => AgentLattice | undefined;
+declare const getAgentConfig: (key: string) => AgentConfig | undefined;
+declare const getAllAgentConfigs: () => AgentConfig[];
+declare const validateAgentInput: (key: string, input: any) => boolean;
+/**
+ * 获取或初始化Agent客户端
+ *
+ * @param key Agent Lattice键名
+ * @param options 构建选项
+ * @returns 返回CompiledGraph对象
+ */
+declare const getAgentClient: (key: string, options?: GraphBuildOptions) => AgentClient;
+
+/**
+ * MemoryLatticeManager
+ *
+ * 记忆Lattice管理器，负责管理和注册检查点保存器
+ */
+
+/**
+ * 记忆Lattice管理器类
+ */
+declare class MemoryLatticeManager extends BaseLatticeManager {
+    private static instance;
+    private static checkpointSavers;
+    /**
+     * 私有构造函数，防止外部直接实例化
+     */
+    private constructor();
+    /**
+     * 获取单例实例
+     */
+    static getInstance(): MemoryLatticeManager;
+    /**
+     * 获取Lattice类型
+     */
+    protected getLatticeType(): string;
+    /**
+     * 注册检查点保存器
+     * @param key 保存器键名
+     * @param saver 检查点保存器实例
+     */
+    registerCheckpointSaver(key: string, saver: BaseCheckpointSaver): void;
+    /**
+     * 获取检查点保存器
+     * @param key 保存器键名
+     */
+    getCheckpointSaver(key: string): BaseCheckpointSaver;
+    /**
+     * 获取所有已注册的检查点保存器键名
+     */
+    getCheckpointSaverKeys(): string[];
+    /**
+     * 检查检查点保存器是否存在
+     * @param key 保存器键名
+     */
+    hasCheckpointSaver(key: string): boolean;
+    /**
+     * 移除检查点保存器
+     * @param key 保存器键名
+     */
+    removeCheckpointSaver(key: string): boolean;
+}
+declare const getCheckpointSaver: (key: string) => BaseCheckpointSaver<number>;
+declare const registerCheckpointSaver: (key: string, saver: BaseCheckpointSaver) => void;
+
+export { type AgentClient, type AgentLattice, AgentLatticeManager, MemoryLatticeManager, type ModelConfig, type ModelLatticeInterface, ModelLatticeManager, agentLatticeManager, getAgentClient, getAgentConfig, getAgentLattice, getAllAgentConfigs, getCheckpointSaver, getModelLattice, modelLatticeManager, registerAgentLattice, registerAgentLattices, registerCheckpointSaver, registerModelLattice, validateAgentInput };