Skip to content

Commit bdaf5a2

Browse files
authored
Update the start command to use validate_schema (#82)
Closes #77 ## By Submitting this PR I confirm: - I am familiar with the [Contributing Guidelines](https://github.com/NVIDIA/AgentIQ/blob/develop/docs/source/advanced/contributing.md). - We require that all contributors "sign-off" on their commits. This certifies that the contribution is your original work, or you have rights to submit it under the same license, or a compatible license. - Any contribution which contains commits that are not Signed-Off will not be accepted. - When the PR is ready for review, new or existing tests cover these changes. - When the PR is ready for review, the documentation is up to date with these changes. Authors: - David Gardner (https://github.com/dagardner-nv) Approvers: - Michael Demoret (https://github.com/mdemoret-nv) URL: #82
1 parent 99a9cba commit bdaf5a2

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/aiq/cli/commands/start.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@
2727
from aiq.cli.type_registry import GlobalTypeRegistry
2828
from aiq.cli.type_registry import RegisteredFrontEndInfo
2929
from aiq.data_models.config import AIQConfig
30+
from aiq.utils.data_models.schema_validator import validate_schema
3031
from aiq.utils.type_utils import DecomposedType
3132

3233
logger = logging.getLogger(__name__)
@@ -184,7 +185,7 @@ def invoke_subcommand(self,
184185
# Get the front end for the command
185186
front_end: RegisteredFrontEndInfo = self._registered_front_ends[cmd_name]
186187

187-
config = AIQConfig.model_validate(config_dict)
188+
config = validate_schema(config_dict, AIQConfig)
188189

189190
# Check that we have the right kind of front end
190191
if (not isinstance(config.general.front_end, front_end.config_type)):

0 commit comments

Comments
 (0)