Refactor Vertex class and update params dictionary
This commit is contained in:
parent
fdfbe810c9
commit
54758662c2
1 changed files with 4 additions and 5 deletions
|
|
@ -2,8 +2,8 @@ import ast
|
||||||
import inspect
|
import inspect
|
||||||
import types
|
import types
|
||||||
from typing import TYPE_CHECKING, Any, Callable, Coroutine, Dict, List, Optional
|
from typing import TYPE_CHECKING, Any, Callable, Coroutine, Dict, List, Optional
|
||||||
from langflow.graph.schema import InterfaceComponentTypes
|
|
||||||
|
|
||||||
|
from langflow.graph.schema import InterfaceComponentTypes
|
||||||
from langflow.graph.utils import UnbuiltObject, UnbuiltResult
|
from langflow.graph.utils import UnbuiltObject, UnbuiltResult
|
||||||
from langflow.graph.vertex.utils import generate_result
|
from langflow.graph.vertex.utils import generate_result
|
||||||
from langflow.interface.initialize import loading
|
from langflow.interface.initialize import loading
|
||||||
|
|
@ -300,9 +300,8 @@ class Vertex:
|
||||||
else:
|
else:
|
||||||
params.pop(key, None)
|
params.pop(key, None)
|
||||||
# Add _type to params
|
# Add _type to params
|
||||||
self._raw_params = params
|
|
||||||
self.params = params
|
self.params = params
|
||||||
# TODO: Hash params dict
|
self._raw_params = params
|
||||||
|
|
||||||
async def _build(self, user_id=None):
|
async def _build(self, user_id=None):
|
||||||
"""
|
"""
|
||||||
|
|
@ -344,7 +343,7 @@ class Vertex:
|
||||||
"""
|
"""
|
||||||
Iterates over each node in the params dictionary and builds it.
|
Iterates over each node in the params dictionary and builds it.
|
||||||
"""
|
"""
|
||||||
for key, value in self.params.copy().items():
|
for key, value in self._raw_params.items():
|
||||||
if self._is_node(value):
|
if self._is_node(value):
|
||||||
if value == self:
|
if value == self:
|
||||||
del self.params[key]
|
del self.params[key]
|
||||||
|
|
@ -390,7 +389,7 @@ class Vertex:
|
||||||
await self.build(requester=requester, user_id=user_id)
|
await self.build(requester=requester, user_id=user_id)
|
||||||
return self._built_object
|
return self._built_object
|
||||||
|
|
||||||
async def _build_node_and_update_params(self, key, node, user_id=None):
|
async def _build_node_and_update_params(self, key, node: "Vertex", user_id=None):
|
||||||
"""
|
"""
|
||||||
Builds a given node and updates the params dictionary accordingly.
|
Builds a given node and updates the params dictionary accordingly.
|
||||||
"""
|
"""
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue