mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-05-06 01:32:36 +00:00
0aeced1fa9
- Added OpenAIConverterAdapter to manage structured output conversion for OpenAI agents, enhancing their ability to process and format results. - Updated OpenAIAgentAdapter to utilize the new converter for configuring structured output and post-processing results. - Removed the deprecated get_output_converter method from OpenAIAgentAdapter. - Added unit tests for BaseAgentAdapter and BaseToolAdapter to ensure proper functionality and integration of new features.
114 lines
3.6 KiB
Python
114 lines
3.6 KiB
Python
from typing import Any, Dict, List, Optional
|
|
|
|
import pytest
|
|
from pydantic import BaseModel
|
|
|
|
from crewai.agent import BaseAgent
|
|
from crewai.agents.agent_adapters.base_agent_adapter import BaseAgentAdapter
|
|
from crewai.tools import BaseTool
|
|
from crewai.utilities.token_counter_callback import TokenProcess
|
|
|
|
|
|
# Concrete implementation for testing
|
|
class ConcreteAgentAdapter(BaseAgentAdapter):
|
|
def configure_tools(
|
|
self, tools: Optional[List[BaseTool]] = None, **kwargs: Any
|
|
) -> None:
|
|
# Simple implementation for testing
|
|
self.tools = tools or []
|
|
|
|
def execute_task(
|
|
self,
|
|
task: Any,
|
|
context: Optional[str] = None,
|
|
tools: Optional[List[Any]] = None,
|
|
) -> str:
|
|
# Dummy implementation needed due to BaseAgent inheritance
|
|
return "Task executed"
|
|
|
|
def create_agent_executor(self, tools: Optional[List[BaseTool]] = None) -> Any:
|
|
# Dummy implementation
|
|
return None
|
|
|
|
def get_delegation_tools(
|
|
self, tools: List[BaseTool], tool_map: Optional[Dict[str, BaseTool]]
|
|
) -> List[BaseTool]:
|
|
# Dummy implementation
|
|
return []
|
|
|
|
def _parse_output(self, agent_output: Any, token_process: TokenProcess):
|
|
# Dummy implementation
|
|
pass
|
|
|
|
def get_output_converter(self, tools: Optional[List[BaseTool]] = None) -> Any:
|
|
# Dummy implementation
|
|
return None
|
|
|
|
|
|
def test_base_agent_adapter_initialization():
|
|
"""Test initialization of the concrete agent adapter."""
|
|
adapter = ConcreteAgentAdapter(
|
|
role="test role", goal="test goal", backstory="test backstory"
|
|
)
|
|
assert isinstance(adapter, BaseAgent)
|
|
assert isinstance(adapter, BaseAgentAdapter)
|
|
assert adapter.role == "test role"
|
|
assert adapter._agent_config is None
|
|
assert adapter.adapted_structured_output is False
|
|
|
|
|
|
def test_base_agent_adapter_initialization_with_config():
|
|
"""Test initialization with agent_config."""
|
|
config = {"model": "gpt-4"}
|
|
adapter = ConcreteAgentAdapter(
|
|
agent_config=config,
|
|
role="test role",
|
|
goal="test goal",
|
|
backstory="test backstory",
|
|
)
|
|
assert adapter._agent_config == config
|
|
|
|
|
|
def test_configure_tools_method_exists():
|
|
"""Test that configure_tools method exists and can be called."""
|
|
adapter = ConcreteAgentAdapter(
|
|
role="test role", goal="test goal", backstory="test backstory"
|
|
)
|
|
# Create dummy tools if needed, or pass None
|
|
tools = []
|
|
adapter.configure_tools(tools)
|
|
assert hasattr(adapter, "tools")
|
|
assert adapter.tools == tools
|
|
|
|
|
|
def test_configure_structured_output_method_exists():
|
|
"""Test that configure_structured_output method exists and can be called."""
|
|
adapter = ConcreteAgentAdapter(
|
|
role="test role", goal="test goal", backstory="test backstory"
|
|
)
|
|
|
|
# Define a dummy structure or pass None/Any
|
|
class DummyOutput(BaseModel):
|
|
data: str
|
|
|
|
structured_output = DummyOutput
|
|
adapter.configure_structured_output(structured_output)
|
|
# Add assertions here if configure_structured_output modifies state
|
|
# For now, just ensuring it runs without error is sufficient
|
|
pass
|
|
|
|
|
|
def test_base_agent_adapter_inherits_base_agent():
|
|
"""Test that BaseAgentAdapter inherits from BaseAgent."""
|
|
assert issubclass(BaseAgentAdapter, BaseAgent)
|
|
|
|
|
|
class ConcreteAgentAdapterWithoutRequiredMethods(BaseAgentAdapter):
|
|
pass
|
|
|
|
|
|
def test_base_agent_adapter_fails_without_required_methods():
|
|
"""Test that BaseAgentAdapter fails without required methods."""
|
|
with pytest.raises(TypeError):
|
|
ConcreteAgentAdapterWithoutRequiredMethods() # type: ignore
|