Refactor inputs parameter to inputs_dict in build_vertex function

This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-03-25 16:17:53 -03:00
commit e14d089527
2 changed files with 2 additions and 3 deletions

View file

@ -159,7 +159,7 @@ async def build_vertex(
set_cache_coro=set_cache_coro,
vertex_id=vertex_id,
user_id=current_user.id,
inputs=inputs.model_dump() if inputs else {},
inputs_dict=inputs.model_dump() if inputs else {},
)
result_data_response = ResultDataResponse(**result_dict.model_dump())

View file

@ -652,7 +652,7 @@ class Graph:
lock: asyncio.Lock,
set_cache_coro: Coroutine,
vertex_id: str,
inputs: Optional[Dict[str, str]] = None,
inputs_dict: Optional[Dict[str, str]] = None,
user_id: Optional[str] = None,
):
"""
@ -675,7 +675,6 @@ class Graph:
vertex = self.get_vertex(vertex_id)
try:
if not vertex.frozen or not vertex._built:
inputs_dict = inputs.model_dump() if inputs else {}
await vertex.build(user_id=user_id, inputs=inputs_dict)
if vertex.result is not None: