refactor: Update Vertex class to handle field initialization for list types

This commit is contained in:
ogabrielluiz 2024-06-18 18:36:32 -03:00
commit 15aa6031b8

View file

@ -2,10 +2,10 @@ import ast
import asyncio import asyncio
import inspect import inspect
import os import os
import traceback
import types import types
from enum import Enum from enum import Enum
from typing import TYPE_CHECKING, Any, AsyncIterator, Callable, Dict, Iterator, List, Mapping, Optional from typing import TYPE_CHECKING, Any, AsyncIterator, Callable, Dict, Iterator, List, Mapping, Optional
import traceback
from loguru import logger from loguru import logger
@ -313,6 +313,11 @@ class Vertex:
"Setting to None." "Setting to None."
) )
params[field_name] = None params[field_name] = None
else:
if field["list"]:
params[field_name] = []
else:
params[field_name] = None
elif field.get("type") in DIRECT_TYPES and params.get(field_name) is None: elif field.get("type") in DIRECT_TYPES and params.get(field_name) is None:
val = field.get("value") val = field.get("value")
@ -619,7 +624,7 @@ class Vertex:
error_detail = f"Error: {exc}\nTraceback: {tb}" error_detail = f"Error: {exc}\nTraceback: {tb}"
logger.exception(error_detail) logger.exception(error_detail)
raise ValueError(f"Error building Component {self.display_name}: {str(exc)}\n{error_detail}") from exc raise ValueError(f"Error building Component {self.display_name}: {error_detail}") from exc
def _update_built_object_and_artifacts(self, result): def _update_built_object_and_artifacts(self, result):
""" """