diff --git a/lib/crewai/src/crewai/lite_agent.py b/lib/crewai/src/crewai/lite_agent.py
index c3c6d2cc7..a0f2a90af 100644
--- a/lib/crewai/src/crewai/lite_agent.py
+++ b/lib/crewai/src/crewai/lite_agent.py
@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import asyncio
 from collections.abc import Callable
 import inspect
diff --git a/lib/crewai/src/crewai/task.py b/lib/crewai/src/crewai/task.py
index 87ce1ed43..0c467dbaf 100644
--- a/lib/crewai/src/crewai/task.py
+++ b/lib/crewai/src/crewai/task.py
@@ -10,7 +10,6 @@ import logging
 from pathlib import Path
 import threading
 from typing import (
-    TYPE_CHECKING,
     Any,
     ClassVar,
     cast,
@@ -31,10 +30,6 @@ from pydantic import (
 from pydantic_core import PydanticCustomError
 from typing_extensions import Self
 
-
-if TYPE_CHECKING:
-    from crewai_files import FileInput
-
 from crewai.agents.agent_builder.base_agent import BaseAgent
 from crewai.events.event_bus import crewai_event_bus
 from crewai.events.types.task_events import (
@@ -57,10 +52,11 @@ from crewai.utilities.file_store import (
 
 
 try:
-    from crewai_files import FilePath
+    from crewai_files import FileInput, FilePath
 
     HAS_CREWAI_FILES = True
 except ImportError:
+    FileInput = Any  # type: ignore[misc,assignment]
     HAS_CREWAI_FILES = False
 from crewai.utilities.guardrail import (
     process_guardrail,