fix: resolve lint issues in ChunkBasedTask implementation

- Remove unused Dict import from typing
- Fix f-string without placeholders
- Remove unused imports and variables in tests

Co-Authored-By: Jo\u00E3o <joao@crewai.com>
This commit is contained in:
Devin AI
2025-07-13 00:34:24 +00:00
parent f836b2dc90
commit 9497715b42
3 changed files with 4 additions and 6 deletions

View File

@@ -1,6 +1,6 @@
import datetime import datetime
from pathlib import Path 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 pydantic import Field, field_validator
from crewai.task import Task from crewai.task import Task
@@ -61,7 +61,7 @@ class ChunkBasedTask(Task):
"""Create a task description for processing a single chunk.""" """Create a task description for processing a single chunk."""
chunk_context = f"Processing chunk {chunk_index + 1} of {total_chunks}" chunk_context = f"Processing chunk {chunk_index + 1} of {total_chunks}"
if self.chunk_results: 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}" return f"{chunk_context}\n\n{self.description}\n\nChunk content:\n{chunk}"

View File

@@ -4,7 +4,6 @@ from pathlib import Path
from unittest.mock import Mock, patch from unittest.mock import Mock, patch
from crewai.tasks.chunk_based_task import ChunkBasedTask from crewai.tasks.chunk_based_task import ChunkBasedTask
from crewai.agent import Agent
from crewai.tasks.task_output import TaskOutput from crewai.tasks.task_output import TaskOutput
@@ -110,7 +109,7 @@ class TestChunkBasedTask:
mock_agent.role = "test_agent" mock_agent.role = "test_agent"
mock_agent.crew = None 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 len(task.chunk_results) > 1
assert mock_execute.call_count > len(task.chunk_results) assert mock_execute.call_count > len(task.chunk_results)

View File

@@ -1,4 +1,3 @@
import pytest
import tempfile import tempfile
from pathlib import Path from pathlib import Path
@@ -42,7 +41,7 @@ class TestChunkBasedTaskIntegration:
agent=agent agent=agent
) )
crew = Crew( Crew(
agents=[agent], agents=[agent],
tasks=[task], tasks=[task],
memory=True memory=True