diff --git a/src/crewai/experimental/evaluation/base_evaluator.py b/src/crewai/experimental/evaluation/base_evaluator.py index 1fe3f58a1..07b522054 100644 --- a/src/crewai/experimental/evaluation/base_evaluator.py +++ b/src/crewai/experimental/evaluation/base_evaluator.py @@ -6,7 +6,7 @@ from typing import Any, Optional from pydantic import BaseModel, Field from crewai.agent import Agent -from crewai.llm import BaseLLM +from crewai.llms.base_llm import BaseLLM from crewai.task import Task from crewai.utilities.llm_utils import create_default_llm, create_llm @@ -19,7 +19,7 @@ class MetricCategory(enum.Enum): PARAMETER_EXTRACTION = "parameter_extraction" TOOL_INVOCATION = "tool_invocation" - def title(self): + def title(self) -> str: return self.value.replace("_", " ").title()