Add session_id parameter to vertex.build()

This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-01 14:01:20 -03:00
commit 652de333db

View file

@ -63,7 +63,7 @@ async def build_vertex(
start_time = time.perf_counter() start_time = time.perf_counter()
try: try:
if isinstance(vertex, StatelessVertex) or not vertex._built: if isinstance(vertex, StatelessVertex) or not vertex._built:
await vertex.build(user_id=None) await vertex.build(user_id=None, session_id=sid)
params = vertex._built_object_repr() params = vertex._built_object_repr()
valid = True valid = True
result_dict = vertex.get_built_result() result_dict = vertex.get_built_result()