andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-11 00:58:30 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Fix import error handling for optional chromadb dependency using TYPE_CHECKING

Browse Source 
Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
Devin AI
2025-04-26 19:12:30 +00:00
parent 86155b4590
commit 2b7b2993ca
2 changed files with 29 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/crewai/knowledge/storage/knowledge_storage.py
View File

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

8
src/crewai/memory/storage/rag_storage.py
View File

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