def __init__()

in genai-on-vertex-ai/gemini/needle_in_a_haystack/needlehaystack/evaluators/google.py [0:0]

• 7 lines of code
• 1 McCabe index (conditional complexity)

    def __init__(self,
                 project_id: str,
                 model_name: str = "gemini-2.0-flash-001",
                 model_kwargs: dict = DEFAULT_MODEL_KWARGS):
        """
        :param project_id: ID of the google cloud platform project to use
        :param model_name: The name of the model.
        :param model_kwargs: Model configuration. Default is {temperature: 0}
        """
        self.model_name = model_name
        self.model_kwargs = model_kwargs
        self.evaluator = ChatVertexAI(model=self.model_name, **self.model_kwargs)