Enhance custom LLM implementation with better error handling, documentation, and test coverage

Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
Devin AI
2025-03-04 17:50:52 +00:00
parent 22aeeaadbe
commit 963ed23b63
4 changed files with 686 additions and 94 deletions

View File

@@ -4,7 +4,7 @@ from crewai.agent import Agent
from crewai.crew import Crew
from crewai.flow.flow import Flow
from crewai.knowledge.knowledge import Knowledge
from crewai.llm import BaseLLM, LLM
from crewai.llm import LLM, BaseLLM
from crewai.process import Process
from crewai.task import Task

View File

@@ -6,10 +6,9 @@ import warnings
from concurrent.futures import Future
from copy import copy as shallow_copy
from hashlib import md5
from typing import Any, Callable, Dict, List, Optional, Set, Tuple, Union, cast, TypeVar
from typing import Any, Callable, Dict, List, Optional, Set, Tuple, TypeVar, Union, cast
from langchain_core.tools import BaseTool as LangchainBaseTool
from crewai.tools.base_tool import BaseTool, Tool
from pydantic import (
UUID4,
BaseModel,
@@ -38,7 +37,7 @@ from crewai.task import Task
from crewai.tasks.conditional_task import ConditionalTask
from crewai.tasks.task_output import TaskOutput
from crewai.tools.agent_tools.agent_tools import AgentTools
from crewai.tools.base_tool import Tool
from crewai.tools.base_tool import BaseTool, Tool
from crewai.types.usage_metrics import UsageMetrics
from crewai.utilities import I18N, FileHandler, Logger, RPMController
from crewai.utilities.constants import TRAINING_DATA_FILE