feat(chroma): Add Search API (#34273)

Author: itaismith

libs/partners/chroma/langchain_chroma/vectorstores.py

@@ -18,7 +18,7 @@
 import chromadb
 import chromadb.config
 import numpy as np
-from chromadb import Settings
+from chromadb import Search, Settings
 from chromadb.api import CreateCollectionConfiguration
 from langchain_core.documents import Document
 from langchain_core.embeddings import Embeddings
@@ -681,6 +681,52 @@ def add_texts(
             )
         return ids
 
+    def hybrid_search(self, search: Search) -> list[Document]:
+        """Run hybrid search with Chroma.
+
+        Args:
+            search: The Search configuration for hybrid search.
+
+        Returns:
+            A list of documents resulting from the search operation.
+
+        Example:
+            from chromadb import Search, K, Knn, Rrf
+
+            # Create RRF ranking with text query
+            hybrid_rank = Rrf(
+                ranks=[
+                    Knn(query="query", return_rank=True, limit=300),
+                    Knn(query="query learning applications", key="sparse_embedding")
+                ],
+                weights=[2.0, 1.0],  # Dense 2x more important
+                k=60
+            )
+
+            # Build complete the search strategy
+            search = (Search()
+                .where(
+                    (K("language") == "en") &
+                    (K("year") >= 2020)
+                )
+                .rank(hybrid_rank)
+                .limit(10)
+                .select(K.DOCUMENT, K.SCORE, "title", "year")
+            )
+
+            results = vector_store.hybrid_search(search)
+        """
+        results = self._collection.search(search)
+        return [
+            Document(
+                page_content=record["document"],
+                metadata=record["metadata"],
+                id=record["id"],
+            )
+            for record in results.rows()[0]
+            if record["document"] is not None
+        ]
+
     def similarity_search(
         self,
         query: str,

libs/partners/chroma/pyproject.toml

@@ -10,7 +10,7 @@ dependencies = [
     "langchain-core>=1.0.0,<2.0.0",
     "numpy>=1.26.0; python_version < '3.13'",
     "numpy>=2.1.0; python_version >= '3.13'",
-    "chromadb>=1.0.20,<2.0.0",
+    "chromadb>=1.3.5,<2.0.0",
 ]
 name = "langchain-chroma"
 version = "1.0.0"