polyaxon 2.2.0rc0__py3-none-any.whl → 2.3.0rc0__py3-none-any.whl

polyaxon/_client/project.py

@@ -12,6 +12,7 @@ from clipped.utils.validation import validate_tags
 
 from polyaxon._client.client import PolyaxonClient
 from polyaxon._client.decorators import client_handler, get_global_or_inline_config
+from polyaxon._client.mixin import ClientMixin
 from polyaxon._constants.globals import DEFAULT
 from polyaxon._contexts import paths as ctx_paths
 from polyaxon._env_vars.getters.user import get_local_owner
@@ -21,13 +22,17 @@ from polyaxon._sdk.schemas.v1_list_project_versions_response import (
 )
 from polyaxon._sdk.schemas.v1_project import V1Project
 from polyaxon._sdk.schemas.v1_project_version import V1ProjectVersion
-from polyaxon._utils.fqn_utils import get_entity_full_name, get_entity_info
+from polyaxon._utils.fqn_utils import (
+    get_entity_full_name,
+    get_entity_info,
+    split_owner_team_space,
+)
 from polyaxon.exceptions import ApiException, PolyaxonClientException
 from polyaxon.logger import logger
 from traceml.artifacts import V1RunArtifact
 
 
-class ProjectClient:
+class ProjectClient(ClientMixin):
     """ProjectClient is a client to communicate with Polyaxon projects endpoints.
 
     If no values are passed to this class,
@@ -94,26 +99,13 @@ class ProjectClient:
         if not owner:
             raise PolyaxonClientException("Please provide a valid owner.")
 
+        owner, team = split_owner_team_space(owner)
         self._client = client
         self._owner = owner or DEFAULT
+        self._team = team
         self._project = project
         self._project_data = V1Project.construct()
 
-    @property
-    def client(self):
-        if self._client:
-            return self._client
-        self._client = PolyaxonClient()
-        return self._client
-
-    @property
-    def owner(self):
-        return self._owner
-
-    @property
-    def project(self):
-        return self._project
-
     @property
     def project_data(self):
         return self._project_data
@@ -139,11 +131,19 @@ class ProjectClient:
         Returns:
             V1Project, project instance from the response.
         """
-        self._project_data = self.client.projects_v1.create_project(
-            self.owner,
-            data,
-            async_req=False,
-        )
+        if self.team:
+            self._project_data = self.client.projects_v1.create_team_project(
+                self.owner,
+                self.team,
+                data,
+                async_req=False,
+            )
+        else:
+            self._project_data = self.client.projects_v1.create_project(
+                self.owner,
+                data,
+                async_req=False,
+            )
         self._project_data.owner = self.owner
         self._project = self._project_data.name
         return self._project_data

polyaxon/_client/run.py

@@ -35,6 +35,7 @@ from polyaxon import settings
 from polyaxon._cli.errors import handle_cli_error
 from polyaxon._client.client import PolyaxonClient
 from polyaxon._client.decorators import client_handler, get_global_or_inline_config
+from polyaxon._client.mixin import ClientMixin
 from polyaxon._client.store import PolyaxonStore
 from polyaxon._constants.metadata import META_COPY_ARTIFACTS, META_RECOMPILE
 from polyaxon._containers.names import MAIN_CONTAINER_NAMES
@@ -63,7 +64,11 @@ from polyaxon._sdk.schemas.v1_operation_body import V1OperationBody
 from polyaxon._sdk.schemas.v1_project_version import V1ProjectVersion
 from polyaxon._sdk.schemas.v1_run import V1Run
 from polyaxon._sdk.schemas.v1_run_settings import V1RunSettings
-from polyaxon._utils.fqn_utils import get_entity_full_name, to_fqn_name
+from polyaxon._utils.fqn_utils import (
+    get_entity_full_name,
+    split_owner_team_space,
+    to_fqn_name,
+)
 from polyaxon._utils.urls_utils import get_proxy_run_url
 from polyaxon.api import K8S_V1_LOCATION, STREAMS_V1_LOCATION
 from polyaxon.exceptions import ApiException, PolyaxonClientException
@@ -80,7 +85,7 @@ if TYPE_CHECKING:
     from traceml.tracking.run import Run
 
 
-class RunClient:
+class RunClient(ClientMixin):
     """RunClient is a client to communicate with Polyaxon runs endpoints.
 
     If no values are passed to this class,
@@ -146,7 +151,7 @@ class RunClient:
             return
 
         try:
-            owner, project = get_project_or_local(
+            owner, _, project = get_project_or_local(
                 get_entity_full_name(owner=owner, entity=project)
             )
         except PolyaxonClientException:
@@ -166,8 +171,10 @@ class RunClient:
         if error_message and not self._is_offline:
             raise PolyaxonClientException(error_message)
 
+        owner, team = split_owner_team_space(owner)
         self._client = client
         self._owner = owner
+        self._team = team
         self._project = project
         self._run_uuid = (
             get_run_or_local(run_uuid)
@@ -218,12 +225,11 @@ class RunClient:
             return client.config.no_op
         return settings.CLIENT_CONFIG.no_op
 
-    @property
-    def client(self):
-        if self._client:
-            return self._client
-        self._client = PolyaxonClient()
-        return self._client
+    def _use_agent_host(self):
+        if self.settings.agent and self.settings.agent.url:
+            self.reset_client(
+                host=self.settings.agent.url, POLYAXON_HOST=self.settings.agent.url
+            )
 
     @property
     def store(self):
@@ -256,20 +262,6 @@ class RunClient:
             return self.settings.artifacts_store.name
         return None
 
-    @property
-    def owner(self) -> str:
-        return self._owner or ""
-
-    def set_owner(self, owner: str):
-        self._owner = owner
-
-    @property
-    def project(self) -> str:
-        return self._project or ""
-
-    def set_project(self, project: str):
-        self._project = project
-
     @property
     def run_uuid(self) -> str:
         return self._run_uuid
@@ -898,6 +890,7 @@ class RunClient:
         """
         if not self.settings:
             self.refresh_data()
+        self._use_agent_host()
         params = get_logs_params(
             last_file=last_file, last_time=last_time, connection=self.artifacts_store
         )
@@ -921,6 +914,7 @@ class RunClient:
     def inspect(self):
         if not self.settings:
             self.refresh_data()
