文档
使用 llm.complete(...) 从已加载的语言模型生成文本补全。文本补全是指向模型发送一个无格式字符串,并期望模型完成剩余文本。
这与多轮对话聊天不同。有关聊天补全的更多信息,请参阅 聊天补全。
首先,你需要加载一个模型来生成补全。这可以通过在 llm 句柄上使用 model 方法来完成。
import { LMStudioClient } from "@lmstudio/sdk";
const client = new LMStudioClient();
const model = await client.llm.model("qwen2.5-7b-instruct");
一旦加载了模型,你可以通过将字符串传递给 llm 句柄上的 complete 方法来生成补全。
const completion = model.complete("My name is", {
maxTokens: 100,
});
for await (const { content } of completion) {
process.stdout.write(content);
}
console.info(); // Write a new line for cosmetic purposes
您还可以打印预测元数据,例如用于生成的模型、生成的标记数量、第一个标记的时间以及停止原因。
console.info("Model used:", completion.modelInfo.displayName);
console.info("Predicted tokens:", completion.stats.predictedTokensCount);
console.info("Time to first token (seconds):", completion.stats.timeToFirstTokenSec);
console.info("Stop reason:", completion.stats.stopReason);
以下是如何使用 complete 方法来模拟终端的示例。
import { LMStudioClient } from "@lmstudio/sdk";
import { createInterface } from "node:readline/promises";
const rl = createInterface({ input: process.stdin, output: process.stdout });
const client = new LMStudioClient();
const model = await client.llm.model();
let history = "";
while (true) {
const command = await rl.question("$ ");
history += "$ " + command + "\n";
const prediction = model.complete(history, { stopStrings: ["$"] });
for await (const { content } of prediction) {
process.stdout.write(content);
}
process.stdout.write("\n");
const { content } = await prediction.result();
history += content;
}
此页面的源代码可在 GitHub 上找到