mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-13 18:18:29 +00:00
be09b519c4
- Add Union types to handle both BaseTool and dict in tool parameters - Update tool utility functions to handle mixed types safely - Filter raw tool definitions in agent adapters - Fix experimental evaluation metrics to handle mixed tool types - Maintain backward compatibility while adding hosted tools support Co-Authored-By: João <joao@crewai.com>
43 lines
1.4 KiB
Python
43 lines
1.4 KiB
Python
from abc import ABC, abstractmethod
|
|
from typing import Any, Dict, List, Optional, Union
|
|
|
|
from pydantic import PrivateAttr
|
|
|
|
from crewai.agent import BaseAgent
|
|
from crewai.tools import BaseTool
|
|
|
|
|
|
class BaseAgentAdapter(BaseAgent, ABC):
|
|
"""Base class for all agent adapters in CrewAI.
|
|
|
|
This abstract class defines the common interface and functionality that all
|
|
agent adapters must implement. It extends BaseAgent to maintain compatibility
|
|
with the CrewAI framework while adding adapter-specific requirements.
|
|
"""
|
|
|
|
adapted_structured_output: bool = False
|
|
_agent_config: Optional[Dict[str, Any]] = PrivateAttr(default=None)
|
|
|
|
model_config = {"arbitrary_types_allowed": True}
|
|
|
|
def __init__(self, agent_config: Optional[Dict[str, Any]] = None, **kwargs: Any):
|
|
super().__init__(adapted_agent=True, **kwargs)
|
|
self._agent_config = agent_config
|
|
|
|
@abstractmethod
|
|
def configure_tools(self, tools: Optional[List[Union[BaseTool, dict]]] = None) -> None:
|
|
"""Configure and adapt tools for the specific agent implementation.
|
|
|
|
Args:
|
|
tools: Optional list of BaseTool instances and raw tool definitions to be configured
|
|
"""
|
|
pass
|
|
|
|
def configure_structured_output(self, structured_output: Any) -> None:
|
|
"""Configure the structured output for the specific agent implementation.
|
|
|
|
Args:
|
|
structured_output: The structured output to be configured
|
|
"""
|
|
pass
|