UNPKG

@arizeai/phoenix-evals

Version:
59 lines 2.73 kB
import { TOOL_RESPONSE_HANDLING_CLASSIFICATION_EVALUATOR_CONFIG } from "../__generated__/default_templates/index.js"; import { createClassificationEvaluator } from "./createClassificationEvaluator.js"; /** * Creates a tool response handling evaluator function. * * This function returns an evaluator that determines whether an AI agent properly * handled a tool's response, including error handling, data extraction, * transformation, and safe information disclosure. * * @param args - The arguments for creating the tool response handling evaluator. * @param args.model - The model to use for classification. * @param args.choices - The possible classification choices (defaults to correct/incorrect). * @param args.promptTemplate - The prompt template to use. * @param args.telemetry - The telemetry to use for the evaluator. * * @returns An evaluator function that takes a {@link ToolResponseHandlingEvaluationRecord} * and returns a classification result indicating whether the tool response handling * is correct or incorrect. * * @example * ```ts * const evaluator = createToolResponseHandlingEvaluator({ model: openai("gpt-4o-mini") }); * * // Example: Correct extraction from tool result * const result = await evaluator.evaluate({ * input: "What's the weather in Seattle?", * toolCall: 'get_weather(location="Seattle")', * toolResult: JSON.stringify({ * temperature: 58, * unit: "fahrenheit", * conditions: "partly cloudy" * }), * output: "The weather in Seattle is 58°F and partly cloudy." * }); * console.log(result.label); // "correct" * * // Example: Hallucinated data (incorrect) * const resultHallucinated = await evaluator.evaluate({ * input: "What restaurants are nearby?", * toolCall: 'search_restaurants(location="downtown")', * toolResult: JSON.stringify({ * results: [{ name: "Cafe Luna", rating: 4.2 }] * }), * output: "I found Cafe Luna (4.2 stars) and Mario's Italian (4.8 stars) nearby." * }); * console.log(resultHallucinated.label); // "incorrect" - Mario's was hallucinated * ``` */ export function createToolResponseHandlingEvaluator(args) { const { choices = TOOL_RESPONSE_HANDLING_CLASSIFICATION_EVALUATOR_CONFIG.choices, promptTemplate = TOOL_RESPONSE_HANDLING_CLASSIFICATION_EVALUATOR_CONFIG.template, optimizationDirection = TOOL_RESPONSE_HANDLING_CLASSIFICATION_EVALUATOR_CONFIG.optimizationDirection, name = TOOL_RESPONSE_HANDLING_CLASSIFICATION_EVALUATOR_CONFIG.name, ...rest } = args; return createClassificationEvaluator({ ...rest, promptTemplate, choices, optimizationDirection, name, }); } //# sourceMappingURL=createToolResponseHandlingEvaluator.js.map