jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Add type hint extraction functions for

Browse source 
GenericAlias
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-11-22 21:09:35 -03:00
commit 5812d3b91e
1 changed files with 19 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

19
src/backend/langflow/interface/custom/utils.py
View file

@ -1,4 +1,6 @@
import re import re
from types import GenericAlias
from typing import Any
def extract_inner_type(return_type: str) -> str: def extract_inner_type(return_type: str) -> str:
@ -10,6 +12,23 @@ def extract_inner_type(return_type: str) -> str:
return return_type return return_type
def extract_inner_type_from_generic_alias(return_type: GenericAlias) -> Any:
"""
Extracts the inner type from a type hint that is a list.
"""
if return_type.__origin__ == list:
return list(return_type.__args__)
return return_type
def extract_union_types_from_generic_alias(return_type: GenericAlias) -> tuple:
"""
Extracts the inner type from a type hint that is a Union.
"""
return return_type.__args__
def extract_union_types(return_type: str) -> list[str]: def extract_union_types(return_type: str) -> list[str]:
""" """
Extracts the inner type from a type hint that is a list. Extracts the inner type from a type hint that is a list.