@axiom-lattice/core 1.0.13 → 1.0.21

package/src/deep_agent/tools.ts

@@ -1,273 +0,0 @@
-/**
- * Tool functions for Deep Agents
- *
- * TypeScript versions of all tools using @langchain/core/tools tool() function.
- * Uses getCurrentTaskInput() for state access and returns Command objects for state updates.
- * Implements mock filesystem operations using state.files similar to Python version.
- */
-
-import { tool, ToolRunnableConfig } from "@langchain/core/tools";
-import { ToolMessage } from "@langchain/core/messages";
-import { Command, getCurrentTaskInput } from "@langchain/langgraph";
-import { z } from "zod";
-import {
-  WRITE_TODOS_DESCRIPTION,
-  EDIT_DESCRIPTION,
-  TOOL_DESCRIPTION,
-} from "./prompts";
-import { DeepAgentStateType } from "./types";
-
-/**
- * Write todos tool - manages todo list with Command return
- * Uses getCurrentTaskInput() instead of Python's InjectedState
- */
-export const writeTodos = tool(
-  (input, config: ToolRunnableConfig) => {
-    return new Command({
-      update: {
-        todos: input.todos,
-        messages: [
-          new ToolMessage({
-            content: `Updated todo list to ${JSON.stringify(input.todos)}`,
-            tool_call_id: config.toolCall?.id as string,
-          }),
-        ],
-      },
-    });
-  },
-
-  {
-    name: "write_todos",
-    description: WRITE_TODOS_DESCRIPTION,
-    schema: z.object({
-      todos: z
-        .array(
-          z.object({
-            content: z.string().describe("Content of the todo item"),
-            status: z
-              .enum(["pending", "in_progress", "completed"])
-              .describe("Status of the todo"),
-          })
-        )
-        .describe("List of todo items to update"),
-    }),
-  }
-);
-
-/**
- * List files tool - returns list of files from state.files
- * Equivalent to Python's ls function
- */
-export const ls = tool(
-  () => {
-    const state = getCurrentTaskInput<DeepAgentStateType>();
-    const files = state.files || {};
-    return Object.keys(files);
-  },
-  {
-    name: "ls",
-    description: "List all files in the mock filesystem",
-    schema: z.object({}),
-  }
-);
-
-/**
- * Read file tool - reads from mock filesystem in state.files
- * Matches Python read_file function behavior exactly
- */
-export const readFile = tool(
-  (input: { file_path: string; offset?: number; limit?: number }) => {
-    const state = getCurrentTaskInput<DeepAgentStateType>();
-    const mockFilesystem = state.files || {};
-    const { file_path, offset = 0, limit = 2000 } = input;
-
-    if (!(file_path in mockFilesystem)) {
-      return `Error: File '${file_path}' not found`;
-    }
-
-    // Get file content
-    const content = mockFilesystem[file_path];
-
-    // Handle empty file
-    if (!content || content.trim() === "") {
-      return "System reminder: File exists but has empty contents";
-    }
-
-    // Split content into lines
-    const lines = content.split("\n");
-
-    // Apply line offset and limit
-    const startIdx = offset;
-    const endIdx = Math.min(startIdx + limit, lines.length);
-
-    // Handle case where offset is beyond file length
-    if (startIdx >= lines.length) {
-      return `Error: Line offset ${offset} exceeds file length (${lines.length} lines)`;
-    }
-
-    // Format output with line numbers (cat -n format)
-    const resultLines: string[] = [];
-    for (let i = startIdx; i < endIdx; i++) {
-      let lineContent = lines[i];
-
-      // Truncate lines longer than 2000 characters
-      if (lineContent.length > 2000) {
-        lineContent = lineContent.substring(0, 2000);
-      }
-
-      // Line numbers start at 1, so add 1 to the index
-      const lineNumber = i + 1;
-      resultLines.push(`${lineNumber.toString().padStart(6)}	${lineContent}`);
-    }
-
-    return resultLines.join("\n");
-  },
-  {
-    name: "read_file",
-    description: TOOL_DESCRIPTION,
-    schema: z.object({
-      file_path: z.string().describe("Absolute path to the file to read"),
-      offset: z
-        .number()
-        .optional()
-        .default(0)
-        .describe("Line offset to start reading from"),
-      limit: z
-        .number()
-        .optional()
-        .default(2000)
-        .describe("Maximum number of lines to read"),
-    }),
-  }
-);
-
-/**
- * Write file tool - writes to mock filesystem with Command return
- * Matches Python write_file function behavior exactly
- */
-export const writeFile = tool(
-  (
-    input: { file_path: string; content: string },
-    config: ToolRunnableConfig
-  ) => {
-    const state = getCurrentTaskInput<DeepAgentStateType>();
-    const files = { ...(state.files || {}) };
-    files[input.file_path] = input.content;
-
-    return new Command({
-      update: {
-        files: files,
-        messages: [
-          new ToolMessage({
-            content: `Updated file ${input.file_path}`,
-            tool_call_id: config.toolCall?.id as string,
-          }),
-        ],
-      },
-    });
-  },
-  {
-    name: "write_file",
-    description: "Write content to a file in the mock filesystem",
-    schema: z.object({
-      file_path: z.string().describe("Absolute path to the file to write"),
-      content: z.string().describe("Content to write to the file"),
-    }),
-  }
-);
-
-/**
- * Edit file tool - edits files in mock filesystem with Command return
- * Matches Python edit_file function behavior exactly
- */
-export const editFile = tool(
-  (
-    input: {
-      file_path: string;
-      old_string: string;
-      new_string: string;
-      replace_all?: boolean;
-    },
-    config: ToolRunnableConfig
-  ) => {
-    const state = getCurrentTaskInput<DeepAgentStateType>();
-    const mockFilesystem = { ...(state.files || {}) };
-    const { file_path, old_string, new_string, replace_all = false } = input;
-
-    // Check if file exists in mock filesystem
-    if (!(file_path in mockFilesystem)) {
-      return `Error: File '${file_path}' not found`;
-    }
-
-    // Get current file content
-    const content = mockFilesystem[file_path];
-
-    // Check if old_string exists in the file
-    if (!content.includes(old_string)) {
-      return `Error: String not found in file: '${old_string}'`;
-    }
-
-    // If not replace_all, check for uniqueness
-    if (!replace_all) {
-      const escapedOldString = old_string.replace(
-        /[.*+?^${}()|[\]\\]/g,
-        "\\$&"
-      );
-      const occurrences = (
-        content.match(new RegExp(escapedOldString, "g")) || []
-      ).length;
-      if (occurrences > 1) {
-        return `Error: String '${old_string}' appears ${occurrences} times in file. Use replace_all=True to replace all instances, or provide a more specific string with surrounding context.`;
-      } else if (occurrences === 0) {
-        return `Error: String not found in file: '${old_string}'`;
-      }
-    }
-
-    // Perform the replacement
-    let newContent: string;
-
-    if (replace_all) {
-      const escapedOldString = old_string.replace(
-        /[.*+?^${}()|[\]\\]/g,
-        "\\$&"
-      );
-      newContent = content.replace(
-        new RegExp(escapedOldString, "g"),
-        new_string
-      );
-    } else {
-      newContent = content.replace(old_string, new_string);
-    }
-
-    // Update the mock filesystem
-    mockFilesystem[file_path] = newContent;
-
-    return new Command({
-      update: {
-        files: mockFilesystem,
-        messages: [
-          new ToolMessage({
-            content: `Updated file ${file_path}`,
-            tool_call_id: config.toolCall?.id as string,
-          }),
-        ],
-      },
-    });
-  },
-  {
-    name: "edit_file",
-    description: EDIT_DESCRIPTION,
-    schema: z.object({
-      file_path: z.string().describe("Absolute path to the file to edit"),
-      old_string: z
-        .string()
-        .describe("String to be replaced (must match exactly)"),
-      new_string: z.string().describe("String to replace with"),
-      replace_all: z
-        .boolean()
-        .optional()
-        .default(false)
-        .describe("Whether to replace all occurrences"),
-    }),
-  }
-);

