|
6 | 6 | from langevals_core.base_evaluator import (
|
7 | 7 | BaseEvaluator,
|
8 | 8 | EvaluationResult,
|
9 |
| - LLMEvaluatorSettings, |
| 9 | + EvaluatorSettings, |
10 | 10 | Money,
|
11 | 11 | EvaluationResultSkipped,
|
12 |
| - EvaluatorEntry |
| 12 | + EvaluatorEntry, |
13 | 13 | )
|
14 |
| -from pydantic import BaseModel, Field |
| 14 | +from pydantic import Field |
15 | 15 | from ragas import evaluate
|
16 | 16 | from ragas.metrics.base import Metric
|
17 | 17 | from ragas.llms import LangchainLLMWrapper
|
|
38 | 38 | from tqdm.notebook import tqdm as tqdm_notebook
|
39 | 39 | from functools import partialmethod
|
40 | 40 |
|
41 |
| -import json |
42 |
| -import re |
43 | 41 | from typing import List, Optional
|
44 | 42 | from datasets import Dataset
|
45 | 43 | from ragas import evaluate
|
46 | 44 | from ragas.metrics import faithfulness, Faithfulness
|
47 | 45 | from ragas.llms import LangchainLLMWrapper
|
48 |
| -from ragas.llms.prompt import PromptValue |
49 |
| -from langchain_core.callbacks import Callbacks |
50 |
| -from pydantic import BaseModel, Field |
51 |
| -import litellm |
52 |
| -from langchain.schema.output import LLMResult |
53 |
| -from langchain_core.outputs.generation import Generation |
| 46 | +from pydantic import Field |
54 | 47 | from langevals_core.utils import calculate_total_tokens
|
55 | 48 |
|
56 | 49 | env_vars = []
|
57 | 50 |
|
58 | 51 |
|
59 |
| -class RagasSettings(LLMEvaluatorSettings): |
60 |
| - model: str = Field( |
61 |
| - default="azure/gpt-35-turbo-16k", |
| 52 | +class RagasSettings(EvaluatorSettings): |
| 53 | + model: Literal[ |
| 54 | + "openai/gpt-3.5-turbo-16k", |
| 55 | + "openai/gpt-4o", |
| 56 | + "openai/gpt-4o-mini", |
| 57 | + "azure/gpt-35-turbo-16k", |
| 58 | + "azure/gpt-4o", |
| 59 | + "anthropic/claude-3-5-sonnet-20240620", |
| 60 | + ] = Field( |
| 61 | + default="openai/gpt-3.5-turbo-16k", |
62 | 62 | description="The model to use for evaluation.",
|
63 | 63 | )
|
64 | 64 | embeddings_model: Literal[
|
65 | 65 | "openai/text-embedding-ada-002",
|
66 | 66 | "azure/text-embedding-ada-002",
|
67 | 67 | ] = Field(
|
68 |
| - default="azure/text-embedding-ada-002", |
| 68 | + default="openai/text-embedding-ada-002", |
69 | 69 | description="The model to use for embeddings.",
|
70 | 70 | )
|
71 | 71 | max_tokens: int = Field(
|
|
0 commit comments