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

Refactor inputs parameter to inputs_dict in build_vertex function

Browse source
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
Download patch file Download diff file Expand all files Collapse all files

2
src/backend/langflow/api/v1/chat.py
View file

@ -159,7 +159,7 @@ async def build_vertex(
set_cache_coro=set_cache_coro, set_cache_coro=set_cache_coro,
vertex_id=vertex_id, vertex_id=vertex_id,
user_id=current_user.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()) result_data_response = ResultDataResponse(**result_dict.model_dump())

3
src/backend/langflow/graph/graph/base.py
View file

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