package/src/deep_agent/types.ts

@@ -1,71 +0,0 @@
-/**
- * TypeScript type definitions for Deep Agents
- *
- * This file contains all the TypeScript interfaces and types that correspond
- * to the Python TypedDict and other type definitions. Defines all necessary
- * TypeScript interfaces and types including StateSchemaType, SubAgent, Todo,
- * and proper generic types for state schemas.
- */
-
-import type {
-  BaseLanguageModelInput,
-  LanguageModelOutput,
-} from "@langchain/core/language_models/base";
-import type { StructuredTool } from "@langchain/core/tools";
-import type { DeepAgentState } from "./state";
-import { z } from "zod";
-import { Runnable } from "@langchain/core/runnables";
-import { ModelLattice } from "../model_lattice/ModelLattice";
-
-export type InferZodObjectShape<T> = T extends z.ZodObject<infer Shape>
-  ? Shape
-  : never;
-
-/**
- * SubAgent interface matching Python's TypedDict structure
- */
-export interface SubAgent {
-  name: string;
-  description: string;
-  prompt: string;
-  tools?: string[];
-}
-
-export type TodoStatus = "pending" | "in_progress" | "completed";
-
-export interface Todo {
-  content: string;
-  status: TodoStatus;
-}
-
-export type DeepAgentStateType = z.infer<typeof DeepAgentState>;
-
-export type LanguageModelLike = Runnable<
-  BaseLanguageModelInput,
-  LanguageModelOutput
->;
-
-/**
- * Parameters for createDeepAgent function with TypeScript types
- */
-export interface CreateDeepAgentParams<
-  StateSchema extends z.ZodObject<any, any, any, any, any>
-> {
-  tools?: StructuredTool[];
-  instructions?: string;
-  model?: ModelLattice;
-  subagents?: SubAgent[];
-  stateSchema?: StateSchema;
-}
-
-/**
- * Parameters for createTaskTool function
- */
-export interface CreateTaskToolParams<
-  StateSchema extends z.ZodObject<any, any, any, any, any>
-> {
-  subagents: SubAgent[];
-  tools?: Record<string, StructuredTool>;
-  model?: ModelLattice;
-  stateSchema?: StateSchema;
-}