+        self._use_agent_host()
         params = get_streams_params(connection=self.artifacts_store, status=self.status)
         return self.client.runs_v1.inspect_run(
             self.namespace, self.owner, self.project, self.run_uuid, **params
@@ -994,6 +988,10 @@ class RunClient:
                 if not container:
                     container = pod_containers[0]
 
+        if not self.settings:
+            self.refresh_data()
+        self._use_agent_host()
+
         url = get_proxy_run_url(
             service=K8S_V1_LOCATION,
             namespace=self.namespace,
@@ -1039,6 +1037,8 @@ class RunClient:
         """
         if not self.settings:
             self.refresh_data()
+        self._use_agent_host()
+
         params = get_streams_params(self.artifacts_store)
         return self.client.runs_v1.get_run_events(
             self.namespace,
@@ -1074,6 +1074,8 @@ class RunClient:
         """
         if not self.settings:
             self.refresh_data()
+        self._use_agent_host()
+
         params = get_streams_params(self.artifacts_store)
         return self.client.runs_v1.get_multi_run_events(
             self.namespace,
@@ -1310,6 +1312,7 @@ class RunClient:
         """
         if not self.settings:
             self.refresh_data()
+        self._use_agent_host()
         params = get_streams_params(self.artifacts_store)
         return self.client.runs_v1.get_run_artifact(
             namespace=self.namespace,
@@ -1343,6 +1346,10 @@ class RunClient:
         if not self.run_uuid:
             return
 
+        if not self.settings:
+            self.refresh_data()
+        self._use_agent_host()
+
         lineage_path = lineage.path or ""
         summary = lineage.summary or {}
         is_event = summary.get("is_event")
@@ -1421,6 +1428,8 @@ class RunClient:
         """
         if not self.settings:
             self.refresh_data()
+        self._use_agent_host()
+
         url = get_proxy_run_url(
             service=STREAMS_V1_LOCATION,
             namespace=self.namespace,
@@ -1460,6 +1469,8 @@ class RunClient:
         """
         if not self.settings:
             self.refresh_data()
+        self._use_agent_host()
+
         url = get_proxy_run_url(
             service=STREAMS_V1_LOCATION,
             namespace=self.namespace,
@@ -1507,6 +1518,7 @@ class RunClient:
         """
         if not self.settings:
             self.refresh_data()
+        self._use_agent_host()
 
         params = get_streams_params(connection=self.artifacts_store)
         url = get_proxy_run_url(
@@ -1596,6 +1608,7 @@ class RunClient:
 
         if not self.settings:
             self.refresh_data()
+        self._use_agent_host()
 
         params = get_streams_params(connection=self.artifacts_store)
         url = get_proxy_run_url(
@@ -1626,6 +1639,8 @@ class RunClient:
         """
         if not self.settings:
             self.refresh_data()
+        self._use_agent_host()
+
         params = get_streams_params(connection=self.artifacts_store)
         self.client.runs_v1.delete_run_artifact(
             namespace=self.namespace,
@@ -1645,6 +1660,8 @@ class RunClient:
         """
         if not self.settings:
             self.refresh_data()
+        self._use_agent_host()
+
         params = get_streams_params(connection=self.artifacts_store)
         return self.client.runs_v1.delete_run_artifacts(
             namespace=self.namespace,
@@ -1667,6 +1684,8 @@ class RunClient:
         """
         if not self.settings:
             self.refresh_data()
+        self._use_agent_host()
+
         params = get_streams_params(connection=self.artifacts_store)
         return self.client.runs_v1.get_run_artifacts_tree(
             namespace=self.namespace,
@@ -2117,9 +2136,9 @@ class RunClient:
         abspath = filepath if is_abs() else os.path.abspath(filepath)
 
         for_patterns = []
-        if getattr(self, "_artifacts_path"):
+        if getattr(self, "_artifacts_path", None):
             for_patterns.append(getattr(self, "_artifacts_path"))
-        if getattr(self, "_store_path"):
+        if getattr(self, "_store_path", None):
             for_patterns.append(getattr(self, "_store_path"))
         context_root = (
             ctx_paths.CONTEXT_OFFLINE_ROOT

polyaxon/_compiler/contexts/ray_job.py

@@ -33,7 +33,9 @@ class RayJobContextsManager(BaseContextsManager):
                 connection_by_names=connection_by_names,
             )
 
-        return {
+        data = {
             "head": _get_replica(job.head),
-            "workers": {wn: _get_replica(job.workers[wn]) for wn in job.workers},
         }
+        if job.workers:
+            data["workers"] = {wn: _get_replica(job.workers[wn]) for wn in job.workers}
+        return data

polyaxon/_env_vars/getters/owner_entity.py

@@ -5,7 +5,7 @@ from clipped.utils.strings import validate_slug
 
 from polyaxon._constants.globals import DEFAULT
 from polyaxon._env_vars.getters.user import get_local_owner
-from polyaxon._utils.fqn_utils import get_entity_info
+from polyaxon._utils.fqn_utils import get_entity_info, split_owner_team_space
 from polyaxon.exceptions import PolyaxonClientException, PolyaxonSchemaError
 
 
@@ -31,6 +31,8 @@ def resolve_entity_info(entity: str, entity_name: str, is_cli: bool = False):
     if not owner:
         owner = settings.AUTH_CONFIG.username if settings.AUTH_CONFIG else None
 
+    owner, team = split_owner_team_space(owner)
+
     if not all([owner, entity_value]):
         message = "Please provide a valid {}.".format(entity_name)
         if is_cli:
@@ -49,4 +51,4 @@ def resolve_entity_info(entity: str, entity_name: str, is_cli: bool = False):
                 entity_name, entity_value
             )
         )
-    return owner, entity_value
+    return owner, team, entity_value

polyaxon/_env_vars/getters/project.py

@@ -7,7 +7,7 @@ from polyaxon._constants.globals import DEFAULT
 from polyaxon._env_vars.getters.user import get_local_owner
 from polyaxon._managers.project import ProjectConfigManager
 from polyaxon._utils.cache import get_local_project
-from polyaxon._utils.fqn_utils import get_entity_info
+from polyaxon._utils.fqn_utils import get_entity_info, split_owner_team_space
 from polyaxon.exceptions import PolyaxonClientException, PolyaxonSchemaError
 
 
@@ -52,6 +52,8 @@ def get_project_or_local(project=None, is_cli: bool = False):
     if not owner and (not settings.CLI_CONFIG or settings.CLI_CONFIG.is_community):
         owner = DEFAULT
 
+    owner, team = split_owner_team_space(owner)
+
     if not all([owner, project_name]):
         error_message = get_project_error_message(owner, project_name)
         if is_cli:
@@ -79,4 +81,4 @@ def get_project_or_local(project=None, is_cli: bool = False):
             sys.exit(1)
         else:
             raise PolyaxonSchemaError(error_message)
-    return owner, project_name
+    return owner, team, project_name

polyaxon/_env_vars/getters/run.py

@@ -37,9 +37,9 @@ def get_run_or_local(run_uuid=None, is_cli: bool = False):
 
 
 def get_project_run_or_local(project=None, run_uuid=None, is_cli: bool = True):
-    user, project_name = get_project_or_local(project, is_cli=is_cli)
+    owner, team, project_name = get_project_or_local(project, is_cli=is_cli)
     run_uuid = get_run_or_local(run_uuid, is_cli=is_cli)
-    return user, project_name, run_uuid
+    return owner, team, project_name, run_uuid
 
 
 def get_collect_artifacts(arg: Optional[bool] = None, default: Optional[bool] = None):

polyaxon/_k8s/converter/base/base.py

@@ -2,6 +2,7 @@ import copy
 
 from typing import Dict, Iterable, List, Optional
 
+from clipped.utils.enums import get_enum_value
 from clipped.utils.sanitizers import sanitize_string_dict
 from clipped.utils.strings import slugify
 
@@ -85,7 +86,7 @@ class BaseConverter(
             "operation.polyaxon.com/name": self.run_name,
             "operation.polyaxon.com/owner": self.owner_name,
             "operation.polyaxon.com/project": self.project_name,
-            "operation.polyaxon.com/kind": self.K8S_ANNOTATIONS_KIND,
+            "operation.polyaxon.com/kind": get_enum_value(self.K8S_ANNOTATIONS_KIND),
         }
 
     def get_annotations(

polyaxon/_k8s/converter/common/accelerators.py

@@ -37,6 +37,9 @@ def requests_gpu(resources: k8s_schemas.V1ResourceRequirements) -> bool:
     if not resources:
         return False
 
+    if not isinstance(resources, k8s_schemas.V1ResourceRequirements):
+        resources = k8s_schemas.V1ResourceRequirements(**resources)
+
     if resources.requests:
         for key, val in resources.requests.items():
             if "gpu" in key and val is not None and val > 0:

polyaxon/_k8s/converter/converters/ray_job.py

@@ -58,7 +58,9 @@ class RayJobConverter(RayJobMixin, BaseConverter):
             config=compiled_operation.plugins, auth=default_auth
         )
         head = _get_replica(job.head)
-        workers = {n: _get_replica(w) for n, w in job.workers.items()}
+        workers = None
+        if job.workers:
+            workers = {n: _get_replica(w) for n, w in job.workers.items()}
         labels = self.get_labels(version=pkg.VERSION, labels={})
 
         return get_ray_job_custom_resource(
@@ -68,7 +70,7 @@ class RayJobConverter(RayJobMixin, BaseConverter):
             workers=workers,
             entrypoint=job.entrypoint,
             metadata=job.metadata,
-            runtime_env=encode(orjson_dumps(job.runtime_env)),
+            runtime_env=orjson_dumps(job.runtime_env),
             ray_version=job.ray_version,
             termination=compiled_operation.termination,
             collect_logs=plugins.collect_logs,

polyaxon/_k8s/custom_resources/dask_job.py

@@ -46,6 +46,9 @@ def get_dask_replicas_template(
     if liveness_probe and replica.main_container.liveness_probe is None:
         replica.main_container.liveness_probe = liveness_probe
 
+    labels = {**labels, **replica.labels}
+    annotations = {**annotations, **replica.annotations}
+
     metadata, pod_spec = get_pod_spec(
         namespace=namespace,
         main_container=replica.main_container,

polyaxon/_k8s/custom_resources/kubeflow/common.py

@@ -16,6 +16,9 @@ def get_kf_replicas_template(
     if not replica:
         return
 
+    labels = {**labels, **replica.labels}
+    annotations = {**annotations, **replica.annotations}
+
     metadata, pod_spec = get_pod_spec(
         namespace=namespace,
         main_container=replica.main_container,

polyaxon/_k8s/custom_resources/ray_job.py

@@ -26,6 +26,9 @@ def _get_ray_replicas_template(
     if not replica:
         return
 
+    labels = {**labels, **replica.labels}
+    annotations = {**annotations, **replica.annotations}
+
     metadata, pod_spec = get_pod_spec(
         namespace=namespace,
         main_container=replica.main_container,

polyaxon/_k8s/custom_resources/setter.py

@@ -46,7 +46,7 @@ def set_notify(custom_object: Dict, notifications: List[V1Notification]) -> Dict
 def set_clean_pod_policy(template_spec: Dict, clean_pod_policy: str) -> Dict:
     if not clean_pod_policy:
         # Sets default clean pod policy
-        clean_pod_policy = "All"
+        clean_pod_policy = "None"
 
     template_spec["cleanPodPolicy"] = clean_pod_policy.capitalize()
     return template_spec

polyaxon/_pql/manager.py

@@ -13,7 +13,7 @@ class PQLManager:
     FIELDS_PROXY = {}
     FIELDS_TRANS = {}
     FIELDS_ORDERING = None
-    FIELDS_ORDERING_PROXY = None
+    FIELDS_ORDERING_PROXY = None  # Do not set a field on both field and proxy
     FIELDS_DEFAULT_ORDERING = None
     FIELDS_DISTINCT = None
     CHECK_ALIVE = True

polyaxon/_sdk/api/runs_v1_api.py

@@ -3014,8 +3014,8 @@ class RunsV1Api(BaseApi):
         self,
         namespace: Annotated[StrictStr, Field(..., description="namespace")],
         owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
-        project: Annotated[
-            StrictStr, Field(..., description="Project where the run will be assigned")
+        entity: Annotated[
+            StrictStr, Field(..., description="Entity where the run will be assigned")
         ],
         kind: Annotated[str, Field(..., description="The artifact kind")],
         names: Annotated[
@@ -3046,15 +3046,15 @@ class RunsV1Api(BaseApi):
         This method makes a synchronous HTTP request by default. To make an
         asynchronous HTTP request, please pass async_req=True
 
-        >>> thread = api.get_multi_run_events(namespace, owner, project, kind, names, runs, orient, force, sample, connection, status, async_req=True)
+        >>> thread = api.get_multi_run_events(namespace, owner, entity, kind, names, runs, orient, force, sample, connection, status, async_req=True)
         >>> result = thread.get()
 
         :param namespace: namespace (required)
         :type namespace: str
         :param owner: Owner of the namespace (required)
         :type owner: str
-        :param project: Project where the run will be assigned (required)
-        :type project: str
+        :param entity: Entity where the run will be assigned (required)
+        :type entity: str
         :param kind: The artifact kind (required)
         :type kind: str
         :param names: Names query param.
@@ -3090,7 +3090,7 @@ class RunsV1Api(BaseApi):
         return self.get_multi_run_events_with_http_info(
             namespace,
             owner,
-            project,
+            entity,
             kind,
             names,
             runs,
@@ -3107,8 +3107,8 @@ class RunsV1Api(BaseApi):
         self,
         namespace: Annotated[StrictStr, Field(..., description="namespace")],
         owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
-        project: Annotated[
-            StrictStr, Field(..., description="Project where the run will be assigned")
+        entity: Annotated[
+            StrictStr, Field(..., description="Entity where the run will be assigned")
         ],
         kind: Annotated[str, Field(..., description="The artifact kind")],
         names: Annotated[
@@ -3139,15 +3139,15 @@ class RunsV1Api(BaseApi):
         This method makes a synchronous HTTP request by default. To make an
         asynchronous HTTP request, please pass async_req=True
 
-        >>> thread = api.get_multi_run_events_with_http_info(namespace, owner, project, kind, names, runs, orient, force, sample, connection, status, async_req=True)
+        >>> thread = api.get_multi_run_events_with_http_info(namespace, owner, entity, kind, names, runs, orient, force, sample, connection, status, async_req=True)
         >>> result = thread.get()
 
         :param namespace: namespace (required)
         :type namespace: str
         :param owner: Owner of the namespace (required)
         :type owner: str
-        :param project: Project where the run will be assigned (required)
-        :type project: str
+        :param entity: Entity where the run will be assigned (required)
+        :type entity: str
         :param kind: The artifact kind (required)
         :type kind: str
         :param names: Names query param.
@@ -3193,7 +3193,7 @@ class RunsV1Api(BaseApi):
         _all_params = [
             "namespace",
             "owner",
-            "project",
+            "entity",
             "kind",
             "names",
             "runs",
@@ -3233,8 +3233,8 @@ class RunsV1Api(BaseApi):
             _path_params["namespace"] = _params["namespace"]
         if _params["owner"]:
             _path_params["owner"] = _params["owner"]
-        if _params["project"]:
-            _path_params["project"] = _params["project"]
+        if _params["entity"]:
+            _path_params["entity"] = _params["entity"]
         if _params["kind"]:
             _path_params["kind"] = _params["kind"]
 
@@ -3281,7 +3281,7 @@ class RunsV1Api(BaseApi):
         }
 
         return self.api_client.call_api(
-            "/streams/v1/{namespace}/{owner}/{project}/runs/multi/events/{kind}",
+            "/streams/v1/{namespace}/{owner}/{entity}/runs/multi/events/{kind}",
             "GET",
             _path_params,
             _query_params,
@@ -3304,8 +3304,8 @@ class RunsV1Api(BaseApi):
         self,
         namespace: Annotated[StrictStr, Field(..., description="namespace")],
         owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
-        project: Annotated[
-            StrictStr, Field(..., description="Project where the run will be assigned")
+        entity: Annotated[
+            StrictStr, Field(..., description="Entity where the run will be assigned")
         ],
         body: Annotated[Dict[str, Any], Field(..., description="Params/Metrics data")],
         **kwargs
@@ -3315,15 +3315,15 @@ class RunsV1Api(BaseApi):
         This method makes a synchronous HTTP request by default. To make an
         asynchronous HTTP request, please pass async_req=True
 
-        >>> thread = api.get_multi_run_importance(namespace, owner, project, body, async_req=True)
+        >>> thread = api.get_multi_run_importance(namespace, owner, entity, body, async_req=True)
         >>> result = thread.get()
 
         :param namespace: namespace (required)
         :type namespace: str
         :param owner: Owner of the namespace (required)
         :type owner: str
-        :param project: Project where the run will be assigned (required)
-        :type project: str
+        :param entity: Entity where the run will be assigned (required)
+        :type entity: str
         :param body: Params/Metrics data (required)
         :type body: object
         :param async_req: Whether to execute the request asynchronously.
@@ -3343,7 +3343,7 @@ class RunsV1Api(BaseApi):
         """
         kwargs["_return_http_data_only"] = True
         return self.get_multi_run_importance_with_http_info(
-            namespace, owner, project, body, **kwargs
+            namespace, owner, entity, body, **kwargs
         )  # noqa: E501
 
     @validate_arguments
@@ -3351,8 +3351,8 @@ class RunsV1Api(BaseApi):
         self,
         namespace: Annotated[StrictStr, Field(..., description="namespace")],
         owner: Annotated[StrictStr, Field(..., description="Owner of the namespace")],
-        project: Annotated[
-            StrictStr, Field(..., description="Project where the run will be assigned")
+        entity: Annotated[
+            StrictStr, Field(..., description="Entity where the run will be assigned")
         ],
         body: Annotated[Dict[str, Any], Field(..., description="Params/Metrics data")],
         **kwargs
@@ -3362,15 +3362,15 @@ class RunsV1Api(BaseApi):
         This method makes a synchronous HTTP request by default. To make an
         asynchronous HTTP request, please pass async_req=True
 
-        >>> thread = api.get_multi_run_importance_with_http_info(namespace, owner, project, body, async_req=True)
+        >>> thread = api.get_multi_run_importance_with_http_info(namespace, owner, entity, body, async_req=True)
         >>> result = thread.get()
 
         :param namespace: namespace (required)
         :type namespace: str
         :param owner: Owner of the namespace (required)
         :type owner: str
-        :param project: Project where the run will be assigned (required)
-        :type project: str
+        :param entity: Entity where the run will be assigned (required)
+        :type entity: str
         :param body: Params/Metrics data (required)
         :type body: object
         :param async_req: Whether to execute the request asynchronously.
@@ -3399,7 +3399,7 @@ class RunsV1Api(BaseApi):
 
         _params = locals()
 
-        _all_params = ["namespace", "owner", "project", "body"]
+        _all_params = ["namespace", "owner", "entity", "body"]
         _all_params.extend(
             [
                 "async_req",
@@ -3432,8 +3432,8 @@ class RunsV1Api(BaseApi):
         if _params["owner"]:
             _path_params["owner"] = _params["owner"]
 
-        if _params["project"]:
-            _path_params["project"] = _params["project"]
+        if _params["entity"]:
+            _path_params["entity"] = _params["entity"]
 
         # process the query parameters
         _query_params = []
@@ -3471,7 +3471,7 @@ class RunsV1Api(BaseApi):
         }
 
         return self.api_client.call_api(
-            "/streams/v1/{namespace}/{owner}/{project}/runs/multi/importance",
+            "/streams/v1/{namespace}/{owner}/{entity}/runs/multi/importance",
             "POST",
             _path_params,
             _query_params,

polyaxon/_sdk/schemas/v1_settings_catalog.py

@@ -9,3 +9,4 @@ class V1SettingsCatalog(BaseAllowSchemaModel):
     uuid: Optional[UUIDStr]
     name: Optional[StrictStr]
     version: Optional[StrictStr]
+    url: Optional[StrictStr]