文档
.act()
调用
LLM 本质上是文本输入、文本输出的程序。因此,你可能会问“LLM 如何使用工具?”。答案是一些 LLM 经过训练,会要求人类为它们调用工具,并期望工具的输出以某种格式返回。
想象一下,你正在通过电话为某人提供计算机支持。你可能会说这样的话:“为我运行这个命令... 好的,它输出了什么?... 好的,现在点击那里,告诉我上面写着什么...”。在这种情况下,你就是 LLM!而你正在通过电话线另一端的人“间接地调用工具”。
rounds
”中运行工具调用我们引入“执行轮次”的概念来描述运行工具、向 LLM 提供其输出,然后等待 LLM 决定下一步操作的组合过程。
执行轮次
• run a tool → ↑ • provide the result to the LLM → │ • wait for the LLM to generate a response │ └────────────────────────────────────────┘ └➔ (return)
模型可能会选择多次运行工具,然后才返回最终结果。例如,如果 LLM 正在编写代码,它可能会选择编译或运行程序、修复错误,然后再运行,如此反复,直到获得期望的结果。
考虑到这一点,我们说 .act()
API 是一个自动“多轮”工具调用 API。
import { LMStudioClient, tool } from "@lmstudio/sdk";
import { z } from "zod";
const client = new LMStudioClient();
const multiplyTool = tool({
name: "multiply",
description: "Given two numbers a and b. Returns the product of them.",
parameters: { a: z.number(), b: z.number() },
implementation: ({ a, b }) => a * b,
});
const model = await client.llm.model("qwen2.5-7b-instruct");
await model.act("What is the result of 12345 multiplied by 54321?", [multiplyTool], {
onMessage: (message) => console.info(message.toString()),
});
为工具使用选择的模型将极大地影响性能。
选择模型时的一些通用指南
以下代码演示了如何在单个 .act()
调用中提供多个工具。
import { LMStudioClient, tool } from "@lmstudio/sdk";
import { z } from "zod";
const client = new LMStudioClient();
const additionTool = tool({
name: "add",
description: "Given two numbers a and b. Returns the sum of them.",
parameters: { a: z.number(), b: z.number() },
implementation: ({ a, b }) => a + b,
});
const isPrimeTool = tool({
name: "isPrime",
description: "Given a number n. Returns true if n is a prime number.",
parameters: { n: z.number() },
implementation: ({ n }) => {
if (n < 2) return false;
const sqrt = Math.sqrt(n);
for (let i = 2; i <= sqrt; i++) {
if (n % i === 0) return false;
}
return true;
},
});
const model = await client.llm.model("qwen2.5-7b-instruct");
await model.act(
"Is the result of 12345 + 45668 a prime? Think step by step.",
[additionTool, isPrimeTool],
{ onMessage: (message) => console.info(message.toString()) },
);
以下代码创建了一个与 LLM Agent 的对话循环,该 Agent 可以创建文件。
import { Chat, LMStudioClient, tool } from "@lmstudio/sdk";
import { existsSync } from "fs";
import { writeFile } from "fs/promises";
import { createInterface } from "readline/promises";
import { z } from "zod";
const rl = createInterface({ input: process.stdin, output: process.stdout });
const client = new LMStudioClient();
const model = await client.llm.model();
const chat = Chat.empty();
const createFileTool = tool({
name: "createFile",
description: "Create a file with the given name and content.",
parameters: { name: z.string(), content: z.string() },
implementation: async ({ name, content }) => {
if (existsSync(name)) {
return "Error: File already exists.";
}
await writeFile(name, content, "utf-8");
return "File created.";
},
});
while (true) {
const input = await rl.question("You: ");
// Append the user input to the chat
chat.append("user", input);
process.stdout.write("Bot: ");
await model.act(chat, [createFileTool], {
// When the model finish the entire message, push it to the chat
onMessage: (message) => chat.append(message),
onPredictionFragment: ({ content }) => {
process.stdout.write(content);
},
});
process.stdout.write("\n");
}