package/src/index.ts

@@ -1,9 +0,0 @@
-// 导出核心模块
-export * from "./model_lattice";
-export * from "./tool_lattice";
-export * from "./agent_lattice";
-export * from "./memory_lattice";
-
-// 重新导出协议接口 - 使用命名导出避免冲突
-import * as Protocols from "@axiom-lattice/protocols";
-export { Protocols };

package/src/logger/Logger.ts

@@ -1,186 +0,0 @@
-import pino from "pino";
-import "pino-pretty";
-import "pino-roll";
-import { AsyncLocalStorage } from "async_hooks";
-
-export interface LoggerContext {
-  "x-user-id"?: string;
-  "x-tenant-id"?: string;
-  "x-request-id"?: string;
-  "x-task-id"?: string;
-  "x-thread-id"?: string;
-}
-
-export interface LoggerOptions {
-  name?: string;
-  serviceName?: string;
-  context?: LoggerContext;
-}
-
-/**
- * 单例的Pino日志工厂类，管理底层pino实例
- */
-class PinoLoggerFactory {
-  private static instance: PinoLoggerFactory;
-  private pinoLogger: pino.Logger;
-
-  private constructor() {
-    const isProd = process.env.NODE_ENV === "production";
-
-    const loggerConfig: pino.LoggerOptions = {
-      // 自定义时间戳格式
-      timestamp: () => `,"@timestamp":"${new Date().toISOString()}"`,
-
-      // 关闭默认的时间戳键
-      base: {
-        "@version": "1",
-        app_name: "lattice",
-        service_name: "lattice/graph-server",
-        thread_name: "main",
-        logger_name: "lattice-graph-logger",
-      },
-
-      formatters: {
-        level: (label, number) => {
-          return {
-            level: label.toUpperCase(),
-            level_value: number * 1000,
-          };
-        },
-      },
-    };
-
-    // 生产环境使用文件日志
-    if (isProd) {
-      try {
-        this.pinoLogger = pino(
-          loggerConfig,
-          pino.transport({
-            target: "pino-roll",
-            options: {
-              file: "./logs/fin_ai_graph_server",
-              frequency: "daily",
-              mkdir: true,
-            },
-          })
-        );
-      } catch (error) {
-        console.error(
-          "无法初始化 pino-roll 日志记录器，回退到控制台日志",
-          error
-        );
-        // 回退到开发环境的配置
-        this.pinoLogger = pino({
-          ...loggerConfig,
-          transport: {
-            target: "pino-pretty",
-            options: {
-              colorize: true,
-            },
-          },
-        });
-      }
-    } else {
-      // 开发环境使用格式化输出
-      this.pinoLogger = pino({
-        ...loggerConfig,
-        transport: {
-          target: "pino-pretty",
-          options: {
-            colorize: true,
-          },
-        },
-      });
-    }
-  }
-
-  public static getInstance(): PinoLoggerFactory {
-    if (!PinoLoggerFactory.instance) {
-      PinoLoggerFactory.instance = new PinoLoggerFactory();
-    }
-    return PinoLoggerFactory.instance;
-  }
-
-  public getPinoLogger(): pino.Logger {
-    return this.pinoLogger;
-  }
-}
-
-/**
- * Logger类，可以创建多个实例，每个实例有自己的上下文
- */
-export class Logger {
-  private context: LoggerContext;
-  private name: string;
-  private serviceName: string;
-
-  constructor(options?: LoggerOptions) {
-    this.context = options?.context || {};
-    this.name = options?.name || "lattice-graph-logger";
-    this.serviceName = options?.serviceName || "lattice/graph-server";
-  }
-
-  /**
-   * 获取合并了上下文的日志对象
-   * @param additionalContext 额外的上下文数据
-   * @returns 带有上下文的pino日志对象
-   */
-  private getContextualLogger(additionalContext?: object): pino.Logger {
-    const pinoLogger = PinoLoggerFactory.getInstance().getPinoLogger();
-
-    // 合并Logger实例的上下文和额外上下文
-    const contextObj = {
-      "x-user-id": this.context["x-user-id"] || "",
-      "x-tenant-id": this.context["x-tenant-id"] || "",
-      "x-request-id": this.context["x-request-id"] || "",
-      "x-task-id": this.context["x-task-id"] || "",
-      "x-thread-id": this.context["x-thread-id"] || "",
-      service_name: this.serviceName,
-      logger_name: this.name,
-      ...additionalContext,
-    };
-
-    // 创建带有上下文的子日志记录器
-    return pinoLogger.child(contextObj);
-  }
-
-  info(msg: string, obj?: object): void {
-    this.getContextualLogger(obj).info(msg);
-  }
-
-  error(msg: string, obj?: object | Error): void {
-    this.getContextualLogger(obj).error(msg);
-  }
-
-  warn(msg: string, obj?: object): void {
-    this.getContextualLogger(obj).warn(msg);
-  }
-
-  debug(msg: string, obj?: object): void {
-    this.getContextualLogger(obj).debug(msg);
-  }
-
-  /**
-   * 更新Logger实例的上下文
-   */
-  updateContext(context: Partial<LoggerContext>): void {
-    this.context = {
-      ...this.context,
-      ...context,
-    };
-  }
-
-  /**
-   * 创建一个新的Logger实例，继承当前Logger的上下文
-   */
-  child(options: Partial<LoggerOptions>): Logger {
-    return new Logger({
-      name: options.name || this.name,
-      serviceName: options.serviceName || this.serviceName,
-      context: {
-        ...this.context,
-        ...options.context,
-      },
-    });
-  }
-}

