Update input type in Graph class

This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-04-01 10:37:51 -03:00
commit c63f603b57

View file

@ -15,7 +15,7 @@ from langflow.graph.vertex.base import Vertex
from langflow.graph.vertex.types import ChatVertex, FileToolVertex, LLMVertex, RoutingVertex, StateVertex, ToolkitVertex from langflow.graph.vertex.types import ChatVertex, FileToolVertex, LLMVertex, RoutingVertex, StateVertex, ToolkitVertex
from langflow.interface.tools.constants import FILE_TOOLS from langflow.interface.tools.constants import FILE_TOOLS
from langflow.schema import Record from langflow.schema import Record
from langflow.schema.schema import INPUT_FIELD_NAME from langflow.schema.schema import INPUT_FIELD_NAME, InputType
if TYPE_CHECKING: if TYPE_CHECKING:
from langflow.graph.schema import ResultData from langflow.graph.schema import ResultData
@ -201,7 +201,7 @@ class Graph:
self, self,
inputs: Dict[str, str], inputs: Dict[str, str],
input_components: list[str], input_components: list[str],
input_type: Literal["chat", "text", "any"] | None, input_type: InputType | None,
outputs: list[str], outputs: list[str],
stream: bool, stream: bool,
session_id: str, session_id: str,
@ -271,7 +271,7 @@ class Graph:
self, self,
inputs: list[Dict[str, str]], inputs: list[Dict[str, str]],
input_components: Optional[list[list[str]]] = None, input_components: Optional[list[list[str]]] = None,
types: Optional[list[Literal["chat", "text", "any"] | None]] = None, types: Optional[list[InputType | None]] = None,
outputs: Optional[list[str]] = None, outputs: Optional[list[str]] = None,
session_id: Optional[str] = None, session_id: Optional[str] = None,
stream: bool = False, stream: bool = False,
@ -309,7 +309,7 @@ class Graph:
self, self,
inputs: list[Dict[str, str]], inputs: list[Dict[str, str]],
inputs_components: Optional[list[list[str]]] = None, inputs_components: Optional[list[list[str]]] = None,
types: Optional[list[Literal["chat", "text", "any"] | None]] = None, types: Optional[list[InputType | None]] = None,
outputs: Optional[list[str]] = None, outputs: Optional[list[str]] = None,
session_id: Optional[str] = None, session_id: Optional[str] = None,
stream: bool = False, stream: bool = False,