2121from deepeval .models import LocalModel
2222from deepeval .test_case import LLMTestCase
2323from deepeval .test_case .llm_test_case import ToolCall
24+ from pydantic import Field
2425from typing_extensions import override
2526
2627from veadk .config import getenv
28+ from veadk .evaluation .types import EvalResultCaseData , EvalResultMetadata
2729from veadk .utils .logger import get_logger
2830
2931from ..base_evaluator import BaseEvaluator , EvalResultData , MetricResult
30- from ..utils .prometheus import (
31- EvalResultCaseData ,
32- EvalResultMetadata ,
33- PrometheusPushgatewayConfig ,
34- push_to_prometheus ,
35- )
32+ from ..utils .prometheus import PrometheusPushgatewayConfig , push_to_prometheus
3633
3734logger = get_logger (__name__ )
3835
@@ -46,14 +43,23 @@ class DeepevalEvaluator(BaseEvaluator):
4643 def __init__ (
4744 self ,
4845 agent ,
49- judge_model_api_key : str = getenv ("MODEL_JUDGE_API_KEY" ),
50- judge_model_name : str = getenv (
51- "MODEL_JUDGE_NAME" ,
52- "doubao-seed-1-6-250615" ,
46+ judge_model_api_key : str = Field (
47+ ...,
48+ default_factory = lambda : getenv ("MODEL_JUDGE_API_KEY" ),
5349 ),
54- judge_model_api_base : str = getenv (
55- "MODEL_JUDGE_API_BASE" ,
56- "https://ark.cn-beijing.volces.com/api/v3/" ,
50+ judge_model_name : str = Field (
51+ ...,
52+ default_factory = lambda : getenv (
53+ "MODEL_JUDGE_NAME" ,
54+ "doubao-seed-1-6-250615" ,
55+ ),
56+ ),
57+ judge_model_api_base : str = Field (
58+ ...,
59+ default_factory = lambda : getenv (
60+ "MODEL_JUDGE_API_BASE" ,
61+ "https://ark.cn-beijing.volces.com/api/v3/" ,
62+ ),
5763 ),
5864 name : str = "veadk_deepeval_evaluator" ,
5965 prometheus_config : PrometheusPushgatewayConfig = None ,
0 commit comments