Feature/bump chromadb dep to 0.5.0 (deepset-ai#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