anyscale 0.24.37__tar.gz → 0.24.39__tar.gz

{anyscale-0.24.37 → anyscale-0.24.39}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: anyscale
-Version: 0.24.37
+Version: 0.24.39
 Summary: Command Line Interface for Anyscale
 Author: Anyscale Inc.
 License: AS License

{anyscale-0.24.37 → anyscale-0.24.39}/anyscale/_private/anyscale_client/anyscale_client.py

@@ -1,11 +1,10 @@
 from abc import ABC, abstractmethod
-import asyncio
 import io
 import logging
 import os
 import pathlib
 import time
-from typing import Any, Callable, cast, Dict, IO, List, Optional, Tuple
+from typing import Any, Callable, Dict, Generator, IO, List, Optional, Tuple
 from urllib.parse import urlparse
 
 import requests
@@ -17,9 +16,9 @@ from anyscale._private.anyscale_client.common import (
     RUNTIME_ENV_PACKAGE_FORMAT,
 )
 from anyscale._private.models.image_uri import ImageURI
-from anyscale.api_utils.job_logs_util import _get_job_logs_from_storage_bucket_streaming
+from anyscale.api_utils.logs_util import _download_log_from_s3_url_sync
 from anyscale.authenticate import AuthenticationBlock, get_auth_api_client
