@axiom-lattice/core 1.0.20 → 1.0.21

package/src/agent_lattice/builders/AgentParamsBuilder.ts

@@ -1,86 +0,0 @@
-/**
- * Agent parameters builder
- *
- * responsible for preparing the parameters required for building the Agent
- */
-
-import { toolLatticeManager } from "../../tool_lattice/ToolLatticeManager";
-import { modelLatticeManager } from "../../model_lattice/ModelLatticeManager";
-import { AgentLattice, AgentBuildParams, GraphBuildOptions } from "../types";
-
-/**
- * Get Agent Lattice function type
- */
-export type GetAgentLatticeFunction = (key: string) => AgentLattice | undefined;
-
-/**
- * Agent parameters builder
- */
-export class AgentParamsBuilder {
-  private getAgentLatticeFunc: GetAgentLatticeFunction;
-
-  /**
-   * constructor
-   *
-   * @param getAgentLatticeFunc get Agent Lattice function
-   */
-  constructor(getAgentLatticeFunc: GetAgentLatticeFunction) {
-    this.getAgentLatticeFunc = getAgentLatticeFunc;
-  }
-
-  /**
-   * build Agent parameters
-   *
-   * @param agentLattice Agent Lattice object
-   * @param options build options
-   * @returns Agent build parameters
-   */
-  public buildParams(
-    agentLattice: AgentLattice,
-    options?: GraphBuildOptions
-  ): AgentBuildParams {
-    // get tools
-    const toolKeys = options?.overrideTools || agentLattice.config.tools || [];
-    const tools = toolKeys.map((toolKey: string) => {
-      const toolLattice = toolLatticeManager.getToolLattice(toolKey);
-      if (!toolLattice) {
-        throw new Error(`Tool "${toolKey}" does not exist`);
-      }
-      return {
-        key: toolKey,
-        definition: toolLattice.config,
-        executor: toolLattice.client,
-      };
-    });
-
-    // get model
-    const modelKey = options?.overrideModel || agentLattice.config.modelKey;
-    const model = modelKey
-      ? modelLatticeManager.getModelLattice(modelKey).client
-      : modelLatticeManager.getModelLattice("default").client;
-    if (!model) {
-      throw new Error(`Model "${modelKey}" does not exist`);
-    }
-
-    // get subAgents
-    const subAgentKeys = agentLattice.config.subAgents || [];
-    const subAgents = subAgentKeys.map((agentKey: string) => {
-      const subAgentLattice = this.getAgentLatticeFunc(agentKey);
-      if (!subAgentLattice) {
-        throw new Error(`SubAgent "${agentKey}" does not exist`);
-      }
-      return {
-        key: agentKey,
-        config: subAgentLattice.config,
-        client: subAgentLattice.client,
-      };
-    });
-
-    return {
-      tools,
-      model,
-      subAgents,
-      prompt: agentLattice.config.prompt,
-    };
-  }
-}

package/src/agent_lattice/builders/DeepAgentGraphBuilder.ts

@@ -1,46 +0,0 @@
-/**
- * Deep Agent Graph Builder
- *
- * 实现 Deep Agent 类型的 Agent Graph 构建
- */
-
-import { CompiledStateGraph } from "@langchain/langgraph";
-import { DynamicTool } from "@langchain/core/tools";
-import { createDeepAgent } from "../../deep_agent";
-import { AgentLattice, AgentBuildParams } from "../types";
-import { AgentGraphBuilder } from "./AgentGraphBuilder";
-import { getToolClient } from "../../tool_lattice/ToolLatticeManager";
-
-export class DeepAgentGraphBuilder implements AgentGraphBuilder {
-  /**
-   * 构建Deep Agent Graph
-   *
-   * @param agentLattice Agent Lattice对象
-   * @param params Agent构建参数
-   * @returns 返回CompiledGraph对象
-   */
-  build(
-    agentLattice: AgentLattice,
-    params: AgentBuildParams
-  ): CompiledStateGraph<any, any, any, any, any> {
-    const tools = params.tools
-      .map((t) => {
-        const toolClient = getToolClient(t.key);
-
-        return toolClient;
-      })
-      .filter((tool) => tool !== undefined);
-
-    return createDeepAgent({
-      tools: tools,
-      model: params.model,
-      instructions: params.prompt,
-      subagents: params.subAgents.map((sa) => ({
-        name: sa.key,
-        description: sa.config.description,
-        prompt: sa.config.prompt,
-        tools: sa.config.tools || [],
-      })),
-    });
-  }
-}

package/src/agent_lattice/builders/PlanExecuteAgentGraphBuilder.ts

