fix: replace union with 3.8 compliant syntax

This commit is contained in:
Gabriel Almeida 2023-03-28 21:57:02 -03:00
commit 64bc7c40ed
2 changed files with 6 additions and 6 deletions

View file

@ -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

View file

@ -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 = {}