aiqtoolkit 1.2.0a20250523__py3-none-any.whl → 1.2.0a20250525__py3-none-any.whl

aiq/front_ends/fastapi/fastapi_front_end_config.py

@@ -16,25 +16,66 @@
 import logging
 import typing
 from datetime import datetime
+from pathlib import Path
 
 from pydantic import BaseModel
 from pydantic import Field
+from pydantic import field_validator
 
 from aiq.data_models.front_end import FrontEndBaseConfig
 from aiq.data_models.step_adaptor import StepAdaptorConfig
 
 logger = logging.getLogger(__name__)
 
+YAML_EXTENSIONS = (".yaml", ".yml")
+
 
 class AIQEvaluateRequest(BaseModel):
     """Request model for the evaluate endpoint."""
     config_file: str = Field(description="Path to the configuration file for evaluation")
     job_id: str | None = Field(default=None, description="Unique identifier for the evaluation job")
-    reps: int = Field(default=1, description="Number of repetitions for the evaluation, defaults to 1")
+    reps: int = Field(default=1, gt=0, description="Number of repetitions for the evaluation, defaults to 1")
     expiry_seconds: int = Field(
         default=3600,
+        gt=0,
         description="Optional time (in seconds) before the job expires. Clamped between 600 (10 min) and 86400 (24h).")
 
+    @field_validator('job_id', mode='after')
+    @classmethod
+    def validate_job_id(cls, job_id: str):
+        job_id = job_id.strip()
+        job_id_path = Path(job_id)
+        if len(job_id_path.parts) > 1 or job_id_path.resolve().name != job_id:
+            raise ValueError(
+                f"Job ID '{job_id}' contains invalid characters. Only alphanumeric characters and underscores are"
+                " allowed.")
+
+        if job_id_path.is_reserved():
+            # reserved names is Windows specific
+            raise ValueError(f"Job ID '{job_id}' is a reserved name. Please choose a different name.")
+
+        return job_id
+
+    @field_validator('config_file', mode='after')
+    @classmethod
+    def validate_config_file(cls, config_file: str):
+        config_file = config_file.strip()
+        config_file_path = Path(config_file).resolve()
+
+        # Ensure the config file is a YAML file
+        if config_file_path.suffix.lower() not in YAML_EXTENSIONS:
+            raise ValueError(f"Config file '{config_file}' must be a YAML file with one of the following extensions: "
+                             f"{', '.join(YAML_EXTENSIONS)}")
+
+        if config_file_path.is_reserved():
+            # reserved names is Windows specific
+            raise ValueError(f"Config file '{config_file}' is a reserved name. Please choose a different name.")
+
+        if not config_file_path.exists():
+            raise ValueError(f"Config file '{config_file}' does not exist. Please provide a valid path.")
+
+        return config_file
+
 
 class AIQEvaluateResponse(BaseModel):
     """Response model for the evaluate endpoint."""