@@ -1,46 +0,0 @@
-/**
- * Plan Execute Agent Graph Builder
- *
- * 实现 Plan Execute 类型的 Agent Graph 构建
- */
-
-import { CompiledStateGraph } from "@langchain/langgraph";
-import { DynamicTool } from "@langchain/core/tools";
-import { AgentLattice, AgentBuildParams } from "../types";
-import { AgentGraphBuilder } from "./AgentGraphBuilder";
-import { getToolClient } from "../../tool_lattice/ToolLatticeManager";
-import { createPlanExecuteAgent } from "@createPlanExecuteAgent";
-
-export class PlanExecuteAgentGraphBuilder implements AgentGraphBuilder {
-  /**
-   * 构建Plan Execute Agent Graph
-   *
-   * @param agentLattice Agent Lattice对象
-   * @param params Agent构建参数
-   * @returns 返回CompiledGraph对象
-   */
-  build(
-    agentLattice: AgentLattice,
-    params: AgentBuildParams
-  ): CompiledStateGraph<any, any, any, any, any> {
-    // 创建符合 DynamicTool 接口的工具对象
-    const tools = params.tools
-      .map((t) => {
-        // 使用 DynamicTool 构造工具对象
-        const tool = getToolClient(t.key);
-        return tool;
-      })
-      .filter((tool) => tool !== undefined);
-
-    // 由于createPlanExecuteAgent的返回类型与CompiledStateGraph不兼容
-    // 这里我们使用createAgentExecutor作为替代方案
-    const agent = createPlanExecuteAgent({
-      llmManager: params.model,
-      tools: tools,
-      prompt: params.prompt,
-      name: agentLattice.config.name,
-      description: agentLattice.config.description,
-    });
-    return agent.getCompiledGraph();
-  }
-}

package/src/agent_lattice/builders/ReActAgentGraphBuilder.ts

@@ -1,42 +0,0 @@
-/**
- * ReAct Agent Graph Builder
- *
- * 实现 ReAct 类型的 Agent Graph 构建
- */
-
-import { CompiledStateGraph } from "@langchain/langgraph";
-import { DynamicTool } from "@langchain/core/tools";
-import { AgentLattice, AgentBuildParams } from "../types";
-import { AgentGraphBuilder } from "./AgentGraphBuilder";
-import { getToolClient } from "../../tool_lattice/ToolLatticeManager";
-import { createReactAgent } from "@langchain/langgraph/prebuilt";
-
-export class ReActAgentGraphBuilder implements AgentGraphBuilder {
-  /**
-   * 构建ReAct Agent Graph
-   *
-   * @param agentLattice Agent Lattice对象
-   * @param params Agent构建参数
-   * @returns 返回CompiledGraph对象
-   */
-  build(
-    agentLattice: AgentLattice,
-    params: AgentBuildParams
-  ): CompiledStateGraph<any, any, any, any, any> {
-    // 创建符合 DynamicTool 接口的工具对象
-    const tools = params.tools
-      .map((t) => {
-        // 使用 DynamicTool 构造工具对象
-        const tool = getToolClient(t.key);
-        return tool;
-      })
-      .filter((tool) => tool !== undefined);
-
-    return createReactAgent({
-      llm: params.model,
-      tools: tools,
-      prompt: params.prompt,
-      name: agentLattice.config.name,
-    });
-  }
-}

package/src/agent_lattice/builders/index.ts

@@ -1,14 +0,0 @@
-/**
- * Agent Builder 模块
- *
- * 提供构建Agent Graph的各种Builder和工具
- */
-
-// 导出接口
-export { AgentGraphBuilder } from "./AgentGraphBuilder";
-export { GetAgentLatticeFunction } from "./AgentParamsBuilder";
-
-// 导出工厂和工具类
-export { AgentGraphBuilderFactory } from "./AgentGraphBuilderFactory";
-export { AgentParamsBuilder } from "./AgentParamsBuilder";
-export { AgentBuilder } from "./AgentBuilder";

package/src/agent_lattice/index.ts

@@ -1,27 +0,0 @@
-/**
- * Agent Lattice模块
- *
- * 提供Agent Lattice的管理和Graph构建功能
- */
-
-// 导出类型定义
-export {
-  AgentType,
-  AgentConfig,
-  AgentClient,
-  AgentLattice,
-  GraphBuildOptions,
-} from "./types";
-
-// 导出AgentLatticeManager相关内容
-export {
-  AgentLatticeManager,
-  agentLatticeManager,
-  registerAgentLattice,
-  registerAgentLattices,
-  getAgentLattice,
-  getAgentConfig,
-  getAllAgentConfigs,
-  validateAgentInput,
-  getAgentClient,
-} from "./AgentLatticeManager";

package/src/agent_lattice/types.ts

