oracle-ads 2.13.8__py3-none-any.whl → 2.13.9rc0__py3-none-any.whl

ads/aqua/extension/deployment_ws_msg_handler.py

@@ -1,54 +0,0 @@
-#!/usr/bin/env python
-
-# Copyright (c) 2024 Oracle and/or its affiliates.
-# Licensed under the Universal Permissive License v 1.0 as shown at https://oss.oracle.com/licenses/upl/
-
-import json
-from typing import List, Union
-
-from ads.aqua.common.decorator import handle_exceptions
-from ads.aqua.extension.aqua_ws_msg_handler import AquaWSMsgHandler
-from ads.aqua.extension.models.ws_models import (
-    ListDeploymentResponse,
-    ModelDeploymentDetailsResponse,
-    RequestResponseType,
-)
-from ads.aqua.modeldeployment import AquaDeploymentApp
-from ads.config import COMPARTMENT_OCID
-
-
-class AquaDeploymentWSMsgHandler(AquaWSMsgHandler):
-    def __init__(self, message: Union[str, bytes]):
-        super().__init__(message)
-
-    @staticmethod
-    def get_message_types() -> List[RequestResponseType]:
-        return [
-            RequestResponseType.ListDeployments,
-            RequestResponseType.DeploymentDetails,
-        ]
-
-    @handle_exceptions
-    def process(self) -> Union[ListDeploymentResponse, ModelDeploymentDetailsResponse]:
-        request = json.loads(self.message)
-        if request.get("kind") == "ListDeployments":
-            deployment_list = AquaDeploymentApp().list(
-                compartment_id=request.get("compartment_id") or COMPARTMENT_OCID,
-                project_id=request.get("project_id"),
-            )
-            response = ListDeploymentResponse(
-                message_id=request.get("message_id"),
-                kind=RequestResponseType.ListDeployments,
-                data=deployment_list,
-            )
-            return response
-        elif request.get("kind") == "DeploymentDetails":
-            deployment_details = AquaDeploymentApp().get(
-                request.get("model_deployment_id")
-            )
-            response = ModelDeploymentDetailsResponse(
-                message_id=request.get("message_id"),
-                kind=RequestResponseType.DeploymentDetails,
-                data=deployment_details,
-            )
-            return response

ads/aqua/extension/errors.py

@@ -1,30 +0,0 @@
-#!/usr/bin/env python
-# Copyright (c) 2024 Oracle and/or its affiliates.
-# Licensed under the Universal Permissive License v 1.0 as shown at https://oss.oracle.com/licenses/upl/
-import uuid
-from typing import Any, Dict, List, Optional
-
-from pydantic import Field
-
-from ads.aqua.config.utils.serializer import Serializable
-
-from ads.aqua.constants import (
-    AQUA_TROUBLESHOOTING_LINK
-)
-
-class Errors(str):
-    INVALID_INPUT_DATA_FORMAT = "Invalid format of input data."
-    NO_INPUT_DATA = "No input data provided."
-    MISSING_REQUIRED_PARAMETER = "Missing required parameter: '{}'"
-    MISSING_ONEOF_REQUIRED_PARAMETER = "Either '{}' or '{}' is required."
-    INVALID_VALUE_OF_PARAMETER = "Invalid value of parameter: '{}'"
-
-class ReplyDetails(Serializable):
-    """Structured reply to be returned to the client."""
-    status: int
-    troubleshooting_tips: str = Field(f"For general tips on troubleshooting: {AQUA_TROUBLESHOOTING_LINK}",
-                                      description="GitHub Link for troubleshooting documentation")
-    message: str = Field("Unknown HTTP Error.", description="GitHub Link for troubleshooting documentation")
-    service_payload: Optional[Dict[str, Any]] = Field(default_factory=dict)
-    reason: str = Field("Unknown error", description="Reason for Error")
-    request_id: str = Field(str(uuid.uuid4()), description="Unique ID for tracking the error.")

ads/aqua/extension/evaluation_handler.py

