mlrun 1.3.3__py3-none-any.whl → 1.4.0__py3-none-any.whl

mlrun/__init__.py

@@ -1,4 +1,4 @@
-# Copyright 2018 Iguazio
+# Copyright 2023 Iguazio
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -35,6 +35,7 @@ from .db import get_run_db
 from .errors import MLRunInvalidArgumentError, MLRunNotFoundError
 from .execution import MLClientCtx
 from .model import RunObject, RunTemplate, new_task
+from .package import ArtifactType, DefaultPackager, Packager, handler
 from .platforms import (
     VolumeMount,
     auto_mount,
@@ -62,14 +63,13 @@ from .run import (
     get_object,
     get_or_create_ctx,
     get_pipeline,
-    handler,
     import_function,
     new_function,
     run_local,
     run_pipeline,
     wait_for_pipeline_completion,
 )
-from .runtimes import ArtifactType, new_model_server
+from .runtimes import new_model_server
 from .secrets import get_secret_or_env
 from .utils.version import Version
 

mlrun/__main__.py

@@ -1,6 +1,6 @@
 #!/usr/bin/env python
 
-# Copyright 2018 Iguazio
+# Copyright 2023 Iguazio
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -34,11 +34,9 @@ from tabulate import tabulate
 
 import mlrun
 
-from .builder import upload_tarball
 from .config import config as mlconf
 from .db import get_run_db
 from .errors import err_to_str
-from .k8s_utils import K8sHelper
 from .model import RunTemplate
 from .platforms import auto_mount as auto_mount_modifier
 from .projects import load_project
@@ -545,7 +543,7 @@ def build(
         logger.info(f"uploading data from {src} to {archive}")
         target = archive if archive.endswith("/") else archive + "/"
         target += f"src-{meta.project}-{meta.name}-{meta.tag or 'latest'}.tar.gz"
-        upload_tarball(src, target)
+        mlrun.datastore.utils.upload_tarball(src, target)
         # todo: replace function.yaml inside the tar
         b.source = target
 
@@ -700,20 +698,6 @@ def deploy(
         fp.write(function.status.nuclio_name)
 
 
-@main.command(context_settings=dict(ignore_unknown_options=True))
-@click.argument("pod", type=str, callback=validate_base_argument)
-@click.option("--namespace", "-n", help="kubernetes namespace")
-@click.option(
-    "--timeout", "-t", default=600, show_default=True, help="timeout in seconds"
-)
-def watch(pod, namespace, timeout):
-    """Read current or previous task (pod) logs."""
-    print("This command will be deprecated in future version !!!\n")
-    k8s = K8sHelper(namespace)
-    status = k8s.watch(pod, namespace, timeout)
-    print(f"Pod {pod} last status is: {status}")
-
-
 @main.command(context_settings=dict(ignore_unknown_options=True))
 @click.argument("kind", type=str, callback=validate_base_argument)
 @click.argument(
@@ -779,7 +763,7 @@ def get(kind, name, selector, namespace, uid, project, tag, db, extra_args):
             name, project=project, tag=tag, labels=selector
         )
         df = artifacts.to_df()[
-            ["tree", "key", "iter", "kind", "path", "hash", "updated"]
+            ["tree", "key", "iter", "kind", "path", "hash", "updated", "uri"]
         ]
         df["tree"] = df["tree"].apply(lambda x: f"..{x[-8:]}")
         df["hash"] = df["hash"].apply(lambda x: f"..{x[-6:]}")
@@ -1052,6 +1036,15 @@ def logs(uid, project, offset, db, watch):
     is_flag=True,
     help="Store the project secrets as k8s secrets",
 )
+@click.option(
+    "--notifications",
+    "--notification",
+    "-nt",
+    multiple=True,
+    help="To have a notification for the run set notification file "
+    "destination define: file=notification.json or a "
+    'dictionary configuration e.g \'{"slack":{"webhook":"<webhook>"}}\'',
+)
 def project(
     context,
     name,
@@ -1077,6 +1070,7 @@ def project(
     timeout,
     ensure_project,
     schedule,
+    notifications,
     overwrite_schedule,
     save_secrets,
     save,
@@ -1095,25 +1089,30 @@ def project(
     if db:
         mlconf.dbpath = db
 
-    proj = load_project(context, url, name, init_git=init_git, clone=clone, save=save)
+    # set the CLI/GIT parameters in load_project() so they can be used by project setup scripts
+    parameters = fill_params(param) if param else {}
+    if git_repo:
+        parameters["git_repo"] = git_repo
+    if git_issue:
+        parameters["git_issue"] = git_issue
+    commit = environ.get("GITHUB_SHA") or environ.get("CI_COMMIT_SHA")
+    if commit and not parameters.get("commit_id"):
+        parameters["commit_id"] = commit
+
+    proj = load_project(
+        context,
+        url,
+        name,
+        init_git=init_git,
+        clone=clone,
+        save=save,
+        parameters=parameters,
+    )
     url_str = " from " + url if url else ""
     print(f"Loading project {proj.name}{url_str} into {context}:\n")
 
     if is_relative_path(artifact_path):
         artifact_path = path.abspath(artifact_path)
-    if param:
-        proj.spec.params = fill_params(param, proj.spec.params)
-    if git_repo:
-        proj.spec.params["git_repo"] = git_repo
-    if git_issue:
-        proj.spec.params["git_issue"] = git_issue
-    commit = (
-        proj.get_param("commit_id")
-        or environ.get("GITHUB_SHA")
-        or environ.get("CI_COMMIT_SHA")
-    )
-    if commit:
-        proj.spec.params["commit_id"] = commit
     if secrets:
         secrets = line2keylist(secrets, "kind", "source")
         secret_store = SecretsStore.from_list(secrets)
@@ -1152,6 +1151,8 @@ def project(
                     "token": proj.get_param("GIT_TOKEN"),
                 },
             )
+        if notifications:
+            load_notification(notifications, proj)
         try:
             proj.run(
                 name=run,
@@ -1169,11 +1170,9 @@ def project(
                 timeout=timeout,
                 overwrite=overwrite_schedule,
             )
-
-        except Exception as exc:
+        except Exception as err:
             print(traceback.format_exc())
-            message = f"failed to run pipeline, {err_to_str(exc)}"
-            proj.notifiers.push(message, "error")
+            send_workflow_error_notification(run, proj, err)
             exit(1)
 
     elif sync:
@@ -1450,5 +1449,48 @@ def func_url_to_runtime(func_url, ensure_project: bool = False):
     return runtime
 
 
+def load_notification(notifications: str, project: mlrun.projects.MlrunProject):
+    """
+    A dictionary or json file containing notification dictionaries can be used by the user to set notifications.
+    Each notification is stored in a tuple called notifications.
+    The code then goes through each value in the notifications tuple and check
+    if the notification starts with "file=", such as "file=notification.json," in those cases it loads the
+    notification.json file and uses add_notification_to_project to add the notifications from the file to
+    the project. If not, it adds the notification dictionary to the project.
+    :param notifications:  Notifications file or a dictionary to be added to the project
+    :param project: The object to which the notifications will be added
+    :return:
+    """
+    for notification in notifications:
+        if notification.startswith("file="):
+            file_path = notification.split("=")[-1]
+            notification = open(file_path, "r")
+            notification = json.load(notification)
+        else:
+            notification = json.loads(notification)
+        add_notification_to_project(notification, project)
+
+
+def add_notification_to_project(
+    notification: str, project: mlrun.projects.MlrunProject
+):
+    for notification_type, notification_params in notification.items():
+        project.notifiers.add_notification(
+            notification_type=notification_type, params=notification_params
+        )
+
+
+def send_workflow_error_notification(
+    run_id: str, project: mlrun.projects.MlrunProject, error: KeyError
+):
+    message = (
+        f":x: Failed to run scheduled workflow {run_id} in Project {project.name} !\n"
+        f"error: ```{err_to_str(error)}```"
+    )
+    project.notifiers.push(
+        message=message, severity=mlrun.common.schemas.NotificationSeverity.ERROR
+    )
+
+
 if __name__ == "__main__":
     main()

mlrun/api/__init__.py

@@ -1,4 +1,4 @@
-# Copyright 2018 Iguazio
+# Copyright 2023 Iguazio
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

mlrun/api/api/__init__.py

@@ -1,4 +1,4 @@
-# Copyright 2018 Iguazio
+# Copyright 2023 Iguazio
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

mlrun/api/api/api.py

@@ -1,4 +1,4 @@
-# Copyright 2018 Iguazio
+# Copyright 2023 Iguazio
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -28,9 +28,9 @@ from mlrun.api.api.endpoints import (
     functions,
     grafana_proxy,
     healthz,
+    hub,
     internal,
     logs,
-    marketplace,
     model_endpoints,
     operations,
     pipelines,
@@ -125,8 +125,8 @@ api_router.include_router(
 api_router.include_router(grafana_proxy.router, tags=["grafana", "model-endpoints"])
 api_router.include_router(model_endpoints.router, tags=["model-endpoints"])
 api_router.include_router(
-    marketplace.router,
-    tags=["marketplace"],
+    hub.router,
+    tags=["hub"],
     dependencies=[Depends(mlrun.api.api.deps.authenticate_request)],
 )
 api_router.include_router(

mlrun/api/api/deps.py

@@ -1,4 +1,4 @@
-# Copyright 2018 Iguazio
+# Copyright 2023 Iguazio
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -20,9 +20,9 @@ from sqlalchemy.orm import Session
 
 import mlrun
 import mlrun.api.db.session
-import mlrun.api.schemas
 import mlrun.api.utils.auth.verifier
 import mlrun.api.utils.clients.iguazio
+import mlrun.common.schemas
 
 
 def get_db_session() -> typing.Generator[Session, None, None]:
@@ -35,7 +35,7 @@ def get_db_session() -> typing.Generator[Session, None, None]:
             mlrun.api.db.session.close_session(db_session)
 
 
-async def authenticate_request(request: Request) -> mlrun.api.schemas.AuthInfo:
+async def authenticate_request(request: Request) -> mlrun.common.schemas.AuthInfo:
     return await mlrun.api.utils.auth.verifier.AuthVerifier().authenticate_request(
         request
     )
@@ -46,7 +46,7 @@ def verify_api_state(request: Request):
         request.scope
     )
     path = path_with_query_string.split("?")[0]
-    if mlrun.mlconf.httpdb.state == mlrun.api.schemas.APIStates.offline:
+    if mlrun.mlconf.httpdb.state == mlrun.common.schemas.APIStates.offline:
         enabled_endpoints = [
             # we want to stay healthy
             "healthz",
@@ -56,10 +56,10 @@ def verify_api_state(request: Request):
         if not any(enabled_endpoint in path for enabled_endpoint in enabled_endpoints):
             raise mlrun.errors.MLRunPreconditionFailedError("API is in offline state")
     if mlrun.mlconf.httpdb.state in [
-        mlrun.api.schemas.APIStates.waiting_for_migrations,
-        mlrun.api.schemas.APIStates.migrations_in_progress,
-        mlrun.api.schemas.APIStates.migrations_failed,
-        mlrun.api.schemas.APIStates.waiting_for_chief,
+        mlrun.common.schemas.APIStates.waiting_for_migrations,
+        mlrun.common.schemas.APIStates.migrations_in_progress,
+        mlrun.common.schemas.APIStates.migrations_failed,
+        mlrun.common.schemas.APIStates.waiting_for_chief,
     ]:
         enabled_endpoints = [
             "healthz",
@@ -70,20 +70,9 @@ def verify_api_state(request: Request):
             "memory-reports",
         ]
         if not any(enabled_endpoint in path for enabled_endpoint in enabled_endpoints):
-            message = (
-                "API is waiting for migrations to be triggered. Send POST request to /api/operations/migrations to"
-                " trigger it"
-            )
-            if (
-                mlrun.mlconf.httpdb.state
-                == mlrun.api.schemas.APIStates.migrations_in_progress
-            ):
-                message = "Migrations are in progress"
-            elif (
+            message = mlrun.common.schemas.APIStates.description(
                 mlrun.mlconf.httpdb.state
-                == mlrun.api.schemas.APIStates.migrations_failed
-            ):
-                message = "Migrations failed, API can't be started"
+            )
             raise mlrun.errors.MLRunPreconditionFailedError(message)
 
 

mlrun/api/api/endpoints/__init__.py

@@ -1,4 +1,4 @@
-# Copyright 2018 Iguazio
+# Copyright 2023 Iguazio
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.

mlrun/api/api/endpoints/artifacts.py

@@ -1,4 +1,4 @@
-# Copyright 2018 Iguazio
+# Copyright 2023 Iguazio
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -22,18 +22,23 @@ from sqlalchemy.orm import Session
 import mlrun.api.crud
 import mlrun.api.utils.auth.verifier
 import mlrun.api.utils.singletons.project_member
-from mlrun.api import schemas
+import mlrun.common.schemas
 from mlrun.api.api import deps
 from mlrun.api.api.utils import log_and_raise
-from mlrun.api.schemas.artifact import ArtifactsFormat
+from mlrun.common.schemas.artifact import ArtifactsFormat
 from mlrun.config import config
-from mlrun.utils import is_legacy_artifact
+from mlrun.utils import is_legacy_artifact, logger
 
 router = APIRouter()
 
 
-# TODO /artifact/{project}/{uid}/{key:path} should be deprecated in 1.4
-@router.post("/artifact/{project}/{uid}/{key:path}")
+# TODO: remove /artifact/{project}/{uid}/{key:path} in 1.6.0
+@router.post(
+    "/artifact/{project}/{uid}/{key:path}",
+    deprecated=True,
+    description="/artifact/{project}/{uid}/{key:path} is deprecated in 1.4.0 and will be removed in 1.6.0, "
+    "use /projects/{project}/artifacts/{uid}/{key:path} instead",
+)
 @router.post("/projects/{project}/artifacts/{uid}/{key:path}")
 async def store_artifact(
     request: Request,
@@ -42,7 +47,7 @@ async def store_artifact(
     key: str,
     tag: str = "",
     iter: int = 0,
-    auth_info: mlrun.api.schemas.AuthInfo = Depends(deps.authenticate_request),
+    auth_info: mlrun.common.schemas.AuthInfo = Depends(deps.authenticate_request),
     db_session: Session = Depends(deps.get_db_session),
 ):
     await run_in_threadpool(
@@ -52,10 +57,10 @@ async def store_artifact(
         auth_info=auth_info,
     )
     await mlrun.api.utils.auth.verifier.AuthVerifier().query_project_resource_permissions(
-        mlrun.api.schemas.AuthorizationResourceTypes.artifact,
+        mlrun.common.schemas.AuthorizationResourceTypes.artifact,
         project,
         key,
-        mlrun.api.schemas.AuthorizationAction.store,
+        mlrun.common.schemas.AuthorizationAction.store,
         auth_info,
     )
 
@@ -65,6 +70,9 @@ async def store_artifact(
     except ValueError:
         log_and_raise(HTTPStatus.BAD_REQUEST.value, reason="bad JSON body")
 
+    logger.debug(
+        "Storing artifact", project=project, uid=uid, key=key, tag=tag, iter=iter
+    )
     await run_in_threadpool(
         mlrun.api.crud.Artifacts().store_artifact,
         db_session,
@@ -81,13 +89,13 @@ async def store_artifact(
 @router.get("/projects/{project}/artifact-tags")
 async def list_artifact_tags(
     project: str,
-    category: schemas.ArtifactCategories = None,
-    auth_info: mlrun.api.schemas.AuthInfo = Depends(deps.authenticate_request),
+    category: mlrun.common.schemas.ArtifactCategories = None,
+    auth_info: mlrun.common.schemas.AuthInfo = Depends(deps.authenticate_request),
     db_session: Session = Depends(deps.get_db_session),
 ):
     await mlrun.api.utils.auth.verifier.AuthVerifier().query_project_permissions(
         project,
-        mlrun.api.schemas.AuthorizationAction.read,
+        mlrun.common.schemas.AuthorizationAction.read,
         auth_info,
     )
     tag_tuples = await run_in_threadpool(
@@ -95,7 +103,7 @@ async def list_artifact_tags(
     )
     artifact_key_to_tag = {tag_tuple[1]: tag_tuple[2] for tag_tuple in tag_tuples}
     allowed_artifact_keys = await mlrun.api.utils.auth.verifier.AuthVerifier().filter_project_resources_by_permissions(
-        mlrun.api.schemas.AuthorizationResourceTypes.artifact,
+        mlrun.common.schemas.AuthorizationResourceTypes.artifact,
         list(artifact_key_to_tag.keys()),
         lambda artifact_key: (
             project,
@@ -115,8 +123,13 @@ async def list_artifact_tags(
     }
 
 
-# TODO /projects/{project}/artifact/{key:path} should be deprecated in 1.4
-@router.get("/projects/{project}/artifact/{key:path}")
+# TODO: remove /projects/{project}/artifact/{key:path} in 1.6.0
+@router.get(
+    "/projects/{project}/artifact/{key:path}",
+    deprecated=True,
+    description="/projects/{project}/artifact/{key:path} is deprecated in 1.4.0 and will be removed in 1.6.0, "
+    "use /projects/{project}/artifacts/{key:path} instead",
+)
 @router.get("/projects/{project}/artifacts/{key:path}")
 async def get_artifact(
     project: str,
@@ -124,7 +137,7 @@ async def get_artifact(
     tag: str = "latest",
     iter: int = 0,
     format_: ArtifactsFormat = Query(ArtifactsFormat.full, alias="format"),
-    auth_info: mlrun.api.schemas.AuthInfo = Depends(deps.authenticate_request),
+    auth_info: mlrun.common.schemas.AuthInfo = Depends(deps.authenticate_request),
     db_session: Session = Depends(deps.get_db_session),
 ):
     data = await run_in_threadpool(
@@ -137,10 +150,10 @@ async def get_artifact(
         format_,
     )
     await mlrun.api.utils.auth.verifier.AuthVerifier().query_project_resource_permissions(
-        mlrun.api.schemas.AuthorizationResourceTypes.artifact,
+        mlrun.common.schemas.AuthorizationResourceTypes.artifact,
         project,
         key,
-        mlrun.api.schemas.AuthorizationAction.read,
+        mlrun.common.schemas.AuthorizationAction.read,
         auth_info,
     )
     return {
@@ -148,22 +161,27 @@ async def get_artifact(
     }
 
 
-# TODO /artifact/{project}/{uid} should be deprecated in 1.4
-@router.delete("/artifact/{project}/{uid}")
+# TODO: remove /artifact/{project}/{uid} in 1.6.0
+@router.delete(
+    "/artifact/{project}/{uid}",
+    deprecated=True,
+    description="/artifact/{project}/{uid} is deprecated in 1.4.0 and will be removed in 1.6.0, "
+    "use /projects/{project}/artifacts/{uid} instead",
+)
 @router.delete("/projects/{project}/artifacts/{uid}")
 async def delete_artifact(
     project: str,
     uid: str,
     key: str,
     tag: str = "",
-    auth_info: mlrun.api.schemas.AuthInfo = Depends(deps.authenticate_request),
+    auth_info: mlrun.common.schemas.AuthInfo = Depends(deps.authenticate_request),
     db_session: Session = Depends(deps.get_db_session),
 ):
     await mlrun.api.utils.auth.verifier.AuthVerifier().query_project_resource_permissions(
-        mlrun.api.schemas.AuthorizationResourceTypes.artifact,
+        mlrun.common.schemas.AuthorizationResourceTypes.artifact,
         project,
         key,
-        mlrun.api.schemas.AuthorizationAction.delete,
+        mlrun.common.schemas.AuthorizationAction.delete,
         auth_info,
     )
     await run_in_threadpool(
@@ -172,27 +190,32 @@ async def delete_artifact(
     return {}
 
 
-# TODO /artifacts should be deprecated in 1.4
-@router.get("/artifacts")
+# TODO: remove /artifacts in 1.6.0
+@router.get(
+    "/artifacts",
+    deprecated=True,
+    description="/artifacts is deprecated in 1.4.0 and will be removed in 1.6.0, "
+    "use /projects/{project}/artifacts instead",
+)
 @router.get("/projects/{project}/artifacts")
 async def list_artifacts(
     project: str = None,
     name: str = None,
     tag: str = None,
     kind: str = None,
-    category: schemas.ArtifactCategories = None,
+    category: mlrun.common.schemas.ArtifactCategories = None,
     labels: List[str] = Query([], alias="label"),
     iter: int = Query(None, ge=0),
     best_iteration: bool = Query(False, alias="best-iteration"),
     format_: ArtifactsFormat = Query(ArtifactsFormat.full, alias="format"),
-    auth_info: mlrun.api.schemas.AuthInfo = Depends(deps.authenticate_request),
+    auth_info: mlrun.common.schemas.AuthInfo = Depends(deps.authenticate_request),
     db_session: Session = Depends(deps.get_db_session),
 ):
     if project is None:
         project = config.default_project
     await mlrun.api.utils.auth.verifier.AuthVerifier().query_project_permissions(
         project,
-        mlrun.api.schemas.AuthorizationAction.read,
+        mlrun.common.schemas.AuthorizationAction.read,
         auth_info,
     )
 
@@ -211,7 +234,7 @@ async def list_artifacts(
     )
 
     artifacts = await mlrun.api.utils.auth.verifier.AuthVerifier().filter_project_resources_by_permissions(
-        mlrun.api.schemas.AuthorizationResourceTypes.artifact,
+        mlrun.common.schemas.AuthorizationResourceTypes.artifact,
         artifacts,
         _artifact_project_and_resource_name_extractor,
         auth_info,
@@ -221,14 +244,19 @@ async def list_artifacts(
     }
 
 
-# TODO /artifacts should be deprecated in 1.4
-@router.delete("/artifacts")
+# TODO: remove /artifacts in 1.6.0
+@router.delete(
+    "/artifacts",
+    deprecated=True,
+    description="/artifacts is deprecated in 1.4.0 and will be removed in 1.6.0, "
+    "use /projects/{project}/artifacts instead",
+)
 async def delete_artifacts_legacy(
     project: str = mlrun.mlconf.default_project,
     name: str = "",
     tag: str = "",
     labels: List[str] = Query([], alias="label"),
-    auth_info: mlrun.api.schemas.AuthInfo = Depends(deps.authenticate_request),
+    auth_info: mlrun.common.schemas.AuthInfo = Depends(deps.authenticate_request),
     db_session: Session = Depends(deps.get_db_session),
 ):
     return await _delete_artifacts(
@@ -247,7 +275,7 @@ async def delete_artifacts(
     name: str = "",
     tag: str = "",
     labels: List[str] = Query([], alias="label"),
-    auth_info: mlrun.api.schemas.AuthInfo = Depends(deps.authenticate_request),
+    auth_info: mlrun.common.schemas.AuthInfo = Depends(deps.authenticate_request),
     db_session: Session = Depends(deps.get_db_session),
 ):
     return await _delete_artifacts(
@@ -265,7 +293,7 @@ async def _delete_artifacts(
     name: str = None,
     tag: str = None,
     labels: List[str] = None,
-    auth_info: mlrun.api.schemas.AuthInfo = None,
+    auth_info: mlrun.common.schemas.AuthInfo = None,
     db_session: Session = None,
 ):
     artifacts = await run_in_threadpool(
@@ -277,10 +305,10 @@ async def _delete_artifacts(
         labels,
     )
     await mlrun.api.utils.auth.verifier.AuthVerifier().query_project_resources_permissions(
-        mlrun.api.schemas.AuthorizationResourceTypes.artifact,
+        mlrun.common.schemas.AuthorizationResourceTypes.artifact,
         artifacts,
         _artifact_project_and_resource_name_extractor,
-        mlrun.api.schemas.AuthorizationAction.delete,
+        mlrun.common.schemas.AuthorizationAction.delete,
         auth_info,
     )
     await run_in_threadpool(

mlrun/api/api/endpoints/auth.py

@@ -1,4 +1,4 @@
-# Copyright 2018 Iguazio
+# Copyright 2023 Iguazio
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -15,16 +15,16 @@
 import fastapi
 
 import mlrun.api.api.deps
-import mlrun.api.schemas
 import mlrun.api.utils.auth.verifier
+import mlrun.common.schemas
 
 router = fastapi.APIRouter()
 
 
 @router.post("/authorization/verifications")
 async def verify_authorization(
-    authorization_verification_input: mlrun.api.schemas.AuthorizationVerificationInput,
-    auth_info: mlrun.api.schemas.AuthInfo = fastapi.Depends(
+    authorization_verification_input: mlrun.common.schemas.AuthorizationVerificationInput,
+    auth_info: mlrun.common.schemas.AuthInfo = fastapi.Depends(
         mlrun.api.api.deps.authenticate_request
     ),
 ):