@axiom-lattice/core 1.0.20 → 1.0.21

package/package.json

@@ -1,9 +1,12 @@
 {
   "name": "@axiom-lattice/core",
-  "version": "1.0.20",
+  "version": "1.0.21",
   "main": "dist/index.js",
   "module": "dist/index.mjs",
   "types": "dist/index.d.ts",
+  "files": [
+    "dist"
+  ],
   "exports": {
     ".": {
       "types": "./dist/index.d.ts",
@@ -33,7 +36,7 @@
     "uuid": "^9.0.1",
     "zod": "^3.24.2",
     "zod-to-json-schema": "^3.24.3",
-    "@axiom-lattice/protocols": "1.0.20"
+    "@axiom-lattice/protocols": "1.0.21"
   },
   "devDependencies": {
     "@types/jest": "^29.5.14",

package/.eslintrc.json

@@ -1,22 +0,0 @@
-{
-  "root": true,
-  "env": {
-    "node": true,
-    "es2021": true
-  },
-  "extends": ["eslint:recommended", "plugin:@typescript-eslint/recommended"],
-  "parser": "@typescript-eslint/parser",
-  "parserOptions": {
-    "ecmaVersion": 12,
-    "sourceType": "module"
-  },
-  "plugins": ["@typescript-eslint"],
-  "rules": {
-    "indent": "off", // 禁用缩进规则
-    "linebreak-style": ["warn", "unix"],
-    "semi": ["warn", "always"],
-    "no-unused-vars": "off",
-    "@typescript-eslint/no-unused-vars": ["warn"],
-    "@typescript-eslint/no-explicit-any": "off" // 禁用对 any 类型的检查
-  }
-}

package/.turbo/turbo-build.log

@@ -1,21 +0,0 @@
-
-> @axiom-lattice/core@1.0.10 build /home/runner/work/agentic/agentic/packages/core
-> tsup src/index.ts --format cjs,esm --dts --clean --sourcemap
-
-CLI Building entry: src/index.ts
-CLI Using tsconfig: tsconfig.json
-CLI tsup v8.5.0
-CLI Target: es2020
-CLI Cleaning output folder
-CJS Build start
-ESM Build start
-CJS dist/index.js     71.87 KB
-CJS dist/index.js.map 125.81 KB
-CJS ⚡️ Build success in 201ms
-ESM dist/index.mjs     67.73 KB
-ESM dist/index.mjs.map 125.90 KB
-ESM ⚡️ Build success in 199ms
-DTS Build start
-DTS ⚡️ Build success in 9506ms
-DTS dist/index.d.ts  9.15 KB
-DTS dist/index.d.mts 9.15 KB

package/jest.config.js

@@ -1,21 +0,0 @@
-module.exports = {
-  preset: "ts-jest",
-  testEnvironment: "node",
-  roots: ["<rootDir>/src"],
-  testMatch: ["**/__tests__/**/*.test.ts", "**/?(*.)+(spec|test).ts"],
-  transform: {
-    "^.+\\.ts$": [
-      "ts-jest",
-      {
-        isolatedModules: true,
-      },
-    ],
-  },
-  moduleNameMapper: {
-    "^@/(.*)$": "<rootDir>/src/$1",
-    "^@utils/(.*)$": "<rootDir>/src/utils/$1",
-  },
-  collectCoverageFrom: ["src/**/*.ts", "!src/**/*.d.ts", "!src/__tests__/**/*"],
-  coverageReporters: ["text", "lcov", "html"],
-  testTimeout: 30000,
-};

package/src/__tests__/AgentManager.test.ts

@@ -1,202 +0,0 @@
-import { AgentManager, agentManager, AgentDefinition } from "../AgentManager";
-import { CompiledStateGraph } from "@langchain/langgraph";
-
-// 模拟编译后的状态图
-const mockExecutor = {
-  invoke: jest.fn().mockResolvedValue({ result: "success" }),
-} as unknown as CompiledStateGraph<any, any, any, any, any, any>;
-
-describe("AgentManager", () => {
-  let manager: AgentManager;
-
-  beforeEach(() => {
-    // 获取单例实例
-    manager = AgentManager.getInstance();
-    // 清空代理
-    manager.clearAgents();
-    // 重置模拟函数
-    jest.clearAllMocks();
-  });
-
-  describe("单例模式", () => {
-    it("应该返回相同的实例", () => {
-      const instance1 = AgentManager.getInstance();
-      const instance2 = AgentManager.getInstance();
-      expect(instance1).toBe(instance2);
-    });
-
-    it("应该返回相同的agentManager实例", () => {
-      expect(agentManager).toBe(manager);
-    });
-  });
-
-  describe("代理注册", () => {
-    const testAgentDefinition: AgentDefinition = {
-      name: "测试代理",
-      description: "用于测试的代理",
-      agent_type: "test",
-      agent_config: { model: "test-model" },
-      agent_tools: ["tool1", "tool2"],
-      agent_flow: ["start", "end"],
-    };
-
-    it("应该能够注册代理", () => {
-      manager.registerAgent(testAgentDefinition, mockExecutor, [
-        "tool1",
-        "tool2",
-      ]);
-      expect(manager.hasAgent("测试代理")).toBe(true);
-      expect(manager.getAgentCount()).toBe(1);
-    });
-
-    it("应该防止重复注册同名代理", () => {
-      manager.registerAgent(testAgentDefinition, mockExecutor, [
-        "tool1",
-        "tool2",
-      ]);
-
-      expect(() => {
-        manager.registerAgent(testAgentDefinition, mockExecutor, [
-          "tool1",
-          "tool2",
-        ]);
-      }).toThrow('代理 "测试代理" 已经存在，无法重复注册');
-    });
-
-    it("应该能够批量注册代理", () => {
-      const agent1 = {
-        definition: { ...testAgentDefinition, name: "代理1" },
-        executor: mockExecutor,
-        tools: ["tool1"],
-      };
-      const agent2 = {
-        definition: { ...testAgentDefinition, name: "代理2" },
-        executor: mockExecutor,
-        tools: ["tool2"],
-      };
-
-      manager.registerAgents([agent1, agent2]);
-      expect(manager.getAgentCount()).toBe(2);
-      expect(manager.hasAgent("代理1")).toBe(true);
-      expect(manager.hasAgent("代理2")).toBe(true);
-    });
-  });
-
-  describe("代理管理", () => {
-    const testAgents = [
-      {
-        definition: {
-          name: "代理1",
-          description: "第一个代理",
-          agent_type: "test",
-          agent_config: {},
-          agent_tools: [],
-          agent_flow: ["start", "end"],
-        },
-        executor: mockExecutor,
-        tools: [],
-      },
-      {
-        definition: {
-          name: "代理2",
-          description: "第二个代理",
-          agent_type: "test",
-          agent_config: {},
-          agent_tools: [],
-          agent_flow: ["start", "end"],
-        },
-        executor: mockExecutor,
-        tools: [],
-      },
-    ];
-
-    beforeEach(() => {
-      manager.registerAgents(testAgents);
-    });
-
-    it("应该能够获取代理定义", () => {
-      const definition = manager.getAgentDefinition("代理1");
-      expect(definition).toBeDefined();
-      expect(definition?.name).toBe("代理1");
-    });
-
-    it("应该能够获取所有代理定义", () => {
-      const definitions = manager.getAllAgentDefinitions();
-      expect(definitions).toHaveLength(2);
-      expect(definitions.map((d) => d.name)).toEqual(["代理1", "代理2"]);
-    });
-
-    it("应该能够获取代理名称列表", () => {
-      const names = manager.getAgentNames();
-      expect(names).toEqual(["代理1", "代理2"]);
-    });
-
-    it("应该能够移除代理", () => {
-      expect(manager.removeAgent("代理1")).toBe(true);
-      expect(manager.hasAgent("代理1")).toBe(false);
-      expect(manager.getAgentCount()).toBe(1);
-    });
-
-    it("应该能够清空所有代理", () => {
-      manager.clearAgents();
-      expect(manager.getAgentCount()).toBe(0);
-      expect(manager.getAgentNames()).toHaveLength(0);
-    });
-
-    it("应该能够检查代理是否存在", () => {
-      expect(manager.hasAgent("代理1")).toBe(true);
-      expect(manager.hasAgent("不存在的代理")).toBe(false);
-    });
-  });
-
-  describe("代理查询", () => {
-    const testAgentDefinition: AgentDefinition = {
-      name: "查询测试代理",
-      description: "用于查询测试的代理",
-      agent_type: "test",
-      agent_config: {},
-      agent_tools: ["tool1"],
-      agent_flow: ["start", "end"],
-    };
-
-    beforeEach(() => {
-      manager.registerAgent(testAgentDefinition, mockExecutor, ["tool1"]);
-    });
-
-    it("应该能够获取代理执行器", () => {
-      const executor = manager.getAgentExecutor("查询测试代理");
-      expect(executor).toBeDefined();
-      expect(executor).toBe(mockExecutor);
-    });
-
-    it("应该能够获取代理工具列表", () => {
-      const tools = manager.getAgentTools("查询测试代理");
-      expect(tools).toEqual(["tool1"]);
-    });
-
-    it("应该返回undefined当代理不存在", () => {
-      const executor = manager.getAgentExecutor("不存在的代理");
-      const tools = manager.getAgentTools("不存在的代理");
-      expect(executor).toBeUndefined();
-      expect(tools).toEqual([]);
-    });
-  });
-
-  describe("便捷函数", () => {
-    it("应该能够通过便捷函数注册代理", () => {
-      const { registerAgent } = require("../AgentManager");
-
-      const agentDef: AgentDefinition = {
-        name: "便捷函数测试代理",
-        description: "测试便捷函数",
-        agent_type: "test",
-        agent_config: {},
-        agent_tools: [],
-        agent_flow: ["start", "end"],
-      };
-
-      registerAgent(agentDef, mockExecutor, []);
-      expect(agentManager.hasAgent("便捷函数测试代理")).toBe(true);
-    });
-  });
-});

package/src/__tests__/setup.ts

@@ -1,5 +0,0 @@
-// 全局测试设置文件
-// 这个文件会在所有测试运行前执行
-
-// 设置Node.js环境变量
-process.env.NODE_ENV = "test";

package/src/agent_lattice/AgentLatticeManager.ts

@@ -1,216 +0,0 @@
-import { BaseLatticeManager } from "../base/BaseLatticeManager";
-import {
-  AgentConfig,
-  AgentClient,
-  AgentLattice,
-  GraphBuildOptions,
-  AgentBuildParams,
-} from "./types";
-import { AgentGraphBuilderFactory } from "./builders/AgentGraphBuilderFactory";
-import { AgentParamsBuilder } from "./builders/AgentParamsBuilder";
-
-// 类型定义已移至 ./types.ts
-
-/**
- * AgentLatticeManager - 单例Agent Lattice管理器
- * 负责注册、管理各种Agent Lattice
- */
-export class AgentLatticeManager extends BaseLatticeManager<AgentLattice> {
-  private static _instance: AgentLatticeManager;
-
-  /**
-   * 获取AgentLatticeManager单例实例
-   */
-  public static getInstance(): AgentLatticeManager {
-    if (!AgentLatticeManager._instance) {
-      AgentLatticeManager._instance = new AgentLatticeManager();
-    }
-    return AgentLatticeManager._instance;
-  }
-
-  /**
-   * 获取Lattice类型前缀
-   */
-  protected getLatticeType(): string {
-    return "agents";
-  }
-
-  /**
-   * 注册Agent Lattice
-   * @param config Agent配置
-   */
-  public registerLattice(config: AgentConfig): void {
-    // 创建Agent Lattice对象
-    const agentLattice: AgentLattice = {
-      config,
-      client: undefined, // 客户端将在需要时由initializeClient创建
-    };
-
-    // 调用基类的register方法
-    this.register(config.key, agentLattice);
-  }
-
-  /**
-   * 获取AgentLattice
-   * @param key Lattice键名
-   */
-  public getAgentLattice(key: string): AgentLattice | undefined {
-    return this.get(key);
-  }
-
-  /**
-   * 获取所有Lattice
-   */
-  public getAllLattices(): AgentLattice[] {
-    return this.getAll();
-  }
-
-  /**
-   * 检查Lattice是否存在
-   * @param key Lattice键名
-   */
-  public hasLattice(key: string): boolean {
-    return this.has(key);
-  }
-
-  /**
-   * 移除Lattice
-   * @param key Lattice键名
-   */
-  public removeLattice(key: string): boolean {
-    return this.remove(key);
-  }
-
-  /**
-   * 获取Agent配置
-   * @param key Lattice键名
-   */
-  public getAgentConfig(key: string): AgentConfig | undefined {
-    return this.getAgentLattice(key)?.config;
-  }
-
-  /**
-   * 获取所有Agent配置
-   */
-  public getAllAgentConfigs(): AgentConfig[] {
-    return this.getAllLattices().map((lattice) => lattice.config);
-  }
-
-  /**
-   * 验证Agent输入参数
-   * @param key Lattice键名
-   * @param input 输入参数
-   */
-  public validateAgentInput(key: string, input: any): boolean {
-    const agentLattice = this.getAgentLattice(key);
-    if (!agentLattice) {
-      return false;
-    }
-
-    try {
-      if (agentLattice.config.schema) {
-        agentLattice.config.schema.parse(input);
-      }
-      return true;
-    } catch {
-      return false;
-    }
-  }
-
-  /**
-   * 构建Agent参数
-   *
-   * @param agentLattice Agent Lattice对象
-   * @param options 构建选项
-   * @returns 返回Agent构建参数
-   */
-  private buildAgentParams(
-    agentLattice: AgentLattice,
-    options?: GraphBuildOptions
-  ): AgentBuildParams {
-    // 创建参数构建器
-    const paramsBuilder = new AgentParamsBuilder((key: string) =>
-      this.getAgentLattice(key)
-    );
-
-    // 构建参数
-    return paramsBuilder.buildParams(agentLattice, options);
-  }
-
-  /**
-   * 初始化Agent客户端
-   *
-   * 使用AgentGraphBuilderFactory构建Graph并设置为客户端
-   *
-   * @param key Lattice键名
-   * @param options 构建选项
-   * @returns 返回CompiledGraph对象
-   */
-  public initializeClient(
-    key: string,
-    options?: GraphBuildOptions
-  ): AgentClient {
-    const agentLattice = this.getAgentLattice(key);
-    if (!agentLattice) {
-      throw new Error(`Agent Lattice "${key}" 不存在`);
-    }
-
-    // 如果客户端已经存在，直接返回
-    if (agentLattice.client) {
-      return agentLattice.client;
-    }
-
-    // 获取工厂实例
-    const factory = AgentGraphBuilderFactory.getInstance();
-
-    // 获取对应类型的Builder
-    const builder = factory.getBuilder(agentLattice.config.type);
-
-    // 构建参数
-    const params = this.buildAgentParams(agentLattice, options);
-
-    // 使用Builder构建Graph
-    const graph = builder.build(agentLattice, params);
-
-    // 设置为客户端
-    agentLattice.client = graph;
-
-    return graph;
-  }
-}
-
-// 导出单例实例
-export const agentLatticeManager = AgentLatticeManager.getInstance();
-
-// 导出便捷方法
-export const registerAgentLattice = (config: AgentConfig) => {
-  agentLatticeManager.registerLattice(config);
-};
-
-export const registerAgentLattices = (configs: AgentConfig[]) => {
-  configs.forEach((config) => {
-    agentLatticeManager.registerLattice(config);
-  });
-};
-
-export const getAgentLattice = (key: string) =>
-  agentLatticeManager.getAgentLattice(key);
-
-export const getAgentConfig = (key: string) =>
-  agentLatticeManager.getAgentConfig(key);
-
-export const getAllAgentConfigs = () =>
-  agentLatticeManager.getAllAgentConfigs();
-
-export const validateAgentInput = (key: string, input: any) =>
-  agentLatticeManager.validateAgentInput(key, input);
-
-/**
- * 获取或初始化Agent客户端
- *
- * @param key Agent Lattice键名
- * @param options 构建选项
- * @returns 返回CompiledGraph对象
- */
-export const getAgentClient = (key: string, options?: GraphBuildOptions) =>
-  agentLatticeManager.initializeClient(key, options);

package/src/agent_lattice/builders/AgentBuilder.ts

@@ -1,79 +0,0 @@
-/**
- * Agent Builder 服务
- *
- * 提供构建和执行Agent的功能
- */
-
-import { CompiledStateGraph } from "@langchain/langgraph";
-import { AgentLattice, GraphBuildOptions } from "../types";
-import { AgentGraphBuilderFactory } from "./AgentGraphBuilderFactory";
-import {
-  AgentParamsBuilder,
-  GetAgentLatticeFunction,
-} from "./AgentParamsBuilder";
-
-/**
- * Agent Builder 服务
- */
-export class AgentBuilder {
-  private graphBuilderFactory: AgentGraphBuilderFactory;
-  private paramsBuilder: AgentParamsBuilder;
-
-  /**
-   * 构造函数
-   *
-   * @param getAgentLatticeFunc 获取Agent Lattice的函数
-   */
-  constructor(getAgentLatticeFunc: GetAgentLatticeFunction) {
-    this.graphBuilderFactory = AgentGraphBuilderFactory.getInstance();
-    this.paramsBuilder = new AgentParamsBuilder(getAgentLatticeFunc);
-  }
-
-  /**
-   * 构建Agent Graph
-   *
-   * @param agentLattice Agent Lattice对象
-   * @param options 构建选项
-   * @returns 返回CompiledGraph对象
-   */
-  public buildGraph(
-    agentLattice: AgentLattice,
-    options?: GraphBuildOptions
-  ): CompiledStateGraph<any, any, any, any, any> {
-    // 准备构建参数
-    const buildParams = this.paramsBuilder.buildParams(agentLattice, options);
-
-    // 获取对应的Builder
-    const builder = this.graphBuilderFactory.getBuilder(
-      agentLattice.config.type
-    );
-
-    // 构建Graph
-    return builder.build(agentLattice, buildParams);
-  }
-
-  /**
-   * 执行Agent
-   *
-   * @param agentLattice Agent Lattice对象
-   * @param input 输入参数
-   * @param options 执行选项
-   * @returns 返回执行结果
-   */
-  public async execute<TInput = any, TOutput = any>(
-    agentLattice: AgentLattice,
-    input: TInput,
-    options?: GraphBuildOptions & { sessionId?: string }
-  ): Promise<TOutput> {
-    // 构建Graph
-    const graph = this.buildGraph(agentLattice, options);
-
-    // 执行Graph
-    const config: any = {};
-    if (options?.metadata) {
-      config.metadata = options.metadata;
-    }
-
-    return (await graph.invoke({ input }, config)) as TOutput;
-  }
-}

package/src/agent_lattice/builders/AgentGraphBuilder.ts

@@ -1,22 +0,0 @@
-/**
- * Agent Graph Builder 接口
- *
- * 策略模式的核心接口，定义了构建Agent Graph的方法
- */
-
-import { CompiledStateGraph } from "@langchain/langgraph";
-import { AgentLattice, AgentBuildParams } from "../types";
-
-export interface AgentGraphBuilder {
-  /**
-   * 构建Agent Graph
-   *
-   * @param agentLattice Agent Lattice对象
-   * @param params Agent构建参数
-   * @returns 返回CompiledGraph对象
-   */
-  build(
-    agentLattice: AgentLattice,
-    params: AgentBuildParams
-  ): CompiledStateGraph<any, any, any, any, any>;
-}

package/src/agent_lattice/builders/AgentGraphBuilderFactory.ts

@@ -1,70 +0,0 @@
-/**
- * Agent Graph Builder Factory
- *
- * 工厂类，根据Agent类型创建对应的Builder
- */
-
-import { AgentType } from "../types";
-import { AgentGraphBuilder } from "./AgentGraphBuilder";
-import { ReActAgentGraphBuilder } from "./ReActAgentGraphBuilder";
-import { DeepAgentGraphBuilder } from "./DeepAgentGraphBuilder";
-import { PlanExecuteAgentGraphBuilder } from "./PlanExecuteAgentGraphBuilder";
-
-export class AgentGraphBuilderFactory {
-  // 单例模式
-  private static instance: AgentGraphBuilderFactory;
-
-  // Builder缓存
-  private builders: Map<AgentType, AgentGraphBuilder>;
-
-  private constructor() {
-    this.builders = new Map();
-    this.registerDefaultBuilders();
-  }
-
-  /**
-   * 获取单例实例
-   */
-  public static getInstance(): AgentGraphBuilderFactory {
-    if (!AgentGraphBuilderFactory.instance) {
-      AgentGraphBuilderFactory.instance = new AgentGraphBuilderFactory();
-    }
-    return AgentGraphBuilderFactory.instance;
-  }
-
-  /**
-   * 注册默认的Builder
-   */
-  private registerDefaultBuilders(): void {
-    this.builders.set(AgentType.REACT, new ReActAgentGraphBuilder());
-    this.builders.set(AgentType.DEEP_AGENT, new DeepAgentGraphBuilder());
-    this.builders.set(
-      AgentType.PLAN_EXECUTE,
-      new PlanExecuteAgentGraphBuilder()
-    );
-  }
-
-  /**
-   * 注册自定义Builder
-   *
-   * @param type Agent类型
-   * @param builder Builder实例
-   */
-  public registerBuilder(type: AgentType, builder: AgentGraphBuilder): void {
-    this.builders.set(type, builder);
-  }
-
-  /**
-   * 获取Builder
-   *
-   * @param type Agent类型
-   * @returns 返回对应的Builder
-   */
-  public getBuilder(type: AgentType): AgentGraphBuilder {
-    const builder = this.builders.get(type);
-    if (!builder) {
-      throw new Error(`不支持的Agent类型: ${type}`);
-    }
-    return builder;
-  }
-}