From 5bd4fdc3d0e0c860bfa0dd54c0a57e142c88d641 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jo=C3=A3o=20Moura?= <joaomdmoura@gmail.com>
Date: Fri, 27 Dec 2024 16:19:56 -0300
Subject: [PATCH] fix types and linter

---
 src/crewai/tools/agent_tools/add_image_tool.py | 3 ++-
 src/crewai/utilities/i18n.py                   | 6 +++---
 2 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/src/crewai/tools/agent_tools/add_image_tool.py b/src/crewai/tools/agent_tools/add_image_tool.py
index 6b081ba2e..83484fcd5 100644
--- a/src/crewai/tools/agent_tools/add_image_tool.py
+++ b/src/crewai/tools/agent_tools/add_image_tool.py
@@ -1,7 +1,8 @@
+from typing import Optional
+
 from crewai.tools.base_tool import BaseTool
 from crewai.utilities import I18N
 from pydantic import BaseModel, Field
-from typing import Optional
 
 i18n = I18N()
 
diff --git a/src/crewai/utilities/i18n.py b/src/crewai/utilities/i18n.py
index e325834f3..ebf1abcda 100644
--- a/src/crewai/utilities/i18n.py
+++ b/src/crewai/utilities/i18n.py
@@ -1,6 +1,6 @@
 import json
 import os
-from typing import Dict, Optional
+from typing import Dict, Optional, Union
 
 from pydantic import BaseModel, Field, PrivateAttr, model_validator
 
@@ -41,8 +41,8 @@ class I18N(BaseModel):
     def errors(self, error: str) -> str:
         return self.retrieve("errors", error)
 
-    def tools(self, error: str) -> str:
-        return self.retrieve("tools", error)
+    def tools(self, tool: str) -> Union[str, Dict[str, str]]:
+        return self.retrieve("tools", tool)
 
     def retrieve(self, kind, key) -> str:
         try: