diff --git a/src/crewai/tasks/chunk_based_task.py b/src/crewai/tasks/chunk_based_task.py index 379e78ddf..180388893 100644 --- a/src/crewai/tasks/chunk_based_task.py +++ b/src/crewai/tasks/chunk_based_task.py @@ -1,6 +1,6 @@ import datetime from pathlib import Path -from typing import Any, Dict, List, Optional, Union +from typing import Any, List, Optional, Union from pydantic import Field, field_validator from crewai.task import Task @@ -61,7 +61,7 @@ class ChunkBasedTask(Task): """Create a task description for processing a single chunk.""" chunk_context = f"Processing chunk {chunk_index + 1} of {total_chunks}" if self.chunk_results: - chunk_context += f"\n\nPrevious chunk insights from memory will be available." + chunk_context += "\n\nPrevious chunk insights from memory will be available." return f"{chunk_context}\n\n{self.description}\n\nChunk content:\n{chunk}" diff --git a/tests/test_chunk_based_task.py b/tests/test_chunk_based_task.py index 9d7696a0c..a55a0a4a4 100644 --- a/tests/test_chunk_based_task.py +++ b/tests/test_chunk_based_task.py @@ -4,7 +4,6 @@ from pathlib import Path from unittest.mock import Mock, patch from crewai.tasks.chunk_based_task import ChunkBasedTask -from crewai.agent import Agent from crewai.tasks.task_output import TaskOutput @@ -110,7 +109,7 @@ class TestChunkBasedTask: mock_agent.role = "test_agent" mock_agent.crew = None - result = task._execute_core(mock_agent, None, None) + task._execute_core(mock_agent, None, None) assert len(task.chunk_results) > 1 assert mock_execute.call_count > len(task.chunk_results) diff --git a/tests/test_chunk_based_task_integration.py b/tests/test_chunk_based_task_integration.py index 03e700e80..25256ab4b 100644 --- a/tests/test_chunk_based_task_integration.py +++ b/tests/test_chunk_based_task_integration.py @@ -1,4 +1,3 @@ -import pytest import tempfile from pathlib import Path @@ -42,7 +41,7 @@ class TestChunkBasedTaskIntegration: agent=agent ) - crew = Crew( + Crew( agents=[agent], tasks=[task], memory=True