andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-05-03 08:12:39 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Fix: Make ChromaDB imports lazy to prevent SQLite3 version check on import (fixes #2682)

Browse Source 
Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
Devin AI
2025-04-24 10:18:03 +00:00
parent 685d20f46c
commit 4434481f03
6 changed files with 446 additions and 1569 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -4,9 +4,11 @@ import logging
import os
import shutil
import uuid
from typing import Any, Dict, List, Optional
from typing import Any, Dict, List, Optional, TYPE_CHECKING
from chromadb.api import ClientAPI
# Type checking imports that don't cause runtime imports
if TYPE_CHECKING:
from chromadb.api import ClientAPI
from crewai.memory.storage.base_rag_storage import BaseRAGStorage
from crewai.utilities import EmbeddingConfigurator
@@ -37,7 +39,7 @@ class RAGStorage(BaseRAGStorage):
search efficiency.
"""
app: ClientAPI | None = None
app: Optional[Any] = None
def __init__(
self, type, allow_reset=True, embedder_config=None, crew=None, path=None
@@ -60,6 +62,7 @@ class RAGStorage(BaseRAGStorage):
self.embedder_config = configurator.configure_embedder(self.embedder_config)
def _initialize_app(self):
# Import chromadb here to avoid importing at module level
import chromadb
from chromadb.config import Settings