@@ -1,129 +0,0 @@
-#!/usr/bin/env python
-# Copyright (c) 2024 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 Optional
-from urllib.parse import urlparse
-
-from tornado.web import HTTPError
-
-from ads.aqua.common.decorator import handle_exceptions
-from ads.aqua.evaluation import AquaEvaluationApp
-from ads.aqua.evaluation.entities import CreateAquaEvaluationDetails
-from ads.aqua.extension.base_handler import AquaAPIhandler
-from ads.aqua.extension.errors import Errors
-from ads.config import COMPARTMENT_OCID
-
-
-class AquaEvaluationHandler(AquaAPIhandler):
-    """Handler for Aqua Model Evaluation REST APIs."""
-
-    @handle_exceptions
-    def get(self, eval_id=""):
-        """Handle GET request."""
-        url_parse = urlparse(self.request.path)
-        paths = url_parse.path.strip("/")
-        if paths.startswith("aqua/evaluation/metrics"):
-            return self.get_default_metrics()
-        if not eval_id:
-            return self.list()
-        return self.read(eval_id)
-
-    @handle_exceptions
-    def post(self, *args, **kwargs):  # noqa
-        """Handles post request for the evaluation APIs
-
-        Raises
-        ------
-        HTTPError
-            Raises HTTPError if inputs are missing or are invalid.
-        """
-        try:
-            input_data = self.get_json_body()
-        except Exception as ex:
-            raise HTTPError(400, Errors.INVALID_INPUT_DATA_FORMAT) from ex
-
-        if not input_data:
-            raise HTTPError(400, Errors.NO_INPUT_DATA)
-
-        self.finish(
-            # TODO: decide what other kwargs will be needed for create aqua evaluation.
-            AquaEvaluationApp().create(
-                create_aqua_evaluation_details=(
-                    CreateAquaEvaluationDetails(**input_data)
-                )
-            )
-        )
-
-    @handle_exceptions
-    def put(self, eval_id):
-        """Handles PUT request for the evaluation APIs"""
-        eval_id = eval_id.split("/")[0]
-        return self.finish(AquaEvaluationApp().cancel(eval_id))
-
-    @handle_exceptions
-    def delete(self, eval_id):
-        return self.finish(AquaEvaluationApp().delete(eval_id))
-
-    def read(self, eval_id):
-        """Read the information of an Aqua model."""
-        return self.finish(AquaEvaluationApp().get(eval_id))
-
-    def list(self):
-        """List Aqua models."""
-        compartment_id = self.get_argument("compartment_id", default=COMPARTMENT_OCID)
-        return self.finish(AquaEvaluationApp().list(compartment_id))
-
-    def get_default_metrics(self):
-        """Lists supported metrics for evaluation."""
-        return self.finish(AquaEvaluationApp().get_supported_metrics())
-
-
-class AquaEvaluationStatusHandler(AquaAPIhandler):
-    """Handler for Aqua Evaluation status REST APIs."""
-
-    @handle_exceptions
-    def get(self, eval_id):
-        """Handle GET request."""
-        eval_id = eval_id.split("/")[0]
-        return self.finish(AquaEvaluationApp().get_status(eval_id))
-
-
-class AquaEvaluationReportHandler(AquaAPIhandler):
-    """Handler for Aqua Evaluation report REST APIs."""
-
-    @handle_exceptions
-    def get(self, eval_id):
-        """Handle GET request."""
-        eval_id = eval_id.split("/")[0]
-        return self.finish(AquaEvaluationApp().download_report(eval_id))
-
-
-class AquaEvaluationMetricsHandler(AquaAPIhandler):
-    """Handler for Aqua Evaluation metrics REST APIs."""
-
-    @handle_exceptions
-    def get(self, eval_id):
-        """Handle GET request."""
-        eval_id = eval_id.split("/")[0]
-        return self.finish(AquaEvaluationApp().load_metrics(eval_id))
-
-
-class AquaEvaluationConfigHandler(AquaAPIhandler):
-    """Handler for Aqua Evaluation Config REST APIs."""
-
-    @handle_exceptions
-    def get(self, container: Optional[str] = None, **kwargs):  # noqa
-        """Handle GET request."""
-
-        return self.finish(AquaEvaluationApp().load_evaluation_config(container))
-
-
-__handlers__ = [
-    ("evaluation/config/?([^/]*)", AquaEvaluationConfigHandler),
-    ("evaluation/?([^/]*)", AquaEvaluationHandler),
-    ("evaluation/?([^/]*/report)", AquaEvaluationReportHandler),
-    ("evaluation/?([^/]*/metrics)", AquaEvaluationMetricsHandler),
-    ("evaluation/?([^/]*/status)", AquaEvaluationStatusHandler),
-    ("evaluation/?([^/]*/cancel)", AquaEvaluationHandler),
-]

ads/aqua/extension/evaluation_ws_msg_handler.py

