Update input type options in schemas and graph base classes

This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-03-31 23:12:18 -03:00
commit 888a6904d5
2 changed files with 4 additions and 4 deletions

View file

@ -259,7 +259,7 @@ class VerticesBuiltResponse(BaseModel):
class InputValueRequest(BaseModel):
components: Optional[List[str]] = []
input_value: Optional[str] = None
type: Optional[Literal["chat", "text", "json", "any"]] = Field(
type: Optional[Literal["chat", "text", "any"]] = Field(
"any",
description="Defines on which components the input value should be applied. 'any' applies to all input components.",
)

View file

@ -201,7 +201,7 @@ class Graph:
self,
inputs: Dict[str, str],
input_components: list[str],
input_type: Literal["chat", "text", "json", "any"] | None,
input_type: Literal["chat", "text", "any"] | None,
outputs: list[str],
stream: bool,
session_id: str,
@ -271,7 +271,7 @@ class Graph:
self,
inputs: list[Dict[str, str]],
input_components: Optional[list[list[str]]] = None,
types: Optional[list[Literal["chat", "text", "json", "any"] | None]] = None,
types: Optional[list[Literal["chat", "text", "any"] | None]] = None,
outputs: Optional[list[str]] = None,
session_id: Optional[str] = None,
stream: bool = False,
@ -309,7 +309,7 @@ class Graph:
self,
inputs: list[Dict[str, str]],
inputs_components: Optional[list[list[str]]] = None,
types: Optional[list[Literal["chat", "text", "json", "any"] | None]] = None,
types: Optional[list[Literal["chat", "text", "any"] | None]] = None,
outputs: Optional[list[str]] = None,
session_id: Optional[str] = None,
stream: bool = False,