Fix issues

This commit is contained in:
Brandon Hancock
2025-02-14 15:11:58 -05:00
parent 185556b7e3
commit 2b438baad4
4 changed files with 31 additions and 15 deletions

View File

@@ -9,7 +9,6 @@ from unittest.mock import MagicMock, patch
import instructor
import pydantic_core
import pytest
from crewai.agent import Agent
from crewai.agents.cache import CacheHandler
from crewai.crew import Crew
@@ -541,9 +540,8 @@ def test_crew_with_delegating_agents():
def test_crew_with_delegating_agents_should_not_override_task_tools():
from typing import Type
from pydantic import BaseModel, Field
from crewai.tools import BaseTool
from pydantic import BaseModel, Field
class TestToolInput(BaseModel):
"""Input schema for TestTool."""
@@ -603,9 +601,8 @@ def test_crew_with_delegating_agents_should_not_override_task_tools():
def test_crew_with_delegating_agents_should_not_override_agent_tools():
from typing import Type
from pydantic import BaseModel, Field
from crewai.tools import BaseTool
from pydantic import BaseModel, Field
class TestToolInput(BaseModel):
"""Input schema for TestTool."""
@@ -667,9 +664,8 @@ def test_crew_with_delegating_agents_should_not_override_agent_tools():
def test_task_tools_override_agent_tools():
from typing import Type
from pydantic import BaseModel, Field
from crewai.tools import BaseTool
from pydantic import BaseModel, Field
class TestToolInput(BaseModel):
"""Input schema for TestTool."""
@@ -725,9 +721,8 @@ def test_task_tools_override_agent_tools_with_allow_delegation():
"""
from typing import Type
from pydantic import BaseModel, Field
from crewai.tools import BaseTool
from pydantic import BaseModel, Field
class TestToolInput(BaseModel):
query: str = Field(..., description="Query to process")
@@ -3429,11 +3424,10 @@ def test_task_tools_preserve_code_execution_tools():
"""
from typing import Type
from crewai.tools import BaseTool
from crewai_tools import CodeInterpreterTool
from pydantic import BaseModel, Field
from crewai.tools import BaseTool
class TestToolInput(BaseModel):
"""Input schema for TestTool."""