package/src/memory_lattice/DefaultMemorySaver.ts

@@ -1,4 +0,0 @@
-import { MemorySaver } from "@langchain/langgraph";
-import { registerCheckpointSaver } from "./MemoryLatticeManager";
-const memory = new MemorySaver();
-registerCheckpointSaver("default", memory);

package/src/memory_lattice/MemoryLatticeManager.ts

@@ -1,105 +0,0 @@
-/**
- * MemoryLatticeManager
- *
- * 记忆Lattice管理器，负责管理和注册检查点保存器
- */
-
-import { BaseCheckpointSaver } from "@langchain/langgraph-checkpoint";
-import { BaseLatticeManager } from "../base/BaseLatticeManager";
-import { MemoryLatticeProtocol, MemoryType } from "@axiom-lattice/protocols";
-
-// Re-export types from protocols
-export { MemoryType };
-
-/**
- * 记忆Lattice管理器类
- */
-export class MemoryLatticeManager extends BaseLatticeManager {
-  private static instance: MemoryLatticeManager;
-
-  // 检查点保存器注册表
-  private static checkpointSavers: Map<string, BaseCheckpointSaver> = new Map();
-
-  /**
-   * 私有构造函数，防止外部直接实例化
-   */
-  private constructor() {
-    super();
-  }
-
-  /**
-   * 获取单例实例
-   */
-  public static getInstance(): MemoryLatticeManager {
-    if (!MemoryLatticeManager.instance) {
-      MemoryLatticeManager.instance = new MemoryLatticeManager();
-    }
-    return MemoryLatticeManager.instance;
-  }
-
-  /**
-   * 获取Lattice类型
-   */
-  protected getLatticeType(): string {
-    return "memory";
-  }
-
-  /**
-   * 注册检查点保存器
-   * @param key 保存器键名
-   * @param saver 检查点保存器实例
-   */
-  public registerCheckpointSaver(
-    key: string,
-    saver: BaseCheckpointSaver
-  ): void {
-    if (MemoryLatticeManager.checkpointSavers.has(key)) {
-      throw new Error(`检查点保存器 "${key}" 已经存在，无法重复注册`);
-    }
-
-    MemoryLatticeManager.checkpointSavers.set(key, saver);
-  }
-
-  /**
-   * 获取检查点保存器
-   * @param key 保存器键名
-   */
-  public getCheckpointSaver(key: string): BaseCheckpointSaver {
-    const saver = MemoryLatticeManager.checkpointSavers.get(key);
-    if (!saver) {
-      throw new Error(`检查点保存器 "${key}" 不存在`);
-    }
-    return saver;
-  }
-
-  /**
-   * 获取所有已注册的检查点保存器键名
-   */
-  public getCheckpointSaverKeys(): string[] {
-    return Array.from(MemoryLatticeManager.checkpointSavers.keys());
-  }
-
-  /**
-   * 检查检查点保存器是否存在
-   * @param key 保存器键名
-   */
-  public hasCheckpointSaver(key: string): boolean {
-    return MemoryLatticeManager.checkpointSavers.has(key);
-  }
-
-  /**
-   * 移除检查点保存器
-   * @param key 保存器键名
-   */
-  public removeCheckpointSaver(key: string): boolean {
-    return MemoryLatticeManager.checkpointSavers.delete(key);
-  }
-}
-
-export const getCheckpointSaver = (key: string) =>
-  MemoryLatticeManager.getInstance().getCheckpointSaver(key);
-
-export const registerCheckpointSaver = (
-  key: string,
-  saver: BaseCheckpointSaver
-) => MemoryLatticeManager.getInstance().registerCheckpointSaver(key, saver);

package/src/memory_lattice/index.ts

@@ -1,9 +0,0 @@
-/**
- * Memory Lattice
- *
- * 记忆Lattice模块，提供检查点保存器管理功能
- */
-
-import "./DefaultMemorySaver";
-
-export * from "./MemoryLatticeManager";