From 6ef5e56c5cedbfe8455ee1e0f4ba60bd8dfa433d Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Thu, 29 Jun 2023 11:16:05 -0300
Subject: [PATCH] =?UTF-8?q?=E2=9C=A8=20feat(graph):=20add=20RetrieverVerte?=
 =?UTF-8?q?x=20to=20the=20list=20of=20available=20vertex=20types=20The=20R?=
 =?UTF-8?q?etrieverVertex=20class=20is=20added=20to=20the=20list=20of=20av?=
 =?UTF-8?q?ailable=20vertex=20types=20in=20the=20langflow.graph.vertex.typ?=
 =?UTF-8?q?es=20module.=20This=20allows=20for=20the=20creation=20of=20vert?=
 =?UTF-8?q?ices=20that=20represent=20retrievers=20in=20the=20graph.?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/backend/langflow/graph/__init__.py     | 2 ++
 src/backend/langflow/graph/vertex/types.py | 5 +++++
 2 files changed, 7 insertions(+)

diff --git a/src/backend/langflow/graph/__init__.py b/src/backend/langflow/graph/__init__.py
index a68e844ee..e63b9dcc0 100644
--- a/src/backend/langflow/graph/__init__.py
+++ b/src/backend/langflow/graph/__init__.py
@@ -14,6 +14,7 @@ from langflow.graph.vertex.types import (
     ToolkitVertex,
     VectorStoreVertex,
     WrapperVertex,
+    RetrieverVertex,
 )
 
 __all__ = [
@@ -32,4 +33,5 @@ __all__ = [
     "ToolkitVertex",
     "VectorStoreVertex",
     "WrapperVertex",
+    "RetrieverVertex",
 ]
diff --git a/src/backend/langflow/graph/vertex/types.py b/src/backend/langflow/graph/vertex/types.py
index b6a8f9a91..846b79bc0 100644
--- a/src/backend/langflow/graph/vertex/types.py
+++ b/src/backend/langflow/graph/vertex/types.py
@@ -112,6 +112,11 @@ class MemoryVertex(Vertex):
         super().__init__(data, base_type="memory")
 
 
+class RetrieverVertex(Vertex):
+    def __init__(self, data: Dict):
+        super().__init__(data, base_type="retrievers")
+
+
 class TextSplitterVertex(Vertex):
     def __init__(self, data: Dict):
         super().__init__(data, base_type="textsplitters")