@@ -1,61 +0,0 @@
-#!/usr/bin/env python
-
-# Copyright (c) 2024 Oracle and/or its affiliates.
-# Licensed under the Universal Permissive License v 1.0 as shown at https://oss.oracle.com/licenses/upl/
-
-import json
-from typing import List, Union
-
-from ads.aqua.common.decorator import handle_exceptions
-from ads.aqua.evaluation import AquaEvaluationApp
-from ads.aqua.extension.aqua_ws_msg_handler import AquaWSMsgHandler
-from ads.aqua.extension.models.ws_models import (
-    EvaluationDetailsResponse,
-    ListEvaluationsResponse,
-    RequestResponseType,
-)
-from ads.config import COMPARTMENT_OCID
-
-
-class AquaEvaluationWSMsgHandler(AquaWSMsgHandler):
-    @staticmethod
-    def get_message_types() -> List[RequestResponseType]:
-        return [
-            RequestResponseType.ListEvaluations,
-            RequestResponseType.EvaluationDetails,
-        ]
-
-    def __init__(self, message: Union[str, bytes]):
-        super().__init__(message)
-
-    @handle_exceptions
-    def process(self) -> Union[ListEvaluationsResponse, EvaluationDetailsResponse]:
-        request = json.loads(self.message)
-        if request["kind"] == "ListEvaluations":
-            return self.list_evaluations(request)
-        if request["kind"] == "EvaluationDetails":
-            return self.evaluation_details(request)
-
-    @staticmethod
-    def list_evaluations(request) -> ListEvaluationsResponse:
-        eval_list = AquaEvaluationApp().list(
-            request.get("compartment_id") or COMPARTMENT_OCID
-        )
-        response = ListEvaluationsResponse(
-            message_id=request["message_id"],
-            kind=RequestResponseType.ListEvaluations,
-            data=eval_list,
-        )
-        return response
-
-    @staticmethod
-    def evaluation_details(request) -> EvaluationDetailsResponse:
-        evaluation_details = AquaEvaluationApp().get(
-            eval_id=request.get("evaluation_id")
-        )
-        response = EvaluationDetailsResponse(
-            message_id=request.get("message_id"),
-            kind=RequestResponseType.EvaluationDetails,
-            data=evaluation_details,
-        )
-        return response

ads/aqua/extension/finetune_handler.py

@@ -1,96 +0,0 @@
-#!/usr/bin/env python
-# 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 urllib.parse import urlparse
-
-from tornado.web import HTTPError
-
-from ads.aqua.common.decorator import handle_exceptions
-from ads.aqua.extension.base_handler import AquaAPIhandler
-from ads.aqua.extension.errors import Errors
-from ads.aqua.finetuning import AquaFineTuningApp
-
-
-class AquaFineTuneHandler(AquaAPIhandler):
-    """Handler for Aqua fine-tuning job REST APIs."""
-
-    @handle_exceptions
-    def get(self, id=""):
-        """Handle GET request."""
-        url_parse = urlparse(self.request.path)
-        paths = url_parse.path.strip("/")
-        if paths.startswith("aqua/finetuning/config"):
-            if not id:
-                raise HTTPError(
-                    400, f"The request {self.request.path} requires model id."
-                )
-            return self.get_finetuning_config(id)
-        else:
-            raise HTTPError(400, f"The request {self.request.path} is invalid.")
-
-    @handle_exceptions
-    def post(self, *args, **kwargs):  # noqa: ARG002
-        """Handles post request for the fine-tuning API
-
-        Raises
-        ------
-        HTTPError
-            Raises HTTPError if inputs are missing or are invalid.
-        """
-        try:
-            input_data = self.get_json_body()
-        except Exception as ex:
-            raise HTTPError(400, Errors.INVALID_INPUT_DATA_FORMAT) from ex
-
-        if not input_data:
-            raise HTTPError(400, Errors.NO_INPUT_DATA)
-
-        self.finish(AquaFineTuningApp().create(**input_data))
-
-    def get_finetuning_config(self, model_id):
-        """Gets the finetuning config for Aqua model."""
-        return self.finish(AquaFineTuningApp().get_finetuning_config(model_id=model_id))
-
-
-class AquaFineTuneParamsHandler(AquaAPIhandler):
-    """Handler for Aqua finetuning params REST APIs."""
-
-    @handle_exceptions
-    def get(self, model_id):
-        """Handle GET request."""
-        return self.finish(
-            AquaFineTuningApp().get_finetuning_default_params(model_id=model_id)
-        )
-
-    @handle_exceptions
-    def post(self, *args, **kwargs):  # noqa: ARG002
-        """Handles post request for the finetuning param handler API.
-
-        Raises
-        ------
-        HTTPError
-            Raises HTTPError if inputs are missing or are invalid.
-        """
-        try:
-            input_data = self.get_json_body()
-        except Exception as ex:
-            raise HTTPError(400, Errors.INVALID_INPUT_DATA_FORMAT) from ex
-
-        if not input_data:
-            raise HTTPError(400, Errors.NO_INPUT_DATA)
-
-        params = input_data.get("params", None)
-        return self.finish(
-            AquaFineTuningApp().validate_finetuning_params(
-                params=params,
-            )
-        )
-
-
-__handlers__ = [
-    ("finetuning/?([^/]*)/params", AquaFineTuneParamsHandler),
-    ("finetuning/?([^/]*)", AquaFineTuneHandler),
-    ("finetuning/config/?([^/]*)", AquaFineTuneHandler),
-]