[Application] Create API Gateway by default (#5515)

mlrun · May 14, 2024 · 6fa0bfb · 6fa0bfb
1 parent 40a74a3
commit 6fa0bfb
Show file tree

Hide file tree

Showing 13 changed files with 358 additions and 125 deletions.
diff --git a/mlrun/config.py b/mlrun/config.py
@@ -232,6 +232,10 @@
         "databricks": {
             "artifact_directory_path": "/mlrun_databricks_runtime/artifacts_dictionaries"
         },
+        "application": {
+            "default_sidecar_internal_port": 8050,
+            "default_authentication_mode": "accessKey",
+        },
     },
     # TODO: function defaults should be moved to the function spec config above
     "function_defaults": {

diff --git a/mlrun/db/base.py b/mlrun/db/base.py
@@ -629,8 +629,8 @@ def verify_authorization(
     @abstractmethod
     def store_api_gateway(
         self,
-        project: str,
         api_gateway: mlrun.common.schemas.APIGateway,
+        project: str = None,
     ):
         pass
 

diff --git a/mlrun/db/nopdb.py b/mlrun/db/nopdb.py
@@ -520,8 +520,11 @@ def get_hub_item(
 
     def store_api_gateway(
         self,
-        project: str,
-        api_gateway: mlrun.runtimes.nuclio.APIGateway,
+        api_gateway: Union[
+            mlrun.common.schemas.APIGateway,
+            mlrun.runtimes.nuclio.api_gateway.APIGateway,
+        ],
+        project: str = None,
     ) -> mlrun.common.schemas.APIGateway:
         pass
 

diff --git a/mlrun/runtimes/__init__.py b/mlrun/runtimes/__init__.py
@@ -43,6 +43,7 @@
     new_v2_model_server,
     nuclio_init_hook,
 )
+from .nuclio.api_gateway import APIGateway
 from .nuclio.application import ApplicationRuntime
 from .nuclio.serving import serving_subkind
 from .remotesparkjob import RemoteSparkRuntime