Fix linting and type-checker issues

This commit is contained in:
Brandon Hancock
2024-07-09 16:52:18 -04:00
parent 9fdaffc073
commit 2abc971035
2 changed files with 1 additions and 5 deletions

View File

@@ -180,7 +180,7 @@ class Task(BaseModel):
tools: Optional[List[Any]] = None,
) -> Future[TaskOutput]:
"""Execute the task asynchronously."""
future = Future()
future: Future[TaskOutput] = Future()
threading.Thread(
target=self._execute_task_async, args=(agent, context, tools, future)
).start()
@@ -388,7 +388,6 @@ class Task(BaseModel):
converter.to_pydantic() if self.output_pydantic else converter.to_json()
)
if isinstance(exported_result, ConverterError):
Printer().print(
content=f"{exported_result.message} Using raw output instead.",

View File

@@ -7,7 +7,6 @@ from unittest.mock import patch
import pydantic_core
import pytest
from crewai.agent import Agent
from crewai.agents.cache import CacheHandler
from crewai.crew import Crew
@@ -709,7 +708,6 @@ def test_kickoff_for_each_single_input():
@pytest.mark.vcr(filter_headers=["authorization"])
def test_kickoff_for_each_multiple_inputs():
"""Tests if kickoff_for_each works with multiple inputs."""
from unittest.mock import patch
inputs = [
{"topic": "dog"},
@@ -1084,7 +1082,6 @@ def test_code_execution_flag_adds_code_tool_upon_kickoff():
@pytest.mark.vcr(filter_headers=["authorization"])
def test_delegation_is_not_enabled_if_there_are_only_one_agent():
from unittest.mock import patch
researcher = Agent(
role="Researcher",