[Bugfix] Fix for Issue #6241 - ChromaDB removed IncludeEnum

python/packages/autogen-ext/pyproject.toml

@@ -41,7 +41,7 @@ llama-cpp = [
 ]
 
 graphrag = ["graphrag>=1.0.1"]
-chromadb = ["chromadb"]
+chromadb = ["chromadb>=1.0.0"]
 web-surfer = [
     "autogen-agentchat==0.5.1",
     "playwright>=1.48.0",
@@ -78,7 +78,7 @@ jupyter-executor = [
     "nbclient>=0.10.2",
 ]
 
-task-centric-memory = ["chromadb>=0.6.3"]
+task-centric-memory = ["chromadb>=1.0.0"]
 
 semantic-kernel-core = [
     "semantic-kernel>=1.17.1",

python/packages/autogen-ext/src/autogen_ext/memory/chromadb.py

@@ -7,9 +7,8 @@
 from autogen_core.model_context import ChatCompletionContext
 from autogen_core.models import SystemMessage
 from chromadb import HttpClient, PersistentClient
-from chromadb.api import ClientAPI
 from chromadb.api.models.Collection import Collection
-from chromadb.api.types import Document, IncludeEnum, Metadata
+from chromadb.api.types import Document, Metadata
 from pydantic import BaseModel, Field
 from typing_extensions import Self
 
@@ -331,7 +330,7 @@ async def query(
             results = self._collection.query(
                 query_texts=[query_text],
                 n_results=self._config.k,
-                include=[IncludeEnum.documents, IncludeEnum.metadatas, IncludeEnum.distances],
+                include=["documents", "metadatas", "distances"],
                 **kwargs,
             )
 

python/samples/gitty/src/gitty/_db.py

@@ -55,8 +55,6 @@ def update_chroma(gitty_dir: str, db_path: str) -> None:
     conn.close()
 
     sentence_transformer_ef = embedding_functions.DefaultEmbeddingFunction()
-    if sentence_transformer_ef is None:
-        raise RuntimeError("Default embedding function is not available.")
 
     for issue_number, title, content in rows:
         meta = {"title": title}
@@ -160,10 +158,6 @@ def fetch_and_update_issues(owner: str, repo: str, db_path: Optional[str] = None
     # New embedding function using sentence_transformers
     sentence_transformer_ef = embedding_functions.DefaultEmbeddingFunction()
 
-    if sentence_transformer_ef is None:
-        print("Error: Default embedding function is not available.")
-        exit(1)
-
     for issue_number, title, content in rows:
         meta = {"title": title}  # metadata for each issue
         embedding = sentence_transformer_ef([content])[0]