{aiqtoolkit-1.2.0a20250523.dist-info → aiqtoolkit-1.2.0a20250525.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aiqtoolkit
-Version: 1.2.0a20250523
+Version: 1.2.0a20250525
 Summary: NVIDIA Agent Intelligence toolkit
 Author: NVIDIA Corporation
 Maintainer: NVIDIA Corporation

{aiqtoolkit-1.2.0a20250523.dist-info → aiqtoolkit-1.2.0a20250525.dist-info}/RECORD

@@ -141,7 +141,7 @@ aiq/front_ends/console/console_front_end_plugin.py,sha256=CzadUoHmzrHC_MWn2Fkgh_
 aiq/front_ends/console/register.py,sha256=a84M0jWUFTgOQVyrUiS7UJcxx84i1zhCb1yRkjhapiQ,1159
 aiq/front_ends/cron/__init__.py,sha256=Xs1JQ16L9btwreh4pdGKwskffAw1YFO48jKrU4ib_7c,685
 aiq/front_ends/fastapi/__init__.py,sha256=Xs1JQ16L9btwreh4pdGKwskffAw1YFO48jKrU4ib_7c,685
-aiq/front_ends/fastapi/fastapi_front_end_config.py,sha256=Ni0u9T1SkDZWRyn1EEYTePaCxhx9FYQpMeNnefZN014,7085
+aiq/front_ends/fastapi/fastapi_front_end_config.py,sha256=TaRppa8h7_noLq_TvLbOOO60vE8ccab90jELD-nVLDE,8742
 aiq/front_ends/fastapi/fastapi_front_end_plugin.py,sha256=NdE5c4pS5sMYopI3PUaE397K8HF-GuZoOmpAi1ReDRk,4002
 aiq/front_ends/fastapi/fastapi_front_end_plugin_worker.py,sha256=UJ4u8d-auPF8Dn-0w6UEExEGe5E8kkIkmTcD9yYzj9Q,27333
 aiq/front_ends/fastapi/intermediate_steps_subscriber.py,sha256=2Y3ZXfiu8d85qJRWbT3-ex6iFDuXO6TnNtjQ6Cjl-tc,3131
@@ -307,10 +307,10 @@ aiq/utils/reactive/base/observer_base.py,sha256=UAlyAY_ky4q2t0P81RVFo2Bs_R7z5Nde
 aiq/utils/reactive/base/subject_base.py,sha256=Ed-AC6P7cT3qkW1EXjzbd5M9WpVoeN_9KCe3OM3FLU4,2521
 aiq/utils/settings/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 aiq/utils/settings/global_settings.py,sha256=U9TCLdoZsKq5qOVGjREipGVv9e-FlStzqy5zv82_VYk,7454
-aiqtoolkit-1.2.0a20250523.dist-info/licenses/LICENSE-3rd-party.txt,sha256=8o7aySJa9CBvFshPcsRdJbczzdNyDGJ8b0J67WRUQ2k,183936
-aiqtoolkit-1.2.0a20250523.dist-info/licenses/LICENSE.md,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
-aiqtoolkit-1.2.0a20250523.dist-info/METADATA,sha256=PnYexmTipPV8DjDEDnEl47OFfd5gZSXmzdGhIRBARMQ,20174
-aiqtoolkit-1.2.0a20250523.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
-aiqtoolkit-1.2.0a20250523.dist-info/entry_points.txt,sha256=gRlPfR5g21t328WNEQ4CcEz80S1sJNS8A7rMDYnzl4A,452
-aiqtoolkit-1.2.0a20250523.dist-info/top_level.txt,sha256=fo7AzYcNhZ_tRWrhGumtxwnxMew4xrT1iwouDy_f0Kc,4
-aiqtoolkit-1.2.0a20250523.dist-info/RECORD,,
+aiqtoolkit-1.2.0a20250525.dist-info/licenses/LICENSE-3rd-party.txt,sha256=8o7aySJa9CBvFshPcsRdJbczzdNyDGJ8b0J67WRUQ2k,183936
+aiqtoolkit-1.2.0a20250525.dist-info/licenses/LICENSE.md,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+aiqtoolkit-1.2.0a20250525.dist-info/METADATA,sha256=VtrrrJKL0ZyCGxF6LN-DCjzOMOUl1YcmDH0WtWGgzMc,20174
+aiqtoolkit-1.2.0a20250525.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
+aiqtoolkit-1.2.0a20250525.dist-info/entry_points.txt,sha256=gRlPfR5g21t328WNEQ4CcEz80S1sJNS8A7rMDYnzl4A,452
+aiqtoolkit-1.2.0a20250525.dist-info/top_level.txt,sha256=fo7AzYcNhZ_tRWrhGumtxwnxMew4xrT1iwouDy_f0Kc,4
+aiqtoolkit-1.2.0a20250525.dist-info/RECORD,,