@axiom-lattice/core 1.0.13 → 1.0.21

package/src/util/getLastHumanMessageData.ts

@@ -1,41 +0,0 @@
-import {
-  BaseMessage,
-  HumanMessage,
-  isHumanMessage,
-} from "@langchain/core/messages";
-export type UserFile = {
-  id: number;
-  name: string;
-  type?: string;
-  path?: string;
-};
-export type UserFileZip = {
-  id: number;
-  name: string;
-  files: UserFile[];
-  type?: "application/zip" | "application/rar";
-  totalFiles?: number;
-  processedFiles?: number;
-  skippedFiles?: number;
-};
-
-export const isCompressedFile = (fileType: string): boolean => {
-  return fileType === "application/zip" || fileType === "application/rar";
-};
-
-export const getLastHumanMessageData = (messages: BaseMessage[]) => {
-  // 从最后一条消息开始向前查找第一个人类消息
-  for (let i = messages.length - 1; i >= 0; i--) {
-    const message = messages[i];
-    if (isHumanMessage(message)) {
-      const files = message?.additional_kwargs?.files as UserFile[];
-      return {
-        files,
-        content: message?.content,
-      };
-    }
-  }
-
-  // 如果没有找到任何人类消息，返回 null
-  return null;
-};

package/src/util/returnAIResponse.ts

@@ -1,30 +0,0 @@
-import { AIMessage } from "@langchain/core/messages";
-import { LangGraphRunnableConfig } from "@langchain/langgraph";
-import { v4 } from "uuid";
-import { genUIMarkdown } from "./genUIMarkdown";
-
-export const returnAIResponse = (
-  config: LangGraphRunnableConfig,
-  content: string,
-  type?: string,
-  data?: Record<string, any>
-) => {
-  const contents = type
-    ? [content, genUIMarkdown(type, data)].join("\n")
-    : content;
-  const message = {
-    messages: [
-      new AIMessage({
-        content: contents,
-      }),
-      // {
-      //   id: v4(),
-      //   role: "ai",
-      //   content: contents,
-      //   type: "message",
-      // }, // 旧的message
-    ],
-  };
-  // config.writer?.(message);
-  return message;
-};

package/src/util/returnErrorResponse.ts

@@ -1,26 +0,0 @@
-import { AIMessage } from "@langchain/core/messages";
-import { LangGraphRunnableConfig } from "@langchain/langgraph";
-import { v4 } from "uuid";
-
-export const returnErrorResponse = (
-  config: LangGraphRunnableConfig,
-  content: string,
-  type?: string,
-  data?: Record<string, any>
-) => {
-  const contents = type
-    ? [content, "```" + type, JSON.stringify(data), "```"].join("\n")
-    : content;
-  const message = {
-    messages: [
-      new AIMessage({
-        content: contents,
-        additional_kwargs: {
-          type: "error",
-        },
-      }),
-    ],
-  };
-  //config.writer?.(message);
-  return message;
-};

package/src/util/returnFeedbackResponse.ts

@@ -1,25 +0,0 @@
-import { interrupt, LangGraphRunnableConfig } from "@langchain/langgraph";
-import { UserFeedbackResponse } from "../types";
-
-export const returnFeedbackResponse = async (
-  content: string,
-  type: string,
-  data: Record<string, any>
-) => {
-  const contents = [
-    content,
-    "```" + type,
-    JSON.stringify({ ...data }),
-    "```",
-  ].join("\n");
-
-  const feedback: UserFeedbackResponse = await interrupt(contents);
-  //  console.log("returnFeedbackResponse_feedback", feedback);
-  const messages = [
-    {
-      role: "user",
-      content: JSON.stringify(feedback),
-    },
-  ];
-  return { feedback, messages };
-};

package/src/util/returnToolResponse.ts

@@ -1,32 +0,0 @@
-import { ToolMessage } from "@langchain/core/messages";
-import { LangGraphRunnableConfig } from "@langchain/langgraph";
-import { v4 } from "uuid";
-import { genUIMarkdown } from "./genUIMarkdown";
-
-export const returnToolResponse = (
-  config: LangGraphRunnableConfig,
-  think: {
-    think_id?: string; // 思考的id,为了解决思考过程分组的问题
-    title?: string;
-    content: string;
-    status?: "success" | "error";
-    type?: string;
-    data?: Record<string, any>;
-  }
-) => {
-  const { think_id = v4(), content, title, status, type, data } = think;
-  const contents = type
-    ? [title, content, genUIMarkdown(type, data)]
-    : [title, content];
-  const message = {
-    messages: [
-      new ToolMessage({
-        content: contents.filter((item) => item).join("\n\n"),
-        tool_call_id: config.configurable?.run_id + "_tool_" + think_id,
-        status: status || "success",
-      }),
-    ],
-  };
-
-  return message;
-};

