UNPKG

@arizeai/phoenix-evals

Version:
39 lines 2.05 kB
import { DOCUMENT_RELEVANCE_CLASSIFICATION_EVALUATOR_CONFIG } from "../__generated__/default_templates/index.js"; import { createClassificationEvaluator } from "./createClassificationEvaluator.js"; /** * Creates a document relevance evaluator function. * * This function returns an evaluator that determines whether a given document text * is relevant to a provided input question. The evaluator uses a classification model * and a prompt template to make its determination. * * @param args - The arguments for creating the document relevance evaluator. * @param args.model - The model to use for classification. * @param args.choices - The possible classification choices (defaults to DOCUMENT_RELEVANCE_CHOICES). * @param args.promptTemplate - The prompt template to use (defaults to DOCUMENT_RELEVANCE_TEMPLATE). * @param args.telemetry - The telemetry to use for the evaluator. * * @returns An evaluator function that takes a {@link DocumentRelevanceExample} and returns a classification result * indicating whether the document is relevant to the input question. * * @example * ```ts * const evaluator = createDocumentRelevanceEvaluator({ model: openai("gpt-4o-mini") }); * const result = await evaluator.evaluate({ * input: "What is the capital of France?", * documentText: "Paris is the capital and most populous city of France.", * }); * console.log(result.label); // "relevant" or "unrelated" * ``` */ export function createDocumentRelevanceEvaluator(args) { const { choices = DOCUMENT_RELEVANCE_CLASSIFICATION_EVALUATOR_CONFIG.choices, promptTemplate = DOCUMENT_RELEVANCE_CLASSIFICATION_EVALUATOR_CONFIG.template, optimizationDirection = DOCUMENT_RELEVANCE_CLASSIFICATION_EVALUATOR_CONFIG.optimizationDirection, name = DOCUMENT_RELEVANCE_CLASSIFICATION_EVALUATOR_CONFIG.name, ...rest } = args; return createClassificationEvaluator({ ...rest, promptTemplate, choices, optimizationDirection, name, }); } //# sourceMappingURL=createDocumentRelevanceEvaluator.js.map