oracle-ads 2.13.0__py3-none-any.whl → 2.13.1__py3-none-any.whl

ads/aqua/config/evaluation/evaluation_service_config.py

@@ -1,9 +1,8 @@
 #!/usr/bin/env python
 
-# Copyright (c) 2024 Oracle and/or its affiliates.
+# Copyright (c) 2024, 2025 Oracle and/or its affiliates.
 # Licensed under the Universal Permissive License v 1.0 as shown at https://oss.oracle.com/licenses/upl/
 
-from copy import deepcopy
 from typing import Any, Dict, List, Optional
 
 from pydantic import Field
@@ -11,139 +10,6 @@ from pydantic import Field
 from ads.aqua.config.utils.serializer import Serializable
 
 
-class ModelParamsOverrides(Serializable):
-    """Defines overrides for model parameters, including exclusions and additional inclusions."""
-
-    exclude: Optional[List[str]] = Field(default_factory=list)
-    include: Optional[Dict[str, Any]] = Field(default_factory=dict)
-
-    class Config:
-        extra = "ignore"
-
-
-class ModelParamsVersion(Serializable):
-    """Handles version-specific model parameter overrides."""
-
-    overrides: Optional[ModelParamsOverrides] = Field(
-        default_factory=ModelParamsOverrides
-    )
-
-    class Config:
-        extra = "ignore"
-
-
-class ModelParamsContainer(Serializable):
-    """Represents a container's model configuration, including tasks, defaults, and versions."""
-
-    name: Optional[str] = None
-    default: Optional[Dict[str, Any]] = Field(default_factory=dict)
-    versions: Optional[Dict[str, ModelParamsVersion]] = Field(default_factory=dict)
-
-    class Config:
-        extra = "ignore"
-
-
-class InferenceParams(Serializable):
-    """Contains inference-related parameters with defaults."""
-
-    class Config:
-        extra = "allow"
-
-
-class InferenceContainer(Serializable):
-    """Represents the inference parameters specific to a container."""
-
-    name: Optional[str] = None
-    params: Optional[Dict[str, Any]] = Field(default_factory=dict)
-
-    class Config:
-        extra = "ignore"
-
-
-class ReportParams(Serializable):
-    """Handles the report-related parameters."""
-
-    default: Optional[Dict[str, Any]] = Field(default_factory=dict)
-
-    class Config:
-        extra = "ignore"
-
-
-class InferenceParamsConfig(Serializable):
-    """Combines default inference parameters with container-specific configurations."""
-
-    default: Optional[InferenceParams] = Field(default_factory=InferenceParams)
-    containers: Optional[List[InferenceContainer]] = Field(default_factory=list)
-
-    def get_merged_params(self, container_name: str) -> InferenceParams:
-        """
-        Merges default inference params with those specific to the given container.
-
-        Parameters
-        ----------
-        container_name (str): The name of the container.
-
-        Returns
-        -------
-        InferenceParams: The merged inference parameters.
-        """
-        merged_params = self.default.to_dict()
-        for containers in self.containers:
-            if containers.name.lower() == container_name.lower():
-                merged_params.update(containers.params or {})
-                break
-        return InferenceParams(**merged_params)
-
-    class Config:
-        extra = "ignore"
-
-
-class InferenceModelParamsConfig(Serializable):
-    """Encapsulates the model parameters for different containers."""
-
-    default: Optional[Dict[str, Any]] = Field(default_factory=dict)
-    containers: Optional[List[ModelParamsContainer]] = Field(default_factory=list)
-
-    def get_merged_model_params(
-        self,
-        container_name: str,
-        version: Optional[str] = None,
-    ) -> Dict[str, Any]:
-        """
-        Gets the model parameters for a given container, version,
-        merged with the defaults.
-
-        Parameters
-        ----------
-        container_name (str): The name of the container.
-        version (Optional[str]): The specific version of the container.
-
-        Returns
-        -------
-        Dict[str, Any]: The merged model parameters.
-        """
-        params = deepcopy(self.default)
-
-        for container in self.containers:
-            if container.name.lower() == container_name.lower():
-                params.update(container.default)
-
-                if version and version in container.versions:
-                    version_overrides = container.versions[version].overrides
-                    if version_overrides:
-                        if version_overrides.include:
-                            params.update(version_overrides.include)
-                        if version_overrides.exclude:
-                            for key in version_overrides.exclude:
-                                params.pop(key, None)
-                break
-
-        return params
-
-    class Config:
-        extra = "ignore"
-
-
 class ShapeFilterConfig(Serializable):
     """Represents the filtering options for a specific shape."""
 
@@ -151,7 +17,7 @@ class ShapeFilterConfig(Serializable):
     evaluation_target: Optional[List[str]] = Field(default_factory=list)
 
     class Config:
-        extra = "ignore"
+        extra = "allow"
 
 
 class ShapeConfig(Serializable):
@@ -178,7 +44,7 @@ class MetricConfig(Serializable):
     tags: Optional[List[str]] = Field(default_factory=list)
 
     class Config:
-        extra = "ignore"
+        extra = "allow"
 
 
 class ModelParamsConfig(Serializable):
@@ -223,7 +89,7 @@ class UIConfig(Serializable):
         ]
 
     class Config:
-        extra = "ignore"
+        extra = "allow"
         protected_namespaces = ()
 
 
@@ -235,49 +101,7 @@ class EvaluationServiceConfig(Serializable):
 
     version: Optional[str] = "1.0"
     kind: Optional[str] = "evaluation_service_config"
-    report_params: Optional[ReportParams] = Field(default_factory=ReportParams)
-    inference_params: Optional[InferenceParamsConfig] = Field(
-        default_factory=InferenceParamsConfig
-    )
-    inference_model_params: Optional[InferenceModelParamsConfig] = Field(
-        default_factory=InferenceModelParamsConfig
-    )
     ui_config: Optional[UIConfig] = Field(default_factory=UIConfig)
 
-    def get_merged_inference_params(self, container_name: str) -> InferenceParams:
-        """
-        Merges default inference params with those specific to the given container.
-
-        Params
-        ------
-        container_name (str): The name of the container.
-
-        Returns
-        -------
-        InferenceParams: The merged inference parameters.
-        """
-        return self.inference_params.get_merged_params(container_name=container_name)
-
-    def get_merged_inference_model_params(
-        self,
-        container_name: str,
-        version: Optional[str] = None,
-    ) -> Dict[str, Any]:
-        """
-        Gets the model parameters for a given container, version, and task, merged with the defaults.
-
-        Parameters
-        ----------
-        container_name (str): The name of the container.
-        version (Optional[str]): The specific version of the container.
-
-        Returns
-        -------
-        Dict[str, Any]: The merged model parameters.
-        """
-        return self.inference_model_params.get_merged_model_params(
-            container_name=container_name, version=version
-        )
-
     class Config:
-        extra = "ignore"
+        extra = "allow"

ads/aqua/constants.py

@@ -3,7 +3,6 @@
 # Licensed under the Universal Permissive License v 1.0 as shown at https://oss.oracle.com/licenses/upl/
 """This module defines constants used in ads.aqua module."""
 
-UNKNOWN = ""
 UNKNOWN_VALUE = ""
 READY_TO_IMPORT_STATUS = "TRUE"
 UNKNOWN_DICT = {}

ads/aqua/evaluation/evaluation.py

