Feature/bump chromadb dep to 0.5.0 (#700)

* [DEP]: Bump chromadb version

Remove specific version pin

* Fix: chromadb tests failing due to sort order variance

Fix: linting error

* Add OllamaEmbeddingFunction to function registry

Linting

integrations/chroma/pyproject.toml

@@ -26,7 +26,7 @@ classifiers = [
 ]
 dependencies = [
   "haystack-ai",
-  "chromadb<0.4.20",  #  FIXME: investigate why filtering tests broke on 0.4.20
+  "chromadb",
   "typing_extensions>=4.8.0"
 ]
 

integrations/chroma/src/haystack_integrations/document_stores/chroma/utils.py

@@ -9,6 +9,7 @@
     GoogleVertexEmbeddingFunction,
     HuggingFaceEmbeddingFunction,
     InstructorEmbeddingFunction,
+    OllamaEmbeddingFunction,
     ONNXMiniLM_L6_V2,
     OpenAIEmbeddingFunction,
     SentenceTransformerEmbeddingFunction,
@@ -25,6 +26,7 @@
     "GoogleVertexEmbeddingFunction": GoogleVertexEmbeddingFunction,
     "HuggingFaceEmbeddingFunction": HuggingFaceEmbeddingFunction,
     "InstructorEmbeddingFunction": InstructorEmbeddingFunction,
+    "OllamaEmbeddingFunction": OllamaEmbeddingFunction,
     "ONNXMiniLM_L6_V2": ONNXMiniLM_L6_V2,
     "OpenAIEmbeddingFunction": OpenAIEmbeddingFunction,
     "Text2VecEmbeddingFunction": Text2VecEmbeddingFunction,

integrations/chroma/tests/test_document_store.py

@@ -1,6 +1,7 @@
 # SPDX-FileCopyrightText: 2023-present John Doe <[email protected]>
 #
 # SPDX-License-Identifier: Apache-2.0
+import operator
 import uuid
 from typing import List
 from unittest import mock
@@ -56,6 +57,9 @@ def assert_documents_are_equal(self, received: List[Document], expected: List[Do
         This can happen for example when the Document Store sets a score to returned Documents.
         Since we can't know what the score will be, we can't compare the Documents reliably.
         """
+        received.sort(key=operator.attrgetter("id"))
+        expected.sort(key=operator.attrgetter("id"))
+
         for doc_received, doc_expected in zip(received, expected):
             assert doc_received.content == doc_expected.content
             assert doc_received.meta == doc_expected.meta