package/src/util/withAgentName.ts

@@ -1,220 +0,0 @@
-import { LanguageModelLike } from "@langchain/core/language_models/base";
-import {
-  AIMessage,
-  BaseMessage,
-  BaseMessageLike,
-  isAIMessage,
-  isAIMessageChunk,
-  isBaseMessage,
-  isBaseMessageChunk,
-  MessageContent,
-} from "@langchain/core/messages";
-import { RunnableLambda, RunnableSequence } from "@langchain/core/runnables";
-
-const NAME_PATTERN = /<name>(.*?)<\/name>/s;
-const CONTENT_PATTERN = /<content>(.*?)<\/content>/s;
-
-export type AgentNameMode = "inline";
-
-/**
- * Attach formatted agent names to the messages passed to and from a language model.
- *
- * This is useful for making a message history with multiple agents more coherent.
- *
- * NOTE: agent name is consumed from the message.name field.
- * If you're using an agent built with createReactAgent, name is automatically set.
- * If you're building a custom agent, make sure to set the name on the AI message returned by the LLM.
- *
- * @param message - Message to add agent name formatting to
- * @returns Message with agent name formatting
- *
- * @internal
- */
-export function _addInlineAgentName<T extends BaseMessageLike>(
-  message: T
-): T | AIMessage {
-  const isAI =
-    isBaseMessage(message) &&
-    (isAIMessage(message) ||
-      (isBaseMessageChunk(message) && isAIMessageChunk(message)));
-
-  if (!isAI || !message.name) {
-    return message;
-  }
-
-  const { name } = message;
-
-  if (typeof message.content === "string") {
-    return new AIMessage({
-      ...(Object.keys(message.lc_kwargs ?? {}).length > 0
-        ? message.lc_kwargs
-        : message),
-      content: `<name>${name}</name><content>${message.content}</content>`,
-      name: undefined,
-    });
-  }
-
-  const updatedContent = [];
-  let textBlockCount = 0;
-
-  for (const contentBlock of message.content) {
-    if (typeof contentBlock === "string") {
-      textBlockCount += 1;
-      updatedContent.push(
-        `<name>${name}</name><content>${contentBlock}</content>`
-      );
-    } else if (
-      typeof contentBlock === "object" &&
-      "type" in contentBlock &&
-      contentBlock.type === "text"
-    ) {
-      textBlockCount += 1;
-      updatedContent.push({
-        ...contentBlock,
-        text: `<name>${name}</name><content>${contentBlock.text}</content>`,
-      });
-    } else {
-      updatedContent.push(contentBlock);
-    }
-  }
-
-  if (!textBlockCount) {
-    updatedContent.unshift({
-      type: "text",
-      text: `<name>${name}</name><content></content>`,
-    });
-  }
-  return new AIMessage({
-    ...message.lc_kwargs,
-    content: updatedContent as MessageContent,
-    name: undefined,
-  });
-}
-
-/**
- * Remove explicit name and content XML tags from the AI message content.
- *
- * Examples:
- *
- * @example
- * ```typescript
- * removeInlineAgentName(new AIMessage({ content: "<name>assistant</name><content>Hello</content>", name: "assistant" }))
- * // AIMessage with content: "Hello"
- *
- * removeInlineAgentName(new AIMessage({ content: [{type: "text", text: "<name>assistant</name><content>Hello</content>"}], name: "assistant" }))
- * // AIMessage with content: [{type: "text", text: "Hello"}]
- * ```
- *
- * @internal
- */
-export function _removeInlineAgentName<T extends BaseMessage>(message: T): T {
-  if (!isAIMessage(message) || !message.content) {
-    return message;
-  }
-
-  let updatedContent: MessageContent = [];
-  let updatedName: string | undefined;
-
-  if (Array.isArray(message.content)) {
-    updatedContent = message.content
-      .filter((block) => {
-        if (block.type === "text") {
-          const nameMatch = block.text.match(NAME_PATTERN);
-          const contentMatch = block.text.match(CONTENT_PATTERN);
-          // don't include empty content blocks that were added because there was no text block to modify
-          if (nameMatch && (!contentMatch || contentMatch[1] === "")) {
-            // capture name from text block
-            // eslint-disable-next-line prefer-destructuring
-            updatedName = nameMatch[1];
-            return false;
-          }
-          return true;
-        }
-        return true;
-      })
-      .map((block) => {
-        if (block.type === "text") {
-          const nameMatch = block.text.match(NAME_PATTERN);
-          const contentMatch = block.text.match(CONTENT_PATTERN);
-
-          if (!nameMatch || !contentMatch) {
-            return block;
-          }
-
-          // capture name from text block
-          // eslint-disable-next-line prefer-destructuring
-          updatedName = nameMatch[1];
-
-          return {
-            ...block,
-            text: contentMatch[1],
-          };
-        }
-        return block;
-      });
-  } else {
-    const content = message.content as string;
-    const nameMatch = content.match(NAME_PATTERN);
-    const contentMatch = content.match(CONTENT_PATTERN);
-
-    if (!nameMatch || !contentMatch) {
-      return message;
-    }
-
-    // eslint-disable-next-line prefer-destructuring
-    updatedName = nameMatch[1];
-    // eslint-disable-next-line prefer-destructuring
-    updatedContent = contentMatch[1];
-  }
-
-  return new AIMessage({
-    ...(Object.keys(message.lc_kwargs ?? {}).length > 0
-      ? message.lc_kwargs
-      : message),
-    content: updatedContent,
-    name: updatedName,
-  }) as T;
-}
-
-/**
- * Attach formatted agent names to the messages passed to and from a language model.
- *
- * This is useful for making a message history with multiple agents more coherent.
- *
- * NOTE: agent name is consumed from the message.name field.
- * If you're using an agent built with createReactAgent, name is automatically set.
- * If you're building a custom agent, make sure to set the name on the AI message returned by the LLM.
- *
- * @param model - Language model to add agent name formatting to
- * @param agentNameMode - How to expose the agent name to the LLM
- *   - "inline": Add the agent name directly into the content field of the AI message using XML-style tags.
- *     Example: "How can I help you" -> "<name>agent_name</name><content>How can I help you?</content>".
- */
-export function withAgentName(
-  model: LanguageModelLike,
-  agentNameMode: AgentNameMode
-): LanguageModelLike {
-  let processInputMessage: (message: BaseMessageLike) => BaseMessageLike;
-  let processOutputMessage: (message: BaseMessage) => BaseMessage;
-
-  if (agentNameMode === "inline") {
-    processInputMessage = _addInlineAgentName;
-    processOutputMessage = _removeInlineAgentName;
-  } else {
-    throw new Error(
-      `Invalid agent name mode: ${agentNameMode}. Needs to be one of: "inline"`
-    );
-  }
-
-  function processInputMessages(
-    messages: BaseMessageLike[]
-  ): BaseMessageLike[] {
-    return messages.map(processInputMessage);
-  }
-
-  return RunnableSequence.from([
-    RunnableLambda.from(processInputMessages),
-    model,
-    RunnableLambda.from(processOutputMessage),
-  ]);
-}