-from anyscale.cli_logger import BlockLogger, LogsLogger
+from anyscale.cli_logger import BlockLogger
 from anyscale.client.openapi_client.api.default_api import DefaultApi as InternalApi
 from anyscale.client.openapi_client.models import (
     ArchiveStatus,
@@ -47,17 +46,16 @@ from anyscale.sdk.anyscale_client.models import (
     ClusterComputeConfig,
     ClusterEnvironment,
     ClusterEnvironmentBuild,
+    ClusterenvironmentbuildListResponse,
     ClusterEnvironmentBuildStatus,
+    ClusterEnvironmentsQuery,
+    CreateClusterEnvironment,
+    CreateClusterEnvironmentBuild,
     Job as APIJobRun,
     Project,
     RollbackServiceModel,
     ServiceModel,
-)
-from anyscale.sdk.anyscale_client.models.create_cluster_environment import (
-    CreateClusterEnvironment,
-)
-from anyscale.sdk.anyscale_client.models.create_cluster_environment_build import (
-    CreateClusterEnvironmentBuild,
+    TextQuery,
 )
 from anyscale.sdk.anyscale_client.models.jobs_query import JobsQuery
 from anyscale.sdk.anyscale_client.models.jobs_sort_field import JobsSortField
@@ -68,6 +66,7 @@ from anyscale.sdk.anyscale_client.models.sort_by_clause_jobs_sort_field import (
 from anyscale.sdk.anyscale_client.models.sort_order import SortOrder
 from anyscale.sdk.anyscale_client.rest import ApiException as ExternalApiException
 from anyscale.shared_anyscale_utils.conf import ANYSCALE_HOST
+from anyscale.shared_anyscale_utils.latest_ray_version import LATEST_RAY_VERSION
 from anyscale.util import (
     get_cluster_model_for_current_workspace,
     get_endpoint,
@@ -601,15 +600,35 @@ class AnyscaleClient(AnyscaleClientInterface):
         ).result
         return result.id
 
-    def _get_cluster_env_by_name(self, name: str) -> Optional[ClusterEnvironment]:
-        cluster_envs = self._external_api_client.search_cluster_environments(
-            {
-                "name": {"equals": name},
-                "paging": {"count": 1},
-                "include_anonymous": True,
-            }
-        ).results
-        return cluster_envs[0] if cluster_envs else None
+    def get_cluster_env_by_name(self, name: str) -> Optional[ClusterEnvironment]:
+        resp = self._external_api_client.search_cluster_environments(
+            ClusterEnvironmentsQuery(
+                name=TextQuery(equals=name),  # pyright: ignore reportGeneralTypeIssues
+                paging=PageQuery(count=1),
+                include_anonymous=True,
+            )
+        )
+        if resp.results:
+            return resp.results[0]
+        return None
+
+    def list_cluster_env_builds(
+        self, cluster_env_id: str,
+    ) -> Generator[ClusterEnvironmentBuild, None, None]:
+        paging_token = None
+        while True:
+            resp: ClusterenvironmentbuildListResponse = self._external_api_client.list_cluster_environment_builds(
+                cluster_environment_id=cluster_env_id,
+                count=self.LIST_ENDPOINT_COUNT,
+                paging_token=paging_token,
+                desc=True,
+            )
+            for build in resp.results:
+                yield build
+
+            if resp.metadata.next_paging_token is None:
+                break
+            paging_token = resp.metadata.next_paging_token
 
     def _wait_for_build_to_succeed(
         self, build_id: str, poll_interval_seconds=3, timeout_secs=3600,
@@ -644,7 +663,7 @@ class AnyscaleClient(AnyscaleClientInterface):
     def _find_or_create_cluster_env(
         self, cluster_env_name: str, anonymous: bool
     ) -> ClusterEnvironment:
-        existing_cluster_env = self._get_cluster_env_by_name(cluster_env_name)
+        existing_cluster_env = self.get_cluster_env_by_name(cluster_env_name)
         if existing_cluster_env is not None:
             return existing_cluster_env
 
@@ -661,7 +680,11 @@ class AnyscaleClient(AnyscaleClientInterface):
         return cluster_environment
 
     def get_cluster_env_build_id_from_containerfile(
-        self, cluster_env_name: str, containerfile: str, anonymous: bool = True
+        self,
+        cluster_env_name: str,
+        containerfile: str,
+        anonymous: bool = True,
+        ray_version: Optional[str] = None,
     ) -> str:
         cluster_env = self._find_or_create_cluster_env(
             cluster_env_name, anonymous=anonymous
@@ -673,14 +696,24 @@ class AnyscaleClient(AnyscaleClientInterface):
             if (
                 build.status == ClusterEnvironmentBuildStatus.SUCCEEDED
                 and build.containerfile == containerfile
+                # we don't need to check the ray_version because checking the containerfile is enough.
             ):
                 return build.id
 
-        build_op = self._external_api_client.create_cluster_environment_build(
-            CreateClusterEnvironmentBuild(
-                cluster_environment_id=cluster_env.id, containerfile=containerfile,
-            )
-        ).result
+        try:
+            build_op = self._external_api_client.create_cluster_environment_build(
+                CreateClusterEnvironmentBuild(
+                    cluster_environment_id=cluster_env.id,
+                    containerfile=containerfile,
+                    ray_version=ray_version,  # we don't use the latest version here if ray_version is Noneb/c the backend will try to parse the base image to decide the ray version.
+                )
+            ).result
+        except ExternalApiException as e:
+            if e.status == 400:
+                raise RuntimeError(
+                    "Invalid containerfile. Please check the syntax and try again.",
+                    e.body,
+                ) from None
 
         build_url = self.get_build_ui_url(
             cluster_env.id, build_op.cluster_environment_build_id
@@ -691,7 +724,12 @@ class AnyscaleClient(AnyscaleClientInterface):
 
         return build_op.cluster_environment_build_id
 
-    def get_cluster_env_build_id_from_image_uri(self, image_uri: ImageURI) -> str:
+    def get_cluster_env_build_id_from_image_uri(
+        self,
+        image_uri: ImageURI,
+        registry_login_secret: Optional[str] = None,
+        ray_version: Optional[str] = None,
+    ) -> str:
         if image_uri.is_cluster_env_image():
             identifier = image_uri.to_cluster_env_identifier()
             try:
@@ -714,7 +752,6 @@ class AnyscaleClient(AnyscaleClientInterface):
         cluster_env = self._find_or_create_cluster_env(cluster_env_name, anonymous=True)
         image_uri_str = str(image_uri)
 
-        # since we encode the image_uri into the cluster env name, there should exist one and only one build that matches the image_uri.
         cluster_env_builds = self._external_api_client.list_cluster_environment_builds(
             cluster_environment_id=cluster_env.id, count=1
         ).results
@@ -722,7 +759,9 @@ class AnyscaleClient(AnyscaleClientInterface):
         if (
             build is not None
             and build.docker_image_name == image_uri_str
+            and build.registry_login_secret == registry_login_secret
             and build.status == ClusterEnvironmentBuildStatus.SUCCEEDED
+            and build.ray_version == ray_version
         ):
             return build.id
 
@@ -732,6 +771,8 @@ class AnyscaleClient(AnyscaleClientInterface):
                 # For historical reasons, we have to use docker_image_name instead of image_uri; but it is just a URI to the image.
                 cluster_environment_id=cluster_env.id,
                 docker_image_name=image_uri_str,
+                registry_login_secret=registry_login_secret,
+                ray_version=ray_version if ray_version else LATEST_RAY_VERSION,
             )
         ).result
 
@@ -909,24 +950,70 @@ class AnyscaleClient(AnyscaleClientInterface):
 
         return info.file_uri
 
-    def logs_for_job_run(self, job_run_id: str) -> str:
-        """Retrieves logs from the streaming job logs folder in S3/GCS"""
-        logs = ""
+    def _fetch_log_chunks(self, job_run_id: str) -> Tuple[List[str], Any]:
+        all_log_chunk_urls = []
+        MAX_PAGE_SIZE = 1000
         next_page_token = None
-        cluster_journal_events_start_line = 0
-
-        (logs, next_page_token, cluster_journal_events_start_line, _,) = asyncio.run(
-            _get_job_logs_from_storage_bucket_streaming(
-                self._internal_api_client,
-                cast(
-                    LogsLogger, self.logger
-                ),  # self.logger is a BlockLogger, cast to LogsLogger subclass
-                job_run_id=job_run_id,
-                remove_escape_chars=False,
+        bearer_token = None
+        while True:
+            log_download_result = self._internal_api_client.get_job_logs_download_v2_api_v2_logs_job_logs_download_v2_job_id_get(
+                job_id=job_run_id,
                 next_page_token=next_page_token,
-                cluster_journal_events_start_line=cluster_journal_events_start_line,
-                no_cluster_journal_events=True,
+                page_size=MAX_PAGE_SIZE,
+            ).result
+
+            if bearer_token is None:
+                bearer_token = log_download_result.bearer_token
+
+            all_log_chunk_urls.extend(
+                [chunk.chunk_url for chunk in log_download_result.log_chunks]
             )
-        )
 
+            next_page_token = log_download_result.next_page_token
+            if next_page_token is None:
+                break
+
+        return all_log_chunk_urls, bearer_token
+
+    def _read_log_lines(
+        self,
+        log_chunk_urls: List[str],
+        head: bool,
+        bearer_token: Any,
+        max_lines: Optional[int],
+    ) -> str:
+        # TODO(mowen): Would be nice to return some placeholder here to allow all new
+        # logs to be read from a particular point onwards.
+        result_lines: List[str] = []
+        step = 1 if head else -1
+        line_count = 0
+        for chunk_url in log_chunk_urls[::step]:
+            log_lines = _download_log_from_s3_url_sync(
+                chunk_url, bearer_token=bearer_token
+            ).splitlines()
+
+            if max_lines is not None:
+                lines_to_add = min(len(log_lines), max_lines - line_count)
+            else:
+                lines_to_add = len(log_lines)
+            line_count += lines_to_add
+
+            if head:
+                result_lines = result_lines + log_lines[:lines_to_add]
+            else:
+                result_lines = log_lines[-1 * lines_to_add :] + result_lines
+
+            if line_count == max_lines:
+                break
+
+        return "\n".join(result_lines) + "\n"
+
+    def logs_for_job_run(
+        self, job_run_id: str, head: bool = False, max_lines: Optional[int] = None
+    ) -> str:
+        """Retrieves logs from the streaming job logs folder in S3/GCS"""
+
+        all_log_chunk_urls, bearer_token = self._fetch_log_chunks(job_run_id)
+
+        logs = self._read_log_lines(all_log_chunk_urls, head, bearer_token, max_lines)
         return logs

{anyscale-0.24.37 → anyscale-0.24.39}/anyscale/_private/anyscale_client/common.py

@@ -1,5 +1,5 @@
 from abc import ABC, abstractmethod
-from typing import Dict, List, Optional, Tuple
+from typing import Dict, Generator, List, Optional, Tuple
 
 from anyscale._private.models.image_uri import ImageURI
 from anyscale.client.openapi_client.models import (
@@ -15,6 +15,7 @@ from anyscale.sdk.anyscale_client.models import (
     ApplyServiceModel,
     Cluster,
     ClusterCompute,
+    ClusterEnvironment,
     Job as APIJobRun,
     ServiceModel,
 )
@@ -184,6 +185,23 @@ class AnyscaleClientInterface(ABC):
         """
         raise NotImplementedError
 
+    @abstractmethod
+    def get_cluster_env_by_name(self, name: str) -> Optional[ClusterEnvironment]:
+        """Get a cluster environment by its name.
+
+        """
+        raise NotImplementedError
+
+    @abstractmethod
+    def list_cluster_env_builds(
+        self, cluster_env_id: str
+    ) -> Generator[ClusterEnvironmentBuild, None, None]:
+        """List cluster environment builds for the provided cluster environment id.
+
+        Returns a list of cluster environment builds.
+        """
+        raise NotImplementedError
+
     @abstractmethod
     def get_cluster_env_build(self, build_id: str) -> Optional[ClusterEnvironmentBuild]:
         """Get the cluster env build.
@@ -192,7 +210,11 @@ class AnyscaleClientInterface(ABC):
 
     @abstractmethod
     def get_cluster_env_build_id_from_containerfile(
-        self, cluster_env_name: str, containerfile: str, anonymous: bool
+        self,
+        cluster_env_name: str,
+        containerfile: str,
+        anonymous: bool,
+        ray_version: Optional[str] = None,
     ) -> str:
         """Get the cluster environment build ID for the cluster environment with provided containerfile.
         Look for an existing cluster environment with the provided name.
@@ -203,7 +225,10 @@ class AnyscaleClientInterface(ABC):
 
     @abstractmethod
     def get_cluster_env_build_id_from_image_uri(
-        self, image_uri: ImageURI,
+        self,
+        image_uri: ImageURI,
+        registry_login_secret: Optional[str] = None,
+        ray_version: Optional[str] = None,
     ):
         """Get the cluster environment build ID for the cluster environment with provided image_uri.
 
@@ -327,7 +352,13 @@ class AnyscaleClientInterface(ABC):
         raise NotImplementedError
 
     @abstractmethod
-    def logs_for_job_run(self, job_run_id: str) -> str:
+    def logs_for_job_run(
+        self,
+        job_run_id: str,
+        head: bool = False,
+        tail: bool = True,
+        max_lines: Optional[int] = None,
+    ) -> str:
         """Returns the logs associated with a particular job run.
 
         The job_run_id is the backend id for the job run.

{anyscale-0.24.37 → anyscale-0.24.39}/anyscale/_private/anyscale_client/fake_anyscale_client.py

@@ -1,7 +1,7 @@
 from collections import defaultdict
 from datetime import datetime
 import logging
-from typing import DefaultDict, Dict, List, Optional, Tuple
+from typing import DefaultDict, Dict, Generator, List, Optional, Tuple
 import uuid
 
 from anyscale._private.anyscale_client.common import (
@@ -29,6 +29,8 @@ from anyscale.sdk.anyscale_client.models import (
     Cluster,
     ClusterCompute,
     ClusterComputeConfig,
+    ClusterEnvironmentBuild,
+    ClusterEnvironmentBuildStatus,
     ComputeNodeType,
     Job as APIJobRun,
     ProductionServiceV2VersionModel,
@@ -36,12 +38,8 @@ from anyscale.sdk.anyscale_client.models import (
     ServiceModel,
     ServiceVersionState,
 )
-from anyscale.sdk.anyscale_client.models.cluster_environment_build import (
-    ClusterEnvironmentBuild,
-)
-from anyscale.sdk.anyscale_client.models.cluster_environment_build_status import (
-    ClusterEnvironmentBuildStatus,
-)
+from anyscale.sdk.anyscale_client.models.cluster_environment import ClusterEnvironment
+from anyscale.shared_anyscale_utils.latest_ray_version import LATEST_RAY_VERSION
 from anyscale.utils.workspace_notification import WorkspaceNotification
 
 
@@ -69,9 +67,38 @@ class FakeAnyscaleClient(AnyscaleClientInterface):
     WORKSPACE_CLUSTER_ENV_BUILD_ID = "fake-workspace-cluster-env-build-id"
 
     def __init__(self):
-        self._image_uri_to_build_id: Dict[ImageURI, str] = {}
-        self._build_id_to_image_uri: Dict[str, ImageURI] = {}
-        self._containerfile_to_build_id: Dict[str, str] = {}
+        self._builds: Dict[str, ClusterEnvironmentBuild] = {
+            self.DEFAULT_CLUSTER_ENV_BUILD_ID: ClusterEnvironmentBuild(
+                id=self.DEFAULT_CLUSTER_ENV_BUILD_ID,
+                cluster_environment_id="default-cluster-env-id",
+                docker_image_name="docker.io/my/base-image:latest",
+                status=ClusterEnvironmentBuildStatus.SUCCEEDED,
+                local_vars_configuration=OPENAPI_NO_VALIDATION,
+            ),
+            self.WORKSPACE_CLUSTER_ENV_BUILD_ID: ClusterEnvironmentBuild(
+                id=self.WORKSPACE_CLUSTER_ENV_BUILD_ID,
+                cluster_environment_id="workspace-cluster-env-id",
+                docker_image_name="docker.io/my/base-ws-image:latest",
+                status=ClusterEnvironmentBuildStatus.SUCCEEDED,
+                local_vars_configuration=OPENAPI_NO_VALIDATION,
+            ),
+        }
+        self._images: Dict[str, ClusterEnvironment] = {
+            "default-cluster-env-id": ClusterEnvironment(
+                id="default-cluster-env-id",
+                name="default-cluster-env",
+                anonymous=True,
+                is_default=True,
+                local_vars_configuration=OPENAPI_NO_VALIDATION,
+            ),
+            "workspace-cluster-env-id": ClusterEnvironment(
+                id="workspace-cluster-env-id",
+                name="default-workspace-cluster-env",
+                anonymous=True,
+                is_default=True,
+                local_vars_configuration=OPENAPI_NO_VALIDATION,
+            ),
+        }
         self._compute_config_name_to_ids: DefaultDict[str, List[str]] = defaultdict(
             list
         )
@@ -270,6 +297,12 @@ class FakeAnyscaleClient(AnyscaleClientInterface):
 
         raise RuntimeError(f"Cloud with name '{cloud_name}' not found.")
 
+    def add_build(self, build: ClusterEnvironmentBuild):
+        self._builds[build.id] = build
+
+    def add_image(self, image: ClusterEnvironment):
+        self._images[image.id] = image
+
     def add_cloud(self, cloud: Cloud):
         self._clouds[cloud.id] = cloud
 
@@ -360,12 +393,23 @@ class FakeAnyscaleClient(AnyscaleClientInterface):
     def get_default_compute_config(self, *, cloud_id: str) -> ClusterCompute:
         return self._default_compute_configs[cloud_id]
 
-    def set_image_uri_mapping(self, image_uri: ImageURI, build_id: str):
-        self._image_uri_to_build_id[image_uri] = build_id
-        self._build_id_to_image_uri[build_id] = image_uri
-
-    def set_containerfile_mapping(self, containerfile: str, build_id: str):
-        self._containerfile_to_build_id[containerfile] = build_id
+    def list_cluster_env_builds(
+        self, cluster_env_id: str,
+    ) -> Generator[ClusterEnvironmentBuild, None, None]:
+        for v in self._builds.values():
+            if v.cluster_environment_id == cluster_env_id:
+                yield v
+
+    def get_non_default_cluster_env_builds(self) -> List[ClusterEnvironmentBuild]:
+        return [
+            v
+            for v in self._builds.values()
+            if v.id
+            not in [
+                self.DEFAULT_CLUSTER_ENV_BUILD_ID,
+                self.WORKSPACE_CLUSTER_ENV_BUILD_ID,
+            ]
+        ]
 
     def get_default_build_id(self) -> str:
         workspace_cluster = self.get_current_workspace_cluster()
@@ -374,32 +418,91 @@ class FakeAnyscaleClient(AnyscaleClientInterface):
         return self.DEFAULT_CLUSTER_ENV_BUILD_ID
 
     def get_cluster_env_build(self, build_id: str) -> Optional[ClusterEnvironmentBuild]:
-        if build_id in self._build_id_to_image_uri:
-            return ClusterEnvironmentBuild(
-                docker_image_name=self._build_id_to_image_uri[build_id],
-                status=ClusterEnvironmentBuildStatus.SUCCEEDED,
-                local_vars_configuration=OPENAPI_NO_VALIDATION,
-            )
+        return self._builds.get(build_id, None)
+
+    def get_cluster_env_by_name(self, name) -> Optional[ClusterEnvironment]:
+        for v in self._images.values():
+            if v.name == name:
+                return v
         return None
 
     def get_cluster_env_build_id_from_containerfile(
-        self, cluster_env_name: str, containerfile: str, anonymous: bool  # noqa: ARG002
+        self,
+        cluster_env_name: str,
+        containerfile: str,
+        anonymous: bool,
+        ray_version: Optional[str] = None,  # noqa: ARG002
     ) -> str:
-        if containerfile in self._containerfile_to_build_id:
-            return self._containerfile_to_build_id[containerfile]
-        else:
-            raise ValueError(f"The containerfile '{containerfile}' does not exist.")
+        for build in self._builds.values():
+            if build.containerfile == containerfile:
+                cluster_env = self._images.get(build.cluster_environment_id, None)  # type: ignore
+                if cluster_env is not None and cluster_env.name == cluster_env_name:
+                    return build.id  # type: ignore
+        # create a new one if not found
+        cluster_env_id = f"cluster-env-id-{uuid.uuid4()!s}"
+        self._images[cluster_env_id] = ClusterEnvironment(
+            id=cluster_env_id,
+            name=cluster_env_name,
+            anonymous=anonymous,
+            local_vars_configuration=OPENAPI_NO_VALIDATION,
+        )
+        latest_build = None
+        for build in self._builds.values():
+            if build.cluster_environment_id == cluster_env_id and (latest_build is None or build.revision > latest_build.revision):  # type: ignore
+                latest_build = build
+        build_id = f"cluster-env-build-id-{uuid.uuid4()!s}"
+        self._builds[build_id] = ClusterEnvironmentBuild(
+            id=build_id,
+            cluster_environment_id=cluster_env_id,
+            containerfile=containerfile,
+            status=ClusterEnvironmentBuildStatus.SUCCEEDED,
+            local_vars_configuration=OPENAPI_NO_VALIDATION,
+            ray_version=ray_version,
+            revision=latest_build.revision + 1 if latest_build is not None else 1,  # type: ignore
+        )
+        return build_id
 
-    def get_cluster_env_build_id_from_image_uri(self, image_uri: ImageURI) -> str:
-        if image_uri in self._image_uri_to_build_id:
-            return self._image_uri_to_build_id[image_uri]
-        else:
-            raise ValueError(f"The image_uri '{image_uri}' does not exist.")
+    def get_cluster_env_build_id_from_image_uri(
+        self,
+        image_uri: ImageURI,
+        registry_login_secret: Optional[str] = None,
+        ray_version: Optional[str] = None,
+    ) -> str:
+        for build in self._builds.values():
+            if (
+                build.docker_image_name == image_uri.image_uri
+                and ray_version == build.ray_version
+                and build.registry_login_secret == registry_login_secret
+            ):
+                return build.id  # type: ignore
+        cluster_env_id = f"cluster-env-id-{uuid.uuid4()!s}"
+        cluster_env_name = image_uri.to_cluster_env_name()
+        self._images[cluster_env_id] = ClusterEnvironment(
+            id=cluster_env_id,
+            name=cluster_env_name,
+            anonymous=False,
+            local_vars_configuration=OPENAPI_NO_VALIDATION,
+        )
+        build_id = f"cluster-env-build-id-{uuid.uuid4()!s}"
+        self._builds[build_id] = ClusterEnvironmentBuild(
+            id=build_id,
+            cluster_environment_id=cluster_env_id,
+            docker_image_name=image_uri.image_uri,
+            status=ClusterEnvironmentBuildStatus.SUCCEEDED,
+            registry_login_secret=registry_login_secret,
+            ray_version=ray_version if ray_version else LATEST_RAY_VERSION,
+            local_vars_configuration=OPENAPI_NO_VALIDATION,
+        )
+        return build_id
 
     def get_cluster_env_build_image_uri(
         self, cluster_env_build_id: str
     ) -> Optional[ImageURI]:
-        return self._build_id_to_image_uri.get(cluster_env_build_id)
+        build = self._builds.get(cluster_env_build_id, None)
+        if build is None:
+            return None
+        else:
+            return ImageURI.from_str(build.docker_image_name)
 
     def update_service(self, model: ServiceModel):
         self._services[model.id] = model
@@ -588,5 +691,16 @@ class FakeAnyscaleClient(AnyscaleClientInterface):
     def add_job_run_logs(self, job_run_id: str, logs: str):
         self._job_run_logs[job_run_id] = logs
 
-    def logs_for_job_run(self, job_run_id: str) -> str:
-        return self._job_run_logs[job_run_id]
+    def logs_for_job_run(
+        self,
+        job_run_id: str,
+        head: bool = False,  # noqa: ARG002
+        max_lines: Optional[int] = None,  # noqa: ARG002
+    ) -> str:
+        log_lines = self._job_run_logs.get(job_run_id, "").splitlines()
+        if max_lines is None:
+            max_lines = len(log_lines)
+        if head:
+            return "\n".join(log_lines[:max_lines])
+        else:
+            return "\n".join(log_lines[-1 * max_lines :])

{anyscale-0.24.37 → anyscale-0.24.39}/anyscale/_private/docgen/README.md

@@ -1,6 +1,6 @@
 Tool to autogenerate documentation for the Anyscale SDK/CLI that can be embedded in a [Docusarus](https://docusaurus.io/) site.
 
-This is used by: https://github.com/anyscale/endpoint-docs.
+This is used by: https://github.com/anyscale/docs/tree/master/docs/reference.
 
 Usage:
 ```bash