@@ -40,11 +40,13 @@ from ads.aqua.common.errors import (
 from ads.aqua.common.utils import (
     extract_id_and_name_from_tag,
     fire_and_forget,
+    get_container_config,
     get_container_image,
     is_valid_ocid,
     upload_local_to_os,
 )
 from ads.aqua.config.config import get_evaluation_service_config
+from ads.aqua.config.container_config import AquaContainerConfig
 from ads.aqua.constants import (
     CONSOLE_LINK_RESOURCE_TYPE_MAPPING,
     EVALUATION_REPORT,
@@ -53,7 +55,6 @@ from ads.aqua.constants import (
     JOB_INFRASTRUCTURE_TYPE_DEFAULT_NETWORKING,
     LIFECYCLE_DETAILS_MISSING_JOBRUN,
     NB_SESSION_IDENTIFIER,
-    UNKNOWN,
 )
 from ads.aqua.evaluation.constants import (
     EVAL_TERMINATION_STATE,
@@ -75,10 +76,9 @@ from ads.aqua.evaluation.entities import (
     CreateAquaEvaluationDetails,
 )
 from ads.aqua.evaluation.errors import EVALUATION_JOB_EXIT_CODE_MESSAGE
-from ads.aqua.ui import AquaContainerConfig
 from ads.common.auth import default_signer
 from ads.common.object_storage_details import ObjectStorageDetails
-from ads.common.utils import get_console_link, get_files, get_log_links
+from ads.common.utils import UNKNOWN, get_console_link, get_files, get_log_links
 from ads.config import (
     AQUA_JOB_SUBNET_ID,
     COMPARTMENT_OCID,
@@ -192,7 +192,7 @@ class AquaEvaluationApp(AquaApp):
                         evaluation_source.runtime.to_dict()
                     )
                     inference_config = AquaContainerConfig.from_container_index_json(
-                        enable_spec=True
+                        config=get_container_config(), enable_spec=True
                     ).inference
                     for container in inference_config.values():
                         if container.name == runtime.image[: runtime.image.rfind(":")]:

ads/aqua/extension/base_handler.py

@@ -37,6 +37,10 @@ class AquaAPIhandler(APIHandler):
         except Exception:
             pass
 
+    def prepare(self, *args, **kwargs):
+        """The base class prepare is not required for Aqua"""
+        pass
+        
     @staticmethod
     def serialize(obj: Any):
         """Serialize the object.

ads/aqua/extension/model_handler.py

@@ -8,20 +8,13 @@ from urllib.parse import urlparse
 from tornado.web import HTTPError
 
 from ads.aqua.common.decorator import handle_exceptions
-from ads.aqua.common.enums import (
-    CustomInferenceContainerTypeFamily,
-)
+from ads.aqua.common.enums import CustomInferenceContainerTypeFamily
 from ads.aqua.common.errors import AquaRuntimeError, AquaValueError
-from ads.aqua.common.utils import (
-    get_hf_model_info,
-    is_valid_ocid,
-    list_hf_models,
-)
+from ads.aqua.common.utils import get_hf_model_info, is_valid_ocid, list_hf_models
 from ads.aqua.extension.base_handler import AquaAPIhandler
 from ads.aqua.extension.errors import Errors
 from ads.aqua.model import AquaModelApp
 from ads.aqua.model.entities import AquaModelSummary, HFModelSummary
-from ads.aqua.ui import ModelFormat
 
 
 class AquaModelHandler(AquaAPIhandler):
@@ -44,26 +37,24 @@ class AquaModelHandler(AquaAPIhandler):
                 raise HTTPError(
                     400, Errors.MISSING_REQUIRED_PARAMETER.format("model_format")
                 )
-            try:
-                model_format = ModelFormat(model_format.upper())
-            except ValueError as err:
-                raise AquaValueError(f"Invalid model format: {model_format}") from err
+            
+            model_format = model_format.upper()
+            
+            if os_path:
+                return self.finish(
+                    AquaModelApp.get_model_files(os_path, model_format)
+                )
+            elif model_name:
+                return self.finish(
+                    AquaModelApp.get_hf_model_files(model_name, model_format)
+                )
             else:
-                if os_path:
-                    return self.finish(
-                        AquaModelApp.get_model_files(os_path, model_format)
-                    )
-                elif model_name:
-                    return self.finish(
-                        AquaModelApp.get_hf_model_files(model_name, model_format)
-                    )
-                else:
-                    raise HTTPError(
-                        400,
-                        Errors.MISSING_ONEOF_REQUIRED_PARAMETER.format(
-                            "os_path", "model_name"
-                        ),
-                    )
+                raise HTTPError(
+                    400,
+                    Errors.MISSING_ONEOF_REQUIRED_PARAMETER.format(
+                        "os_path", "model_name"
+                    ),
+                )
         elif not model_id:
             return self.list()
 

ads/aqua/finetuning/finetuning.py

@@ -30,7 +30,6 @@ from ads.aqua.constants import (
     DEFAULT_FT_REPLICA,
     DEFAULT_FT_VALIDATION_SET_SIZE,
     JOB_INFRASTRUCTURE_TYPE_DEFAULT_NETWORKING,
-    UNKNOWN,
     UNKNOWN_DICT,
 )
 from ads.aqua.data import AquaResourceIdentifier
@@ -45,7 +44,7 @@ from ads.aqua.finetuning.entities import (
 )
 from ads.common.auth import default_signer
 from ads.common.object_storage_details import ObjectStorageDetails
-from ads.common.utils import get_console_link
+from ads.common.utils import UNKNOWN, get_console_link
 from ads.config import (
     AQUA_FINETUNING_CONTAINER_OVERRIDE_FLAG_METADATA_NAME,
     AQUA_JOB_SUBNET_ID,
@@ -592,7 +591,7 @@ class AquaFineTuningApp(AquaApp):
         Dict:
             A dict of allowed finetuning configs.
         """
-        config = self.get_config(model_id, AQUA_MODEL_FINETUNING_CONFIG)
+        config = self.get_config(model_id, AQUA_MODEL_FINETUNING_CONFIG).config
         if not config:
             logger.debug(
                 f"Fine-tuning config for custom model: {model_id} is not available. Use defaults."

ads/aqua/model/entities.py

@@ -23,7 +23,6 @@ from ads.aqua.constants import LIFECYCLE_DETAILS_MISSING_JOBRUN, UNKNOWN_VALUE
 from ads.aqua.data import AquaResourceIdentifier
 from ads.aqua.model.enums import FineTuningDefinedMetadata
 from ads.aqua.training.exceptions import exit_code_dict
-from ads.aqua.ui import ModelFormat
 from ads.common.serializer import DataClassSerializable
 from ads.common.utils import get_log_links
 from ads.model.datascience_model import DataScienceModel
@@ -46,7 +45,7 @@ class AquaFineTuneValidation(DataClassSerializable):
 @dataclass(repr=False)
 class ModelValidationResult:
     model_file: Optional[str] = None
-    model_formats: List[ModelFormat] = field(default_factory=list)
+    model_formats: List[str] = field(default_factory=list)
     telemetry_model_name: str = None
     tags: Optional[dict] = None
 
@@ -89,7 +88,7 @@ class AquaModelSummary(DataClassSerializable):
     nvidia_gpu_supported: bool = False
     arm_cpu_supported: bool = False
     model_file: Optional[str] = None
-    model_formats: List[ModelFormat] = field(default_factory=list)
+    model_formats: List[str] = field(default_factory=list)
 
 
 @dataclass(repr=False)

ads/aqua/model/model.py

@@ -19,6 +19,8 @@ from ads.aqua.common.enums import (
     CustomInferenceContainerTypeFamily,
     FineTuningContainerTypeFamily,
     InferenceContainerTypeFamily,
+    ModelFormat,
+    Platform,
     Tags,
 )
 from ads.aqua.common.errors import (
@@ -40,6 +42,7 @@ from ads.aqua.common.utils import (
     read_file,
     upload_folder,
 )
+from ads.aqua.config.container_config import AquaContainerConfig
 from ads.aqua.constants import (
     AQUA_MODEL_ARTIFACT_CONFIG,
     AQUA_MODEL_ARTIFACT_CONFIG_MODEL_NAME,
@@ -56,7 +59,6 @@ from ads.aqua.constants import (
     READY_TO_IMPORT_STATUS,
     TRAINING_METRICS_FINAL,
     TRINING_METRICS,
-    UNKNOWN,
     VALIDATION_METRICS,
     VALIDATION_METRICS_FINAL,
 )
@@ -73,13 +75,11 @@ from ads.aqua.model.entities import (
     AquaModelLicense,
     AquaModelSummary,
     ImportModelDetails,
-    ModelFormat,
     ModelValidationResult,
 )
-from ads.aqua.ui import AquaContainerConfig, AquaContainerConfigItem
 from ads.common.auth import default_signer
 from ads.common.oci_resource import SEARCH_TYPE, OCIResource
-from ads.common.utils import get_console_link
+from ads.common.utils import UNKNOWN, get_console_link
 from ads.config import (
     AQUA_DEPLOYMENT_CONTAINER_CMD_VAR_METADATA_NAME,
     AQUA_DEPLOYMENT_CONTAINER_METADATA_NAME,
@@ -585,7 +585,7 @@ class AquaModelApp(AquaApp):
         """
         config = self.get_config(
             model_id, AQUA_MODEL_TOKENIZER_CONFIG, ConfigFolder.ARTIFACT
-        )
+        ).config
         if not config:
             logger.debug(f"Tokenizer config for model: {model_id} is not available.")
         return config
@@ -667,28 +667,24 @@ class AquaModelApp(AquaApp):
         except Exception:
             model_file = UNKNOWN
 
-        inference_containers = AquaContainerConfig.from_container_index_json().inference
+        inference_containers = AquaContainerConfig.from_container_index_json(
+            config=get_container_config()
+        ).inference
 
         model_formats_str = freeform_tags.get(
-            Tags.MODEL_FORMAT, ModelFormat.SAFETENSORS.value
+            Tags.MODEL_FORMAT, ModelFormat.SAFETENSORS
         ).upper()
-        model_formats = [
-            ModelFormat[model_format] for model_format in model_formats_str.split(",")
-        ]
+        model_formats = model_formats_str.split(",")
 
-        supported_platform: Set[AquaContainerConfigItem.Platform] = set()
+        supported_platform: Set[str] = set()
 
         for container in inference_containers.values():
             for model_format in model_formats:
                 if model_format in container.model_formats:
                     supported_platform.update(container.platforms)
 
-        nvidia_gpu_supported = (
-            AquaContainerConfigItem.Platform.NVIDIA_GPU in supported_platform
-        )
-        arm_cpu_supported = (
-            AquaContainerConfigItem.Platform.ARM_CPU in supported_platform
-        )
+        nvidia_gpu_supported = Platform.NVIDIA_GPU in supported_platform
+        arm_cpu_supported = Platform.ARM_CPU in supported_platform
 
         return {
             "compartment_id": model.compartment_id,
@@ -898,8 +894,7 @@ class AquaModelApp(AquaApp):
             tags.update(
                 {
                     Tags.MODEL_FORMAT: ",".join(
-                        model_format.value
-                        for model_format in validation_result.model_formats
+                        model_format for model_format in validation_result.model_formats
                     )
                 }
             )
@@ -936,9 +931,9 @@ class AquaModelApp(AquaApp):
                     category="Other",
                 )
 
-            inference_containers = (
-                AquaContainerConfig.from_container_index_json().inference
-            )
+            inference_containers = AquaContainerConfig.from_container_index_json(
+                config=get_container_config()
+            ).inference
             smc_container_set = {
                 container.family for container in inference_containers.values()
             }
@@ -1013,13 +1008,13 @@ class AquaModelApp(AquaApp):
         return model
 
     @staticmethod
-    def get_model_files(os_path: str, model_format: ModelFormat) -> List[str]:
+    def get_model_files(os_path: str, model_format: str) -> List[str]:
         """
         Get a list of model files based on the given OS path and model format.
 
         Args:
             os_path (str): The OS path where the model files are located.
-            model_format (ModelFormat): The format of the model files.
+            model_format (str): The format of the model files.
 
         Returns:
             List[str]: A list of model file names.
@@ -1059,13 +1054,13 @@ class AquaModelApp(AquaApp):
         return model_files
 
     @staticmethod
-    def get_hf_model_files(model_name: str, model_format: ModelFormat) -> List[str]:
+    def get_hf_model_files(model_name: str, model_format: str) -> List[str]:
         """
         Get a list of model files based on the given OS path and model format.
 
         Args:
             model_name (str): The huggingface model name.
-            model_format (ModelFormat): The format of the model files.
+            model_format (str): The format of the model files.
 
         Returns:
             List[str]: A list of model file names.
@@ -1097,7 +1092,7 @@ class AquaModelApp(AquaApp):
                 and model_sibling.rfilename == AQUA_MODEL_ARTIFACT_CONFIG
             ):
                 model_files.append(model_sibling.rfilename)
-            if extension == model_format.value:
+            if extension == model_format:
                 model_files.append(model_sibling.rfilename)
 
         logger.debug(
@@ -1152,8 +1147,8 @@ class AquaModelApp(AquaApp):
 
         if not (safetensors_model_files or gguf_model_files):
             raise AquaRuntimeError(
-                f"The model {model_name} does not contain either {ModelFormat.SAFETENSORS.value} "
-                f"or {ModelFormat.GGUF.value} files in {import_model_details.os_path} or Hugging Face repository. "
+                f"The model {model_name} does not contain either {ModelFormat.SAFETENSORS} "
+                f"or {ModelFormat.GGUF} files in {import_model_details.os_path} or Hugging Face repository. "
                 f"Please check if the path is correct or the model artifacts are available at this location."
             )
 
@@ -1261,7 +1256,7 @@ class AquaModelApp(AquaApp):
             ):
                 raise AquaRuntimeError(
                     f"The model {model_name} does not contain {AQUA_MODEL_ARTIFACT_CONFIG} file as required "
-                    f"by {ModelFormat.SAFETENSORS.value} format model."
+                    f"by {ModelFormat.SAFETENSORS} format model."
                     f" Please check if the model name is correct in Hugging Face repository."
                 )
             validation_result.telemetry_model_name = model_name

ads/aqua/modeldeployment/deployment.py

@@ -7,10 +7,7 @@ from typing import Dict, List, Optional, Union
 
 from ads.aqua.app import AquaApp, logger
 from ads.aqua.common.entities import ContainerSpec
-from ads.aqua.common.enums import (
-    InferenceContainerTypeFamily,
-    Tags,
-)
+from ads.aqua.common.enums import InferenceContainerTypeFamily, ModelFormat, Tags
 from ads.aqua.common.errors import AquaRuntimeError, AquaValueError
 from ads.aqua.common.utils import (
     get_combined_params,
@@ -30,19 +27,14 @@ from ads.aqua.constants import (
     AQUA_MODEL_TYPE_CUSTOM,
     AQUA_MODEL_TYPE_SERVICE,
     MODEL_BY_REFERENCE_OSS_PATH_KEY,
-    UNKNOWN,
     UNKNOWN_DICT,
 )
 from ads.aqua.data import AquaResourceIdentifier
 from ads.aqua.finetuning.finetuning import FineTuneCustomMetadata
 from ads.aqua.model import AquaModelApp
-from ads.aqua.modeldeployment.entities import (
-    AquaDeployment,
-    AquaDeploymentDetail,
-)
-from ads.aqua.ui import ModelFormat
+from ads.aqua.modeldeployment.entities import AquaDeployment, AquaDeploymentDetail
 from ads.common.object_storage_details import ObjectStorageDetails
-from ads.common.utils import get_log_links
+from ads.common.utils import UNKNOWN, get_log_links
 from ads.config import (
     AQUA_DEPLOYMENT_CONTAINER_CMD_VAR_METADATA_NAME,
     AQUA_DEPLOYMENT_CONTAINER_METADATA_NAME,
@@ -293,13 +285,13 @@ class AquaDeploymentApp(AquaApp):
             )
 
         model_formats_str = aqua_model.freeform_tags.get(
-            Tags.MODEL_FORMAT, ModelFormat.SAFETENSORS.value
+            Tags.MODEL_FORMAT, ModelFormat.SAFETENSORS
         ).upper()
         model_format = model_formats_str.split(",")
 
         # Figure out a better way to handle this in future release
         if (
-            ModelFormat.GGUF.value in model_format
+            ModelFormat.GGUF in model_format
             and container_type_key.lower()
             == InferenceContainerTypeFamily.AQUA_LLAMA_CPP_CONTAINER_FAMILY
         ):
@@ -661,7 +653,7 @@ class AquaDeploymentApp(AquaApp):
         Dict:
             A dict of allowed deployment configs.
         """
-        config = self.get_config(model_id, AQUA_MODEL_DEPLOYMENT_CONFIG)
+        config = self.get_config(model_id, AQUA_MODEL_DEPLOYMENT_CONFIG).config
         if not config:
             logger.debug(
                 f"Deployment config for custom model: {model_id} is not available. Use defaults."

ads/aqua/modeldeployment/entities.py

@@ -11,10 +11,10 @@ from oci.data_science.models import (
 )
 
 from ads.aqua.common.enums import Tags
-from ads.aqua.constants import UNKNOWN, UNKNOWN_DICT
+from ads.aqua.constants import UNKNOWN_DICT
 from ads.aqua.data import AquaResourceIdentifier
 from ads.common.serializer import DataClassSerializable
-from ads.common.utils import get_console_link
+from ads.common.utils import UNKNOWN, get_console_link
 
 
 @dataclass

ads/aqua/server/__init__.py

@@ -0,0 +1,4 @@
+#!/usr/bin/env python
+
+# Copyright (c) 2025 Oracle and/or its affiliates.
+# Licensed under the Universal Permissive License v 1.0 as shown at https://oss.oracle.com/licenses/upl/

ads/aqua/server/__main__.py

@@ -0,0 +1,24 @@
+#!/usr/bin/env python
+
+# Copyright (c) 2025 Oracle and/or its affiliates.
+# Licensed under the Universal Permissive License v 1.0 as shown at https://oss.oracle.com/licenses/upl/
+
+import os
+from logging import getLogger
+
+from dotenv import load_dotenv
+
+from ads.aqua.server.app import start_server
+
+logger = getLogger(__name__)
+config_location = os.path.join(os.getcwd(), ".env")
+if os.path.exists(config_location):
+    logger.info(f"Loading environment variables from {config_location}")
+    load_dotenv(dotenv_path=config_location)
+    logger.info("Environment variables loaded successfully")
+else:
+    logger.warning(
+        f"{config_location} not found. Consider using `.env` file to setup default environment variables"
+    )
+
+start_server()