package/src/util/zod-to-prompt.ts

@@ -1,50 +0,0 @@
-// filename: zod-to-prompt.ts
-import { z, ZodObject, ZodTypeAny } from "zod";
-
-export function zodToPrompt(schema: ZodObject<any>): string {
-  const shape = schema.shape;
-  const lines: string[] = [];
-
-  for (const key in shape) {
-    const field: ZodTypeAny = shape[key];
-    const description = field.description || "";
-    const isOptional = field.isOptional();
-    let type = field._def.typeName;
-
-    // 处理常见类型
-    switch (type) {
-      case "ZodString":
-        type = "字符串";
-        break;
-      case "ZodNumber":
-        type = "数字";
-        break;
-      case "ZodBoolean":
-        type = "布尔值";
-        break;
-      case "ZodArray":
-        type = "数组";
-        break;
-      case "ZodObject":
-        type = "对象";
-        break;
-      default:
-        type = "未知类型";
-    }
-
-    lines.push(
-      `- ${key}（${type}${isOptional ? "，可选" : "，必填"}）：${description}`
-    );
-  }
-
-  return `请提供以下参数：\n${lines.join("\n")}`;
-}
-
-// // 示例
-// const userSchema = z.object({
-//   username: z.string().min(3).describe("用户名，至少3个字符"),
-//   age: z.number().int().min(0).describe("年龄，必须为非负整数"),
-//   email: z.string().email().optional().describe("电子邮箱，可选"),
-// });
-
-// console.log(zodToPrompt(userSchema));

package/tsconfig.json

@@ -1,26 +0,0 @@
-{
-  "compilerOptions": {
-    "target": "ES2020",
-    "module": "preserve",
-    "lib": ["ES2020"],
-    "outDir": "./dist",
-    "rootDir": "./src",
-    "strict": true,
-    "moduleResolution": "Bundler",
-    "esModuleInterop": true,
-    "skipLibCheck": true,
-    "forceConsistentCasingInFileNames": true,
-    "resolveJsonModule": true,
-    "declaration": true,
-    "declarationMap": true,
-    "types": ["node", "jest"],
-    "paths": {
-      "@*": ["./src/*"]
-    },
-    "sourceMap": true,
-    "incremental": true, // 确保启用增量编译
-    "tsBuildInfoFile": "./.tsbuildinfo" // 指定构建信息文件位置
-  },
-  "include": ["src/index.ts"],
-  "exclude": ["node_modules", "dist"]
-}