fix linter and types

src/crewai/tools/agent_tools/add_image_tool.py

@@ -1,12 +1,13 @@
 from crewai.tools.base_tool import BaseTool
 from crewai.utilities import I18N
 from pydantic import BaseModel, Field
+from typing import Optional
 
 i18n = I18N()
 
 class AddImageToolSchema(BaseModel):
     image_url: str = Field(..., description="The URL or path of the image to add")
-    action: str = Field(
+    action: Optional[str] = Field(
         default=None,
         description="Optional context or question about the image"
     )
@@ -15,17 +16,17 @@ class AddImageToolSchema(BaseModel):
 class AddImageTool(BaseTool):
     """Tool for adding images to the content"""
 
-    name: str = Field(default_factory=lambda: i18n.tools("add_image")["name"])
+    name: str = Field(default_factory=lambda: i18n.tools("add_image").get("name"))
-    description: str = Field(default_factory=lambda: i18n.tools("add_image")["description"])
+    description: str = Field(default_factory=lambda: i18n.tools("add_image").get("description"))
     args_schema: type[BaseModel] = AddImageToolSchema
 
     def _run(
         self,
         image_url: str,
-        action: str = None,
+        action: Optional[str] = None,
         **kwargs,
     ) -> dict:
-        action = action or i18n.tools("add_image")["default_action"]
+        action = action or i18n.tools("add_image").get("default_action")
         content = [
             {"type": "text", "text": action},
             {

tests/knowledge/knowledge_test.py

@@ -5,7 +5,6 @@ from typing import List, Union
 from unittest.mock import patch
 
 import pytest
-
 from crewai.knowledge.source.crew_docling_source import CrewDoclingSource
 from crewai.knowledge.source.csv_knowledge_source import CSVKnowledgeSource
 from crewai.knowledge.source.excel_knowledge_source import ExcelKnowledgeSource

tests/test_task_guardrails.py

@@ -3,7 +3,6 @@
 from unittest.mock import Mock
 
 import pytest
-
 from crewai.task import Task
 from crewai.tasks.task_output import TaskOutput