评估是一种测试形式,可帮助您验证 LLM 的回答并确保其符合您的质量标准。
Firebase Genkit 通过插件支持第三方评估工具,并搭配强大的可观测性功能,可深入了解采用 LLM 的应用的运行时状态。Genkit 工具可帮助您自动从中间步骤中提取数据(包括输入、输出和信息),以评估 LLM 回答的端到端质量,以及了解系统构建块的性能。
例如,如果您有一个 RAG 流程,Genkit 将提取检索器返回的一组文档,以便您在流程上下文中运行检索器时评估其质量,如以下 Genkit 忠实度和回答相关性指标所示:
import { genkit } from 'genkit';
import { genkitEval, GenkitMetric } from '@genkit-ai/evaluator';
import { vertexAI, textEmbedding004, gemini15Flash } from '@genkit-ai/vertexai';
const ai = genkit({
plugins: [
vertexAI(),
genkitEval({
judge: gemini15Flash,
metrics: [GenkitMetric.FAITHFULNESS, GenkitMetric.ANSWER_RELEVANCY],
embedder: textEmbedding004, // GenkitMetric.ANSWER_RELEVANCY requires an embedder
}),
],
// ...
});
注意:上述配置需要安装 genkit
、@genkit-ai/google-ai
、@genkit-ai/evaluator
和 @genkit-ai/vertexai
软件包。
npm install @genkit-ai/evaluator @genkit-ai/vertexai
首先,定义一组要用作输入数据集的输入(称为 testInputs.json
)。此输入数据集代表您将用于生成输出以进行评估的测试用例。
["Cheese", "Broccoli", "Spinach and Kale"]
然后,您可以使用 eval:flow
命令根据 testInputs.json
中提供的测试用例评估您的流程。
genkit eval:flow menuSuggestionFlow --input testInputs.json
然后,您可以通过运行以下命令在开发者界面中查看评估结果:
genkit start
然后,前往 localhost:4000/evaluate
。
或者,您也可以提供输出文件,以便在 JSON 文件中检查输出。
genkit eval:flow menuSuggestionFlow --input testInputs.json --output eval-result.json
注意:下面的示例展示了 LLM 如何帮助您生成测试用例。
支持的评估器
Genkit 评估程序
Genkit 包含少数几个受 RAGAS 启发的原生评估器,可帮助您入门:
- 忠诚
- 回答相关性
- 恶意
评估器插件
Genkit 通过 VertexAI 插件支持 VertexAI Rapid 评估器等插件提供的其他评估器。
高级用法
eval:flow
是一种快速评估流程的便捷方式,但有时您可能需要更好地控制评估步骤。如果您使用的是其他框架,并且已经有了要评估的输出,就可能会出现这种情况。您可以半自动执行 eval:flow
执行的所有步骤。
您可以批量运行 Genkit 流程,并为运行添加唯一标签,该标签将用于提取评估数据集(一组输入、输出和上下文)。
针对测试输入运行流程:
genkit flow:batchRun myRagFlow test_inputs.json --output flow_outputs.json --label customLabel
提取评估数据:
genkit eval:extractData myRagFlow --label customLabel --output customLabel_dataset.json
导出的数据将以 JSON 文件的形式输出,其中每个 testCase 均采用以下格式:
[
{
"testCaseId": string,
"input": string,
"output": string,
"context": array of strings,
"traceIds": array of strings,
}
]
数据提取器会自动定位检索器,并将生成的文档添加到上下文数组。默认情况下,eval:run
将针对所有已配置的评估器运行,并且与 eval:flow
一样,eval:run
的结果将显示在位于 localhost:4000/evaluate
的开发者界面的评估页面中。
自定义提取器
您还可以提供要在 eval:extractData
和 eval:flow
命令中使用的自定义提取器。借助自定义提取器,您可以替换默认的提取逻辑,从而更轻松地创建数据集并对其进行评估。
如需配置自定义提取器,请在项目根目录中添加一个名为 genkit-tools.conf.js
的工具配置文件(如果尚无)。
cd $GENKIT_PROJECT_HOME
touch genkit-tools.conf.js
在 tools 配置文件中,添加以下代码:
module.exports = {
evaluators: [
{
actionRef: '/flow/myFlow',
extractors: {
context: { outputOf: 'foo-step' },
output: 'bar-step',
},
},
],
};
在此示例中,您将为 myFlow
流配置提取器。该配置会替换 context
和 output
字段的提取器,并对 input
字段使用默认逻辑。
评估提取器的规范如下:
evaluators
字段接受一个 EvaluatorConfig 对象数组,这些对象受flowName
的作用域控制extractors
是一个对象,用于指定提取器替换项。extractors
中目前支持的键为[input, output, context]
。可接受的值类型包括:string
- 这应为步骤名称,以字符串形式指定。系统会为此键提取此步骤的输出。{ inputOf: string }
或{ outputOf: string }
- 这些对象代表步骤的特定渠道(输入或输出)。例如,{ inputOf: 'foo-step' }
会提取此键对应的步骤foo-step
的输入。(trace) => string;
- 为了进一步提高灵活性,您可以提供一个接受 Genkit 轨迹并返回string
的函数,并在此函数内指定提取逻辑。如需了解确切的 TraceData 架构,请参阅genkit/genkit-tools/common/src/types/trace.ts
。
注意:所有这些步骤的提取数据将是 JSON 字符串。工具会在评估时自动解析此 JSON 字符串。如果提供函数提取器,请确保输出是有效的 JSON 字符串。例如:"Hello, world!"
不是有效的 JSON;"\"Hello, world!\""
是有效的。
在现有数据集上运行
如需对已提取的数据集运行评估,请执行以下操作:
genkit eval:run customLabel_dataset.json
如需输出到其他位置,请使用 --output
标志。
genkit eval:flow menuSuggestionFlow --input testInputs.json --output customLabel_evalresult.json
如需在部分已配置的评估器上运行,请使用 --evaluators
标志,并按名称提供以英文逗号分隔的评估器列表:
genkit eval:run customLabel_dataset.json --evaluators=genkit/faithfulness,genkit/answer_relevancy
使用 LLM 合成测试数据
下面的示例流程使用 PDF 文件生成用户可能会询问的相关问题。
import { genkit, run, z } from "genkit";
import { googleAI, gemini15Flash } from "@genkit-ai/googleai";
import { chunk } from "llm-chunk";
const ai = genkit({ plugins: [googleAI()] });
export const synthesizeQuestions = ai.defineFlow(
{
name: "synthesizeQuestions",
inputSchema: z.string().describe("PDF file path"),
outputSchema: z.array(z.string()),
},
async (filePath) => {
filePath = path.resolve(filePath);
const pdfTxt = await run("extract-text", () => extractText(filePath));
const chunks = await run("chunk-it", async () =>
chunk(pdfTxt, chunkingConfig)
);
const questions: string[] = [];
for (var i = 0; i < chunks.length; i++) {
const qResponse = await ai.generate({
model: gemini15Flash,
prompt: {
text: `Generate one question about the text below: ${chunks[i]}`,
},
});
questions.push(qResponse.text);
}
return questions;
}
);
然后,您可以使用此命令将数据导出到文件中,以供评估。
genkit flow:run synthesizeQuestions '"my_input.pdf"' --output synthesizedQuestions.json