forecasting invoke bug fix

Author: agrimk

ads/aqua/common/enums.py

@@ -1,5 +1,5 @@
 #!/usr/bin/env python
-# Copyright (c) 2024, 2025 Oracle and/or its affiliates.
+# Copyright (c) 2024, 2026 Oracle and/or its affiliates.
 # Licensed under the Universal Permissive License v 1.0 as shown at https://oss.oracle.com/licenses/upl/
 
 from typing import Dict, List
@@ -25,6 +25,7 @@ class PredictEndpoints(ExtendedEnum):
     TEXT_COMPLETIONS_ENDPOINT = "/v1/completions"
     EMBEDDING_ENDPOINT = "/v1/embedding"
     RESPONSES = "/v1/responses"
+    FORECAST = "v1/forecast"
 
 
 class Tags(ExtendedEnum):
@@ -47,6 +48,7 @@ class Tags(ExtendedEnum):
     MULTIMODEL_TYPE_TAG = "aqua_multimodel"
     STACKED_MODEL_TYPE_TAG = "aqua_stacked_model"
     AQUA_FINE_TUNE_MODEL_VERSION = "fine_tune_model_version"
+    MODEL_DEPLOY_PREDICT_ENDPOINT = "model_deploy_predict_endpoint"
 
 
 class InferenceContainerType(ExtendedEnum):

ads/aqua/modeldeployment/deployment.py

@@ -21,7 +21,12 @@
     ComputeShapeSummary,
     ContainerPath,
 )
-from ads.aqua.common.enums import InferenceContainerTypeFamily, ModelFormat, Tags
+from ads.aqua.common.enums import (
+    InferenceContainerTypeFamily,
+    ModelFormat,
+    PredictEndpoints,
+    Tags,
+)
 from ads.aqua.common.errors import AquaRuntimeError, AquaValueError
 from ads.aqua.common.utils import (
     DEFINED_METADATA_TO_FILE_MAP,
@@ -274,6 +279,11 @@ def create(
                 create_deployment_details.env_var.update(
                     {Tags.TASK.upper(): ModelTask.TIME_SERIES_FORECASTING}
                 )
+                create_deployment_details.env_var.update(
+                    {
+                        Tags.MODEL_DEPLOY_PREDICT_ENDPOINT.upper(): PredictEndpoints.FORECAST
+                    }
+                )
             return self._create(
                 aqua_model=aqua_model,
                 create_deployment_details=create_deployment_details,