fix: update import to use crewai.llms.base_llm

This commit is contained in:
Greyson LaLonde
2025-09-05 08:58:43 -04:00
parent 909b2fd0ef
commit 5f776bbb0a

View File

@@ -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()