support to launch Aqua services without jupyterlab (#1075)

Co-authored-by: Allen Hosler <[email protected]>

Author: mayoor

Diff for: ads/aqua/extension/base_handler.py

@@ -37,6 +37,10 @@ def __init__(
         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.

Diff for: 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/

Diff for: 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()

Diff for: ads/aqua/server/app.py

@@ -0,0 +1,47 @@
+#!/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
+
+import tornado.ioloop
+import tornado.web
+
+from ads.aqua.extension import __handlers__
+
+logger = getLogger(__name__)
+AQUA_PORT = "AQUA_PORT"
+AQUA_HOST = "AQUA_HOST"
+AQUA_PROCESS_COUNT = "AQUA_PROCESS_COUNT"
+AQUA_CORS_ENABLE = "AQUA_CORS_ENABLE"
+
+URL_PATTERN = r"/aqua/"
+
+
+def prepare(self):
+    self.set_header("Access-Control-Allow-Origin", "*")
+
+
+def make_app():
+    # Patch the prepare method to allow CORS request
+    if os.environ.get(AQUA_CORS_ENABLE, "0") == "1":
+        for _, handler in __handlers__:
+            handler.prepare = prepare
+    handlers = [(URL_PATTERN + url, handler) for url, handler in __handlers__]
+    # logger.debug(handlers)
+    return tornado.web.Application(handlers)
+
+
+def start_server():
+    app = make_app()
+    server = tornado.httpserver.HTTPServer(app)
+    port = int(os.environ.get(AQUA_PORT, 8080))
+    host = os.environ.get(AQUA_HOST, "0.0.0.0")
+    processes = int(os.environ.get(AQUA_PROCESS_COUNT, 0))
+    server.bind(port=port, address=host)
+    server.start(processes)
+    logger.info(f"Starting the server from directory: {os.getcwd()}")
+    logger.info(f"Aqua API server running on http://{host}:{port}")
+    tornado.ioloop.IOLoop.current().start()