From f2db68ef0b0d11d4779605b45d2adabda9c7c8e4 Mon Sep 17 00:00:00 2001
From: Vladimir Blagojevic <dovlex@gmail.com>
Date: Thu, 2 Nov 2023 10:56:52 +0100
Subject: [PATCH] fix: Add new rankers to nodes __init__.py (#6219)

* Add new rankers to nodes __init__.py

* Add release note
---
 haystack/nodes/__init__.py                               | 9 ++++++++-
 ...d-new-rankers-schema-generation-7ad92fd5c5de8937.yaml | 5 +++++
 2 files changed, 13 insertions(+), 1 deletion(-)
 create mode 100644 releasenotes/notes/add-new-rankers-schema-generation-7ad92fd5c5de8937.yaml

diff --git a/haystack/nodes/__init__.py b/haystack/nodes/__init__.py
index 062acfb26..a969e245b 100644
--- a/haystack/nodes/__init__.py
+++ b/haystack/nodes/__init__.py
@@ -26,7 +26,14 @@ from haystack.nodes.prompt import PromptNode, PromptTemplate, PromptModel, BaseO
 from haystack.nodes.prompt.invocation_layer import PromptModelInvocationLayer
 from haystack.nodes.query_classifier import TransformersQueryClassifier
 from haystack.nodes.question_generator import QuestionGenerator
-from haystack.nodes.ranker import BaseRanker, SentenceTransformersRanker, CohereRanker
+from haystack.nodes.ranker import (
+    BaseRanker,
+    SentenceTransformersRanker,
+    CohereRanker,
+    LostInTheMiddleRanker,
+    DiversityRanker,
+    RecentnessRanker,
+)
 from haystack.nodes.reader import BaseReader, FARMReader, TransformersReader, TableReader, RCIReader
 from haystack.nodes.retriever import (
     BaseRetriever,
diff --git a/releasenotes/notes/add-new-rankers-schema-generation-7ad92fd5c5de8937.yaml b/releasenotes/notes/add-new-rankers-schema-generation-7ad92fd5c5de8937.yaml
new file mode 100644
index 000000000..b77bdf0cc
--- /dev/null
+++ b/releasenotes/notes/add-new-rankers-schema-generation-7ad92fd5c5de8937.yaml
@@ -0,0 +1,5 @@
+---
+fixes:
+  - |
+    Adds LostInTheMiddleRanker, DiversityRanker, and RecentnessRanker to `haystack/nodes/__init__.py` and thus
+    ensures that they are included in JSON schema generation.