@@ -1,42 +0,0 @@
-/**
- * Agent Lattice 类型定义
- *
- * 包含Agent Lattice相关的所有类型定义，供其他模块共享使用
- */
-
-import { z } from "zod";
-import { ModelLattice } from "@model_lattice/ModelLattice";
-import { CompiledStateGraph } from "@langchain/langgraph";
-import {
-  AgentType,
-  AgentConfig,
-  GraphBuildOptions,
-} from "@axiom-lattice/protocols";
-
-// Re-export types from protocols
-export { AgentType, AgentConfig, GraphBuildOptions };
-
-// Agent客户端类型 - CompiledStateGraph
-export type AgentClient = CompiledStateGraph<any, any, any, any, any>;
-
-// AgentLattice接口定义
-export interface AgentLattice {
-  config: AgentConfig;
-  client: AgentClient | undefined;
-}
-
-// Agent构建参数接口
-export interface AgentBuildParams {
-  tools: Array<{
-    key: string;
-    definition: any;
-    executor: any;
-  }>;
-  model: ModelLattice;
-  subAgents: Array<{
-    key: string;
-    config: AgentConfig;
-    client: AgentClient | undefined;
-  }>;
-  prompt: string;
-}

package/src/base/BaseLatticeManager.ts

@@ -1,145 +0,0 @@
-/**
- * BaseLatticeManager - 抽象基类，为各种Lattice管理器提供通用功能
- * 使用统一的Lattice注册表，不同类型的Lattice通过前缀区分
- * @template TItem - 管理的项目类型
- */
-export abstract class BaseLatticeManager<TItem = any> {
-  // 全局统一的Lattice注册表
-  protected static registry: Map<string, any> = new Map();
-
-  /**
-   * 受保护的构造函数，防止外部直接实例化
-   */
-  protected constructor() {
-    // 空实现
-  }
-
-  /**
-   * 获取管理器实例（由子类实现）
-   */
-  public static getInstance(): BaseLatticeManager<any> {
-    throw new Error("必须由子类实现");
-  }
-
-  /**
-   * 获取Lattice类型，用于构造键前缀
-   * 子类必须重写此方法以提供唯一的类型标识符
-   */
-  protected abstract getLatticeType(): string;
-
-  /**
-   * 构造完整的键名，包含类型前缀
-   * @param key 原始键名
-   */
-  protected getFullKey(key: string): string {
-    return `${this.getLatticeType()}.${key}`;
-  }
-
-  /**
-   * 注册项目
-   * @param key 项目键名（不含前缀）
-   * @param item 项目实例
-   */
-  public register(key: string, item: TItem): void {
-    const fullKey = this.getFullKey(key);
-    if (BaseLatticeManager.registry.has(fullKey)) {
-      throw new Error(`项目 "${fullKey}" 已经存在，无法重复注册`);
-    }
-
-    BaseLatticeManager.registry.set(fullKey, item);
-  }
-
-  /**
-   * 获取指定项目
-   * @param key 项目键名（不含前缀）
-   */
-  public get(key: string): TItem | undefined {
-    const fullKey = this.getFullKey(key);
-    return BaseLatticeManager.registry.get(fullKey) as TItem | undefined;
-  }
-
-  /**
-   * 获取所有当前类型的项目
-   */
-  public getAll(): TItem[] {
-    const prefix = `${this.getLatticeType()}.`;
-    const result: TItem[] = [];
-
-    for (const [key, value] of BaseLatticeManager.registry.entries()) {
-      if (key.startsWith(prefix)) {
-        result.push(value as TItem);
-      }
-    }
-
-    return result;
-  }
-
-  /**
-   * 检查项目是否存在
-   * @param key 项目键名（不含前缀）
-   */
-  public has(key: string): boolean {
-    const fullKey = this.getFullKey(key);
-    return BaseLatticeManager.registry.has(fullKey);
-  }
-
-  /**
-   * 移除项目
-   * @param key 项目键名（不含前缀）
-   */
-  public remove(key: string): boolean {
-    const fullKey = this.getFullKey(key);
-    return BaseLatticeManager.registry.delete(fullKey);
-  }
-
-  /**
-   * 清空当前类型的所有项目
-   */
-  public clear(): void {
-    const prefix = `${this.getLatticeType()}.`;
-    const keysToDelete: string[] = [];
-
-    for (const key of BaseLatticeManager.registry.keys()) {
-      if (key.startsWith(prefix)) {
-        keysToDelete.push(key);
-      }
-    }
-
-    for (const key of keysToDelete) {
-      BaseLatticeManager.registry.delete(key);
-    }
-  }
-
-  /**
-   * 获取当前类型的项目数量
-   */
-  public count(): number {
-    const prefix = `${this.getLatticeType()}.`;
-    let count = 0;
-
-    for (const key of BaseLatticeManager.registry.keys()) {
-      if (key.startsWith(prefix)) {
-        count++;
-      }
-    }
-
-    return count;
-  }
-
-  /**
-   * 获取当前类型的项目键名列表（不含前缀）
-   */
-  public keys(): string[] {
-    const prefix = `${this.getLatticeType()}.`;
-    const prefixLength = prefix.length;
-    const result: string[] = [];
-
-    for (const key of BaseLatticeManager.registry.keys()) {
-      if (key.startsWith(prefix)) {
-        result.push(key.substring(prefixLength));
-      }
-    }
-
-    return result;
-  }
-}

package/src/base/index.ts

@@ -1 +0,0 @@
-export * from "./BaseLatticeManager";