Fix import error handling for optional chromadb dependency using TYPE_CHECKING

Co-Authored-By: Joe Moura <joao@crewai.com>

src/crewai/knowledge/storage/knowledge_storage.py

@@ -6,13 +6,20 @@ import os
 import shutil
 from typing import TYPE_CHECKING, Any, Dict, List, Optional, Union
 
-try:
+if TYPE_CHECKING:
     import chromadb
     import chromadb.errors
     from chromadb.api import ClientAPI
     from chromadb.api.types import OneOrMany
     from chromadb.config import Settings
-except ImportError:
+else:
+    try:
+        import chromadb
+        import chromadb.errors
+        from chromadb.api import ClientAPI
+        from chromadb.api.types import OneOrMany
+        from chromadb.config import Settings
+    except ImportError:
         chromadb = None
         ClientAPI = None
         OneOrMany = None

src/crewai/memory/storage/rag_storage.py

@@ -6,11 +6,19 @@ import shutil
 import uuid
 from typing import TYPE_CHECKING, Any, Dict, List, Optional
 
-try:
+if TYPE_CHECKING:
     import chromadb
     from chromadb.api import ClientAPI
     from chromadb.config import Settings
-except ImportError:
+else:
+    try:
+        import chromadb
+        from chromadb.api import ClientAPI
+        from chromadb.config import Settings
+    except ImportError:
+        chromadb = None
+        ClientAPI = None
+        Settings = None
         raise ImportError(
             "ChromaDB is not installed. Please install it with `pip install crewai[chromadb]`."
         )