From 64bc7c40edbd2d813eb8169cb7c108909f21385d Mon Sep 17 00:00:00 2001
From: Gabriel Almeida <gabriel@logspace.ai>
Date: Tue, 28 Mar 2023 21:57:02 -0300
Subject: [PATCH] fix: replace union with 3.8 compliant syntax

---
 src/backend/langflow/graph/graph.py | 4 ++--
 src/backend/langflow/utils/util.py  | 8 ++++----
 2 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/src/backend/langflow/graph/graph.py b/src/backend/langflow/graph/graph.py
index 9275351a9..8e3c06ef5 100644
--- a/src/backend/langflow/graph/graph.py
+++ b/src/backend/langflow/graph/graph.py
@@ -257,7 +257,7 @@ class PromptNode(Node):
     def build(
         self,
         force: bool = False,
-        tools: Optional[List[Node]] | Optional[List[ToolNode]] = None,
+        tools: Optional[Union[List[Node], List[ToolNode]]] = None,
     ) -> Any:
         if not self._built or force:
             # Check if it is a ZeroShotPrompt and needs a tool
@@ -280,7 +280,7 @@ class ChainNode(Node):
     def build(
         self,
         force: bool = False,
-        tools: Optional[List[Node]] | Optional[List[ToolNode]] = None,
+        tools: Optional[Union[List[Node], List[ToolNode]]] = None,
     ) -> Any:
         if not self._built or force:
             # Check if the chain requires a PromptNode
diff --git a/src/backend/langflow/utils/util.py b/src/backend/langflow/utils/util.py
index 8fdee645f..664630890 100644
--- a/src/backend/langflow/utils/util.py
+++ b/src/backend/langflow/utils/util.py
@@ -2,7 +2,7 @@ import ast
 import importlib
 import inspect
 import re
-from typing import Dict, Optional
+from typing import Dict, Optional, Union
 
 from langchain.agents.load_tools import (
     _BASE_TOOLS,
@@ -170,7 +170,7 @@ def get_tool_by_name(name: str):
     return tools[name]
 
 
-def get_tool_params(tool, **kwargs) -> Dict | None:
+def get_tool_params(tool, **kwargs) -> Union[Dict, None]:
     # Parse the function code into an abstract syntax tree
     # Define if it is a function or a class
     if inspect.isfunction(tool):
@@ -183,7 +183,7 @@ def get_tool_params(tool, **kwargs) -> Dict | None:
         raise ValueError("Tool must be a function or class.")
 
 
-def get_func_tool_params(func, **kwargs) -> Dict | None:
+def get_func_tool_params(func, **kwargs) -> Union[Dict, None]:
     tree = ast.parse(inspect.getsource(func))
 
     # Iterate over the statements in the abstract syntax tree
@@ -226,7 +226,7 @@ def get_func_tool_params(func, **kwargs) -> Dict | None:
     return None
 
 
-def get_class_tool_params(cls, **kwargs) -> Dict | None:
+def get_class_tool_params(cls, **kwargs) -> Union[Dict, None]:
     tree = ast.parse(inspect.getsource(cls))
 
     tool_params = {}