added LocalRerankDriver

__init__.py

@@ -168,6 +168,7 @@ from .nodes.drivers.gtUILeonardoImageGenerationDriver import (
 # - LM Studio
 from .nodes.drivers.gtUILMStudioChatPromptDriver import gtUILMStudioChatPromptDriver
 from .nodes.drivers.gtUILMStudioEmbeddingDriver import gtUILMStudioEmbeddingDriver
+from .nodes.drivers.gtUILocalRerankDriver import gtUILocalRerankDriver
 
 # - Local
 from .nodes.drivers.gtUILocalVectorStoreDriver import gtUILocalVectorStoreDriver
@@ -393,6 +394,7 @@ NODE_CLASS_MAPPINGS = {
     "Griptape Embedding Driver: Voyage AI": gtUIVoyageAiEmbeddingDriver,
     # RERANK DRIVER
     "Griptape Rerank Driver: Cohere": gtUICohereRerankDriver,
+    "Griptape Rerank Driver: Local": gtUILocalRerankDriver,
     # VECTOR STORE DRIVERS
     "Griptape Vector Store Driver: Amazon OpenSearch": gtUIAmazonOpenSearchVectorStoreDriver,
     "Griptape Vector Store Driver: Azure MongoDB": gtUIAzureMongoDbVectorStoreDriver,

nodes/drivers/gtUILocalRerankDriver.py

@@ -0,0 +1,25 @@
+from typing import Any, Tuple
+
+from griptape.drivers.rerank.local import LocalRerankDriver
+
+from .gtUIBaseRerankDriver import gtUIBaseRerankDriver
+
+
+class gtUILocalRerankDriver(gtUIBaseRerankDriver):
+    DESCRIPTION = "Griptape Local Rerank Driver"
+
+    @classmethod
+    def INPUT_TYPES(cls):
+        inputs = super().INPUT_TYPES()
+        inputs["required"].update()
+        inputs["optional"].update()
+
+        return inputs
+
+    CATEGORY = "Griptape/Agent Drivers/Rerank"
+
+    RETURN_TYPES = ("RERANK_DRIVER",)
+
+    def create(self, **kwargs) -> Tuple[Any, ...]:
+        driver = LocalRerankDriver()
+        return (driver,)