diff --git a/.github/workflows/docker-push-images.yml b/.github/workflows/docker-push-images.yml index 36cfcdbd..567df2b5 100644 --- a/.github/workflows/docker-push-images.yml +++ b/.github/workflows/docker-push-images.yml @@ -21,12 +21,12 @@ jobs: with: username: ${{ secrets.DOCKER_HUB_USERNAME }} password: ${{ secrets.DOCKER_HUB_ACCESS_TOKEN }} - - name: Login to the Container registry - uses: docker/login-action@65b78e6e13532edd9afa3aa52ac7964289d1a9c1 - with: - registry: ghcr.io - username: ${{ github.actor }} - password: ${{ secrets.GITHUB_TOKEN }} +# - name: Login to the Container registry +# uses: docker/login-action@65b78e6e13532edd9afa3aa52ac7964289d1a9c1 +# with: +# registry: ghcr.io +# username: ${{ github.actor }} +# password: ${{ secrets.GITHUB_TOKEN }} - name: Build Container Images run: | @@ -36,10 +36,11 @@ jobs: run: | make docker-push-all - - name: Build Container Images with Github Container Registry prefix - run: | - GIT_COMMIT_HASH=${{ github.sha }} AIBRIX_CONTAINER_REGISTRY_NAMESPACE=ghcr.io/aibrix make docker-build-all - - - name: Push Container Images to Github Container Registry - run: | - GIT_COMMIT_HASH=${{ github.sha }} AIBRIX_CONTAINER_REGISTRY_NAMESPACE=ghcr.io/aibrix make docker-push-all +# TODO: Disable GHCR at this moment after org transfer, let's enable it later. +# - name: Build Container Images with Github Container Registry prefix +# run: | +# GIT_COMMIT_HASH=${{ github.sha }} AIBRIX_CONTAINER_REGISTRY_NAMESPACE=ghcr.io/aibrix make docker-build-all +# +# - name: Push Container Images to Github Container Registry +# run: | +# GIT_COMMIT_HASH=${{ github.sha }} AIBRIX_CONTAINER_REGISTRY_NAMESPACE=ghcr.io/aibrix make docker-push-all diff --git a/.github/workflows/release-build.yaml b/.github/workflows/release-build.yaml index 3626fdff..33d35480 100644 --- a/.github/workflows/release-build.yaml +++ b/.github/workflows/release-build.yaml @@ -30,12 +30,12 @@ jobs: password: ${{ secrets.DOCKER_HUB_ACCESS_TOKEN }} # Log in to Github Registry - - name: Login to the Container registry - uses: docker/login-action@65b78e6e13532edd9afa3aa52ac7964289d1a9c1 - with: - registry: ghcr.io - username: ${{ github.actor }} - password: ${{ secrets.GITHUB_TOKEN }} +# - name: Login to the Container registry +# uses: docker/login-action@65b78e6e13532edd9afa3aa52ac7964289d1a9c1 +# with: +# registry: ghcr.io +# username: ${{ github.actor }} +# password: ${{ secrets.GITHUB_TOKEN }} # Build container images with docker registry namespace - name: Build Container Images @@ -48,14 +48,14 @@ jobs: IS_MAIN_BRANCH=false GIT_COMMIT_HASH=${{ github.ref_name }} make docker-push-all # Build container images with Github registry namespace - - name: Build Container Images with Github Container Registry prefix - run: | - IS_MAIN_BRANCH=false GIT_COMMIT_HASH=${{ github.ref_name }} AIBRIX_CONTAINER_REGISTRY_NAMESPACE=ghcr.io/aibrix make docker-build-all +# - name: Build Container Images with Github Container Registry prefix +# run: | +# IS_MAIN_BRANCH=false GIT_COMMIT_HASH=${{ github.ref_name }} AIBRIX_CONTAINER_REGISTRY_NAMESPACE=ghcr.io/aibrix make docker-build-all # Push container image to Github container registry - - name: Push Container Images to Github Container Registry - run: | - IS_MAIN_BRANCH=false GIT_COMMIT_HASH=${{ github.ref_name }} AIBRIX_CONTAINER_REGISTRY_NAMESPACE=ghcr.io/aibrix make docker-push-all +# - name: Push Container Images to Github Container Registry +# run: | +# IS_MAIN_BRANCH=false GIT_COMMIT_HASH=${{ github.ref_name }} AIBRIX_CONTAINER_REGISTRY_NAMESPACE=ghcr.io/aibrix make docker-push-all python-wheel-release: runs-on: ubuntu-latest diff --git a/CONTRIBUTING.md b/CONTRIBUTING.md index df920ed9..b7fa1e40 100644 --- a/CONTRIBUTING.md +++ b/CONTRIBUTING.md @@ -12,25 +12,25 @@ please see the [Development Guide](development/README.md) in the `docs/developme ### 1. Understanding the Repository Structure -You can refer to the [Design Doc](https://github.com/aibrix/aibrix/tree/main/docs/tutorial) to quickly understand the layout of the AIBrix directory structure. +You can refer to the [Design Doc](https://github.com/vllm-project/aibrix/tree/main/docs/tutorial) to quickly understand the layout of the AIBrix directory structure. ### 2. Picking an Issue -Start by picking an [issue](https://github.com/aibrix/aibrix/issues) tagged with "good first issue" to get familiar with the project. Claim the issue by commenting to avoid duplicate efforts. +Start by picking an [issue](https://github.com/vllm-project/aibrix/issues) tagged with "good first issue" to get familiar with the project. Claim the issue by commenting to avoid duplicate efforts. ### 3. Submitting Codes - **Creating Branches**: Use the following naming convention: `//[feat|patch|bug_fix] description` - **Making Commits**: Write detailed commit messages that explain your changes. - **Pull Requests**: - - Create a pull request on [PR Page](https://github.com/aibrix/aibrix/pulls) against the master branch. + - Create a pull request on [PR Page](https://github.com/vllm-project/aibrix/pulls) against the master branch. - Use the provided PR template to describe your changes. - **Code Review Process**: Your PR will be reviewed by related contributors. Address feedback to improve and finalize your contribution. ## Advanced Contributions - **Feature Proposals and Bug Reporting**: Report bugs or propose new features by creating an issue on our GitHub page. Provide as much detail as possible to facilitate discussions. -- **Contributing to Documentation**: Help improve our documentation by proposing changes to issue list. Our documentation is at [AIBrix Doc](https://github.com/aibrix/aibrix/tree/main/docs/tutorial) +- **Contributing to Documentation**: Help improve our documentation by proposing changes to issue list. Our documentation is at [AIBrix Doc](https://github.com/vllm-project/aibrix/tree/main/docs/tutorial) - **Organizing Community Events**: If you are interested in organizing meetups or webinars, contact our community manager (TODO: contact information). ## Community and Communication diff --git a/PROJECT b/PROJECT index 2c95fe00..9b8b97ad 100644 --- a/PROJECT +++ b/PROJECT @@ -7,7 +7,7 @@ layout: - go.kubebuilder.io/v4 multigroup: true projectName: aibrix -repo: github.com/aibrix/aibrix +repo: github.com/vllm-project/aibrix resources: - api: crdVersion: v1 @@ -16,7 +16,7 @@ resources: domain: aibrix.ai group: autoscaling kind: PodAutoscaler - path: github.com/aibrix/aibrix/api/autoscaling/v1alpha1 + path: github.com/vllm-project/aibrix/api/autoscaling/v1alpha1 version: v1alpha1 - api: crdVersion: v1 @@ -25,7 +25,7 @@ resources: domain: aibrix.ai group: model kind: ModelAdapter - path: github.com/aibrix/aibrix/api/model/v1alpha1 + path: github.com/vllm-project/aibrix/api/model/v1alpha1 version: v1alpha1 - api: crdVersion: v1 @@ -34,7 +34,7 @@ resources: domain: aibrix.ai group: orchestration kind: RayClusterReplicaSet - path: github.com/aibrix/aibrix/api/orchestration/v1alpha1 + path: github.com/vllm-project/aibrix/api/orchestration/v1alpha1 version: v1alpha1 - api: crdVersion: v1 @@ -43,7 +43,7 @@ resources: domain: aibrix.ai group: orchestration kind: RayClusterFleet - path: github.com/aibrix/aibrix/api/orchestration/v1alpha1 + path: github.com/vllm-project/aibrix/api/orchestration/v1alpha1 version: v1alpha1 - api: crdVersion: v1 @@ -52,6 +52,6 @@ resources: domain: aibrix.ai group: orchestration kind: KVCache - path: github.com/aibrix/aibrix/api/orchestration/v1alpha1 + path: github.com/vllm-project/aibrix/api/orchestration/v1alpha1 version: v1alpha1 version: "3" diff --git a/README.md b/README.md index ef62e872..b295105b 100644 --- a/README.md +++ b/README.md @@ -26,7 +26,7 @@ To get started with AIBrix, clone this repository and follow the setup instructi ```shell # Local Testing -git clone https://github.com/aibrix/aibrix.git +git clone https://github.com/vllm-project/aibrix.git cd aibrix # Install nightly aibrix dependencies @@ -39,10 +39,10 @@ kubectl create -k config/default Install stable distribution ```shell # Install component dependencies -kubectl create -k "github.com/aibrix/aibrix/config/dependency?ref=v0.2.0" +kubectl create -k "github.com/vllm-project/aibrix/config/dependency?ref=v0.2.0" # Install aibrix components -kubectl create -k "github.com/aibrix/aibrix/config/overlays/release?ref=v0.2.0" +kubectl create -k "github.com/vllm-project/aibrix/config/overlays/release?ref=v0.2.0" ``` ## Documentation @@ -51,16 +51,16 @@ For detailed documentation on installation, configuration, and usage, please vis ## Contributing -We welcome contributions from the community! Check out our [contributing guidelines](https://github.com/aibrix/aibrix/CONTRIBUTING.md) to see how you can make a difference. +We welcome contributions from the community! Check out our [contributing guidelines](https://github.com/vllm-project/aibrix/CONTRIBUTING.md) to see how you can make a difference. Slack Channel: https://vllm-dev.slack.com/archives/C07QP347J4D ## License -AIBrix is licensed under the [APACHE License](https://github.com/aibrix/aibrix/LICENSE.md). +AIBrix is licensed under the [APACHE License](https://github.com/vllm-project/aibrix/LICENSE.md). ## Support -If you have any questions or encounter any issues, please submit an issue on our [GitHub issues page](https://github.com/aibrix/aibrix/issues). +If you have any questions or encounter any issues, please submit an issue on our [GitHub issues page](https://github.com/vllm-project/aibrix/issues). Thank you for choosing AIBrix for your GenAI infrastructure needs! diff --git a/benchmarks/generator/README.md b/benchmarks/generator/README.md index 0d766d65..c7fdba15 100644 --- a/benchmarks/generator/README.md +++ b/benchmarks/generator/README.md @@ -19,7 +19,7 @@ python workload_generator.py --prompt-file $SHAREGPT_FILE_PATH --interval-ms 100 ### Generate a workload file based on workload patterns (synthetic patterns) -The can generate workload file based on synthetic traffic (qps), input lengths (prompt lengths) and output lengths (completion lengths) patterns. Currently we support 4 patterns (`'quick_rising`, `'slow_rising'`, `'slight_fluctuation'`, `'severe_fluctuation'`), described [here](https://github.com/aibrix/aibrix/blob/main/benchmarks/autoscaling/bench_workload_generator.py).: +The can generate workload file based on synthetic traffic (qps), input lengths (prompt lengths) and output lengths (completion lengths) patterns. Currently we support 4 patterns (`'quick_rising`, `'slow_rising'`, `'slight_fluctuation'`, `'severe_fluctuation'`), described [here](https://github.com/vllm-project/aibrix/blob/main/benchmarks/autoscaling/bench_workload_generator.py).: ```shell python workload_generator.py --prompt-file $SHAREGPT_FILE_PATH --interval-ms 1000 --duration-ms 300000 --trace-type synthetic --traffic-pattern "slight_fluctuation" --prompt-len-pattern "slight_fluctuation" --completion-len-pattern "slight_fluctuation" --model "Qwen/Qwen2.5-Coder-7B-Instruct" --output-dir "./output" --output-format jsonl ``` diff --git a/cmd/controllers/main.go b/cmd/controllers/main.go index caa3047d..cf8bd45d 100644 --- a/cmd/controllers/main.go +++ b/cmd/controllers/main.go @@ -25,11 +25,11 @@ import ( clientgoscheme "k8s.io/client-go/kubernetes/scheme" - autoscalingv1alpha1 "github.com/aibrix/aibrix/api/autoscaling/v1alpha1" - modelv1alpha1 "github.com/aibrix/aibrix/api/model/v1alpha1" - orchestrationv1alpha1 "github.com/aibrix/aibrix/api/orchestration/v1alpha1" - "github.com/aibrix/aibrix/pkg/features" rayclusterv1 "github.com/ray-project/kuberay/ray-operator/apis/ray/v1" + autoscalingv1alpha1 "github.com/vllm-project/aibrix/api/autoscaling/v1alpha1" + modelv1alpha1 "github.com/vllm-project/aibrix/api/model/v1alpha1" + orchestrationv1alpha1 "github.com/vllm-project/aibrix/api/orchestration/v1alpha1" + "github.com/vllm-project/aibrix/pkg/features" metav1 "k8s.io/apimachinery/pkg/apis/meta/v1" "k8s.io/apimachinery/pkg/runtime" @@ -45,9 +45,9 @@ import ( metricsserver "sigs.k8s.io/controller-runtime/pkg/metrics/server" "sigs.k8s.io/controller-runtime/pkg/webhook" - "github.com/aibrix/aibrix/pkg/cache" - "github.com/aibrix/aibrix/pkg/config" - "github.com/aibrix/aibrix/pkg/controller" + "github.com/vllm-project/aibrix/pkg/cache" + "github.com/vllm-project/aibrix/pkg/config" + "github.com/vllm-project/aibrix/pkg/controller" //+kubebuilder:scaffold:imports ) diff --git a/cmd/metadata/main.go b/cmd/metadata/main.go index debfb8fb..977b5fe0 100644 --- a/cmd/metadata/main.go +++ b/cmd/metadata/main.go @@ -17,8 +17,8 @@ limitations under the License. package main import ( - "github.com/aibrix/aibrix/pkg/metadata" - "github.com/aibrix/aibrix/pkg/utils" + "github.com/vllm-project/aibrix/pkg/metadata" + "github.com/vllm-project/aibrix/pkg/utils" "k8s.io/klog/v2" ) diff --git a/cmd/plugins/main.go b/cmd/plugins/main.go index 47a16c0f..5ea4a7d4 100644 --- a/cmd/plugins/main.go +++ b/cmd/plugins/main.go @@ -31,10 +31,10 @@ import ( "k8s.io/client-go/tools/clientcmd" "k8s.io/klog/v2" - "github.com/aibrix/aibrix/pkg/cache" - "github.com/aibrix/aibrix/pkg/plugins/gateway" - "github.com/aibrix/aibrix/pkg/utils" extProcPb "github.com/envoyproxy/go-control-plane/envoy/service/ext_proc/v3" + "github.com/vllm-project/aibrix/pkg/cache" + "github.com/vllm-project/aibrix/pkg/plugins/gateway" + "github.com/vllm-project/aibrix/pkg/utils" healthPb "google.golang.org/grpc/health/grpc_health_v1" ) diff --git a/development/README.md b/development/README.md index b3537546..12c222a0 100644 --- a/development/README.md +++ b/development/README.md @@ -4,8 +4,8 @@ Here are some essential resources for anyone interested in AIBrix: -- **Documentation and Tutorials**: [View Tutorials](https://github.com/aibrix/aibrix/tree/main/development/tutorial) -- **Issue Tracker**: [View Issues](https://github.com/aibrix/aibrix/issues) +- **Documentation and Tutorials**: [View Tutorials](https://github.com/vllm-project/aibrix/tree/main/development/tutorial) +- **Issue Tracker**: [View Issues](https://github.com/vllm-project/aibrix/issues) - **Project Roadmap**: TODO Additional resources for contributors: @@ -35,7 +35,7 @@ Alternatively, you can use [Kind](https://kind.sigs.k8s.io/) or [Minikube](https - Clone your fork locally: ```sh - git clone https://github.com/aibrix/aibrix.git + git clone https://github.com/vllm-project/aibrix.git cd aibrix ``` diff --git a/development/tutorials/distributed/README.md b/development/tutorials/distributed/README.md index 997812c4..9d2fd4a7 100644 --- a/development/tutorials/distributed/README.md +++ b/development/tutorials/distributed/README.md @@ -12,7 +12,7 @@ COPY utils.py /usr/local/lib/python3.12/dist-packages/vllm/executor/ray_utils.py ENTRYPOINT [""] ``` -> Note: copy uitls.py from upstream version and remove the placement group validation logic. See [#228](https://github.com/aibrix/aibrix/issues/228) for more details. +> Note: copy uitls.py from upstream version and remove the placement group validation logic. See [#228](https://github.com/vllm-project/aibrix/issues/228) for more details. > Note: No need to downgrade ray to v2.10.0. Seem only ray-project/ray image has issues. Container Image Combination which supports the distributed multi-host inference. diff --git a/docs/source/community/contribution.rst b/docs/source/community/contribution.rst index 7bd3931c..005fcc8e 100644 --- a/docs/source/community/contribution.rst +++ b/docs/source/community/contribution.rst @@ -12,7 +12,7 @@ To ensure a smooth process, please follow a few simple guidelines. Getting Started --------------- -Please make sure to read and observe our `Code of Conduct `_. +Please make sure to read and observe our `Code of Conduct `_. Make your first contribution @@ -31,8 +31,8 @@ Good Start issues To find AIBrix issues that make good entry points: -- Start with issues labeled `good first issue `_. -- For issues that require deeper knowledge of one or more technical aspects, look at issues labeled `help wanted `_. +- Start with issues labeled `good first issue `_. +- For issues that require deeper knowledge of one or more technical aspects, look at issues labeled `help wanted `_. Joining the community --------------------- diff --git a/docs/source/community/research.rst b/docs/source/community/research.rst index 21741779..6c91ad91 100644 --- a/docs/source/community/research.rst +++ b/docs/source/community/research.rst @@ -13,7 +13,7 @@ Opportunities for Research - **Unresolved Production Challenges**: The AI production environment presents numerous unresolved problems, from efficient resource allocation to scalable inference architectures. We can provide case studies and real-world scenarios for researchers interested in exploring and solving these challenges. -- **Research Paper Implementation**: Some research ideas have been integrated into AIBrix, making it a practical landing ground for cutting-edge innovations. We regularly publish `Help Wanted `_ issues on our GitHub, highlighting open research opportunities—feel free to take a look and contribute! +- **Research Paper Implementation**: Some research ideas have been integrated into AIBrix, making it a practical landing ground for cutting-edge innovations. We regularly publish `Help Wanted `_ issues on our GitHub, highlighting open research opportunities—feel free to take a look and contribute! - **AIBrix as a Research Testbed**: Our system is designed to serve as a research testbed for system-level problems in AI infrastructure. Whether it's testing new scheduling algorithms, optimizing inference latency, or improving resource efficiency, we provide hands-on support to help set up experiments and validate hypotheses. diff --git a/docs/source/conf.py b/docs/source/conf.py index 5a397b7f..bc9e5848 100644 --- a/docs/source/conf.py +++ b/docs/source/conf.py @@ -45,7 +45,7 @@ html_static_path = ['_static'] html_theme_options = { # repository level setting - 'repository_url': 'https://github.com/aibrix/aibrix', + 'repository_url': 'https://github.com/vllm-project/aibrix', "use_repository_button": True, "repository_branch": "main", 'path_to_docs': 'docs/source', diff --git a/docs/source/development/release.rst b/docs/source/development/release.rst index f9751c3a..a5fe2c55 100644 --- a/docs/source/development/release.rst +++ b/docs/source/development/release.rst @@ -45,7 +45,7 @@ for patch release, we do not rebase ``main`` because it will introduce new featu Cut a PR -------- -Make sure the manifest images tags and updated and python version is updated. A sample PR is `Cut v0.1.0-rc.5 release `_. +Make sure the manifest images tags and updated and python version is updated. A sample PR is `Cut v0.1.0-rc.5 release `_. Merge the PR. .. note:: @@ -91,7 +91,7 @@ Monitor the pipeline's progress to ensure it completes successfully Publish the release on Github ----------------------------- -Release pipeline will cut a draft pre-release in `Github Releases `_. +Release pipeline will cut a draft pre-release in `Github Releases `_. Go to the "Releases" section in the repository, select the draft release corresponding to the tag you created. Include release notes summarizing the changes (new features, bug fixes, breaking changes, etc.). Optionally attach binaries, documentation, or other assets. In the end, let's publish the release. @@ -120,4 +120,4 @@ we need to retag the images and push to VKE Container Registry. Update released tags in main branch docs ---------------------------------------- -A sample PR is `here `_. +A sample PR is `here `_. diff --git a/docs/source/features/autoscaling/metric-based-autoscaling.rst b/docs/source/features/autoscaling/metric-based-autoscaling.rst index 1d272e22..084f40f5 100644 --- a/docs/source/features/autoscaling/metric-based-autoscaling.rst +++ b/docs/source/features/autoscaling/metric-based-autoscaling.rst @@ -40,7 +40,7 @@ All the sample files can be found in the following directory. .. code-block:: bash - https://github.com/aibrix/aibrix/tree/main/samples/autoscaling + https://github.com/vllm-project/aibrix/tree/main/samples/autoscaling Example HPA yaml config ^^^^^^^^^^^^^^^^^^^^^^^ diff --git a/docs/source/features/autoscaling/optimizer-based-autoscaling.rst b/docs/source/features/autoscaling/optimizer-based-autoscaling.rst index a3844766..e63e1b8e 100644 --- a/docs/source/features/autoscaling/optimizer-based-autoscaling.rst +++ b/docs/source/features/autoscaling/optimizer-based-autoscaling.rst @@ -22,7 +22,7 @@ How It Works Step 1: Offline GPU-Model Benchmark per Input-Output Pattern ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ -Benchmark model. For each type of GPU, run ``aibrix_benchmark``. See `benchmark.sh `_ for more options. +Benchmark model. For each type of GPU, run ``aibrix_benchmark``. See `benchmark.sh `_ for more options. .. code-block:: bash @@ -64,7 +64,7 @@ All the sample files can be found in the following directory. .. code-block:: bash - https://github.com/aibrix/aibrix/tree/main/samples/autoscaling + https://github.com/vllm-project/aibrix/tree/main/samples/autoscaling Example Optimizer-based KPA yaml config ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ diff --git a/docs/source/features/heterogeneous-gpu.rst b/docs/source/features/heterogeneous-gpu.rst index 74881d2a..2ac204ba 100644 --- a/docs/source/features/heterogeneous-gpu.rst +++ b/docs/source/features/heterogeneous-gpu.rst @@ -23,7 +23,7 @@ Example Step 1: Deploy the heterogeneous deployments. One deployment and corresponding PodAutoscaler should be deployed for each GPU type. -See `sample heterogeneous configuration `_ for an example of heterogeneous configuration composed of two GPU types. The following codes +See `sample heterogeneous configuration `_ for an example of heterogeneous configuration composed of two GPU types. The following codes deploy heterogeneous deployments using L20 and V100 GPU. .. code-block:: bash @@ -59,9 +59,9 @@ Step 2: Install aibrix python module: The GPU Optimizer runs continuously in the background, dynamically adjusting GPU allocation for each model based on workload patterns. Note that GPU optimizer requires offline inference performance benchmark data for each type of GPU on each specific LLM model. -If local heterogeneous deployments is used, you can find the prepared benchmark data under `python/aibrix/aibrix/gpu_optimizer/optimizer/profiling/result/ `_ and skip Step 3. See :ref:`Development` for details on deploying a local heterogeneous deployments. +If local heterogeneous deployments is used, you can find the prepared benchmark data under `python/aibrix/aibrix/gpu_optimizer/optimizer/profiling/result/ `_ and skip Step 3. See :ref:`Development` for details on deploying a local heterogeneous deployments. -Step 3: Benchmark model. For each type of GPU, run ``aibrix_benchmark``. See `benchmark.sh `_ for more options. +Step 3: Benchmark model. For each type of GPU, run ``aibrix_benchmark``. See `benchmark.sh `_ for more options. .. code-block:: bash diff --git a/docs/source/getting_started/installation/installation.rst b/docs/source/getting_started/installation/installation.rst index 48230498..1179b849 100644 --- a/docs/source/getting_started/installation/installation.rst +++ b/docs/source/getting_started/installation/installation.rst @@ -23,10 +23,10 @@ Stable Version .. code:: bash # Install component dependencies - kubectl apply -f https://github.com/aibrix/aibrix/releases/download/v0.2.0/aibrix-dependency-v0.2.0.yaml + kubectl apply -f https://github.com/vllm-project/aibrix/releases/download/v0.2.0/aibrix-dependency-v0.2.0.yaml # Install aibrix components - kubectl apply -f https://github.com/aibrix/aibrix/releases/download/v0.2.0/aibrix-core-v0.2.0.yaml + kubectl apply -f https://github.com/vllm-project/aibrix/releases/download/v0.2.0/aibrix-core-v0.2.0.yaml Nightly Version @@ -35,7 +35,7 @@ Nightly Version .. code:: bash # clone the latest repo - git clone https://github.com/aibrix/aibrix.git + git clone https://github.com/vllm-project/aibrix.git cd aibrix # Install component dependencies diff --git a/docs/source/getting_started/installation/lambda.rst b/docs/source/getting_started/installation/lambda.rst index 9f51397a..ccaf5681 100644 --- a/docs/source/getting_started/installation/lambda.rst +++ b/docs/source/getting_started/installation/lambda.rst @@ -120,10 +120,10 @@ Once the cluster is up and running, install AIBrix components: .. code-block:: bash # install dependencies - kubectl create -k "github.com/aibrix/aibrix/config/dependency?ref=v0.2.0" + kubectl create -k "github.com/vllm-project/aibrix/config/dependency?ref=v0.2.0" # install core components - kubectl create -k "github.com/aibrix/aibrix/config/overlays/release?ref=v0.2.0" + kubectl create -k "github.com/vllm-project/aibrix/config/overlays/release?ref=v0.2.0" Verify that the AIBrix components are installed successfully: diff --git a/docs/source/getting_started/quickstart.rst b/docs/source/getting_started/quickstart.rst index ebf52f7c..f693374c 100644 --- a/docs/source/getting_started/quickstart.rst +++ b/docs/source/getting_started/quickstart.rst @@ -14,8 +14,8 @@ Get your kubernetes cluster ready, run following commands to install aibrix comp .. code-block:: bash - kubectl apply -f https://github.com/aibrix/aibrix/releases/download/v0.2.0/aibrix-dependency-v0.2.0.yaml - kubectl apply -f https://github.com/aibrix/aibrix/releases/download/v0.2.0/aibrix-core-v0.2.0.yaml + kubectl apply -f https://github.com/vllm-project/aibrix/releases/download/v0.2.0/aibrix-dependency-v0.2.0.yaml + kubectl apply -f https://github.com/vllm-project/aibrix/releases/download/v0.2.0/aibrix-core-v0.2.0.yaml Wait for few minutes and run `kubectl get pods -n aibrix-system` to check pod status util they are ready. diff --git a/go.mod b/go.mod index b66c96bc..5cd92cb2 100644 --- a/go.mod +++ b/go.mod @@ -1,10 +1,11 @@ -module github.com/aibrix/aibrix +module github.com/vllm-project/aibrix go 1.22.5 toolchain go1.22.6 require ( + github.com/cespare/xxhash v1.1.0 github.com/envoyproxy/go-control-plane v0.12.0 github.com/go-playground/validator/v10 v10.22.1 github.com/google/uuid v1.6.0 @@ -13,6 +14,8 @@ require ( github.com/onsi/ginkgo/v2 v2.17.2 github.com/onsi/gomega v1.33.1 github.com/openai/openai-go v0.1.0-alpha.37 + github.com/pkoukk/tiktoken-go v0.1.7 + github.com/pkoukk/tiktoken-go-loader v0.0.1 github.com/prometheus/client_golang v1.19.1 github.com/prometheus/client_model v0.6.1 github.com/prometheus/common v0.55.0 @@ -35,7 +38,6 @@ require ( require ( github.com/beorn7/perks v1.0.1 // indirect - github.com/cespare/xxhash v1.1.0 // indirect github.com/cespare/xxhash/v2 v2.3.0 // indirect github.com/cncf/xds/go v0.0.0-20240423153145-555b57ec207b // indirect github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc // indirect @@ -74,8 +76,6 @@ require ( github.com/mwitkow/go-conntrack v0.0.0-20190716064945-2f068394615f // indirect github.com/nxadm/tail v1.4.8 // indirect github.com/pkg/errors v0.9.1 // indirect - github.com/pkoukk/tiktoken-go v0.1.7 // indirect - github.com/pkoukk/tiktoken-go-loader v0.0.1 // indirect github.com/pmezard/go-difflib v1.0.0 // indirect github.com/prometheus/procfs v0.15.1 // indirect github.com/spf13/pflag v1.0.5 // indirect diff --git a/go.sum b/go.sum index d8c269e5..108cd729 100644 --- a/go.sum +++ b/go.sum @@ -1,5 +1,6 @@ dario.cat/mergo v0.3.16 h1:wrt7QIfeqlABnUvmf9WpFwB0mGBwtySAJKTgCpnsbOE= dario.cat/mergo v0.3.16/go.mod h1:WBLT9ZmE3lPoWsEzCh9LPo3TiwVN+ZKEjmz+hD27ysY= +github.com/OneOfOne/xxhash v1.2.2 h1:KMrpdQIwFcEqXDklaen+P1axHaj9BSKzvpUUfnHldSE= github.com/OneOfOne/xxhash v1.2.2/go.mod h1:HSdplMjZKSmBqAxg5vPj2TmRDmfkzw+cTzAElWljhcU= github.com/beorn7/perks v1.0.1 h1:VlbKKnNfV8bJzeqoa4cOKqO6bYr3WgKZxO8Z16+hsOM= github.com/beorn7/perks v1.0.1/go.mod h1:G2ZrVWU2WbWT9wwq4/hrbKbnv/1ERSJQ0ibhJ6rlkpw= @@ -156,6 +157,7 @@ github.com/redis/go-redis/v9 v9.6.1 h1:HHDteefn6ZkTtY5fGUE8tj8uy85AHk6zP7CpzIAM0 github.com/redis/go-redis/v9 v9.6.1/go.mod h1:0C0c6ycQsdpVNQpxb1njEQIqkx5UcsM8FJCQLgE9+RA= github.com/rogpeppe/go-internal v1.11.0 h1:cWPaGQEPrBb5/AsnsZesgZZ9yb1OQ+GOISoDNXVBh4M= github.com/rogpeppe/go-internal v1.11.0/go.mod h1:ddIwULY96R17DhadqLgMfk9H9tvdUzkipdSkR5nkCZA= +github.com/spaolacci/murmur3 v0.0.0-20180118202830-f09979ecbc72 h1:qLC7fQah7D6K1B0ujays3HV9gkFtllcxhzImRR7ArPQ= github.com/spaolacci/murmur3 v0.0.0-20180118202830-f09979ecbc72/go.mod h1:JwIasOWyU6f++ZhiEuf87xNszmSA2myDM2Kzu9HwQUA= github.com/spf13/pflag v1.0.5 h1:iy+VFUOCP1a+8yFto/drg2CJ5u0yRoB7fZw3DKv/JXA= github.com/spf13/pflag v1.0.5/go.mod h1:McXfInJRrz4CZXVZOBLb0bTZqETkiAhM9Iw0y3An2Bg= diff --git a/hack/update-codegen.sh b/hack/update-codegen.sh index c44de536..8a3ec483 100755 --- a/hack/update-codegen.sh +++ b/hack/update-codegen.sh @@ -8,7 +8,7 @@ set -o nounset set -o pipefail SCRIPT_ROOT=$(dirname "${BASH_SOURCE[0]}")/.. -ROOT_PKG=github.com/aibrix/aibrix +ROOT_PKG=github.com/vllm-project/aibrix # Grab code-generator version from go.sum CODEGEN_VERSION=$(grep 'k8s.io/code-generator' go.sum | awk '{print $2}' | sed 's/\/go.mod//g' | head -1) @@ -45,7 +45,7 @@ chmod +x ${CODEGEN_PKG}/generate-internal-groups.sh # cd ${SCRIPT_ROOT} ${CODEGEN_PKG}/generate-groups.sh applyconfiguration,client,informer,lister \ - github.com/aibrix/aibrix/pkg/client github.com/aibrix/aibrix/api \ + github.com/vllm-project/aibrix/pkg/client github.com/vllm-project/aibrix/api \ "model:v1alpha1 autoscaling:v1alpha1 orchestration:v1alpha1" \ --output-base "${TEMP_DIR}" \ --go-header-file hack/boilerplate.go.txt diff --git a/pkg/cache/cache.go b/pkg/cache/cache.go index 4a6abbb8..94093cf3 100644 --- a/pkg/cache/cache.go +++ b/pkg/cache/cache.go @@ -27,8 +27,8 @@ import ( "sync/atomic" "time" - crdinformers "github.com/aibrix/aibrix/pkg/client/informers/externalversions" "github.com/redis/go-redis/v9" + crdinformers "github.com/vllm-project/aibrix/pkg/client/informers/externalversions" v1 "k8s.io/api/core/v1" "k8s.io/apimachinery/pkg/util/runtime" "k8s.io/client-go/informers" @@ -37,13 +37,13 @@ import ( "k8s.io/client-go/tools/cache" "k8s.io/klog/v2" - modelv1alpha1 "github.com/aibrix/aibrix/api/model/v1alpha1" - v1alpha1 "github.com/aibrix/aibrix/pkg/client/clientset/versioned" - v1alpha1scheme "github.com/aibrix/aibrix/pkg/client/clientset/versioned/scheme" - "github.com/aibrix/aibrix/pkg/metrics" - "github.com/aibrix/aibrix/pkg/utils" prometheusv1 "github.com/prometheus/client_golang/api/prometheus/v1" dto "github.com/prometheus/client_model/go" + modelv1alpha1 "github.com/vllm-project/aibrix/api/model/v1alpha1" + v1alpha1 "github.com/vllm-project/aibrix/pkg/client/clientset/versioned" + v1alpha1scheme "github.com/vllm-project/aibrix/pkg/client/clientset/versioned/scheme" + "github.com/vllm-project/aibrix/pkg/metrics" + "github.com/vllm-project/aibrix/pkg/utils" "k8s.io/client-go/kubernetes/scheme" ) diff --git a/pkg/client/applyconfiguration/autoscaling/v1alpha1/metricsource.go b/pkg/client/applyconfiguration/autoscaling/v1alpha1/metricsource.go index 0252c367..4051e6a8 100644 --- a/pkg/client/applyconfiguration/autoscaling/v1alpha1/metricsource.go +++ b/pkg/client/applyconfiguration/autoscaling/v1alpha1/metricsource.go @@ -18,7 +18,7 @@ limitations under the License. package v1alpha1 import ( - v1alpha1 "github.com/aibrix/aibrix/api/autoscaling/v1alpha1" + v1alpha1 "github.com/vllm-project/aibrix/api/autoscaling/v1alpha1" ) // MetricSourceApplyConfiguration represents an declarative configuration of the MetricSource type for use diff --git a/pkg/client/applyconfiguration/autoscaling/v1alpha1/podautoscalerspec.go b/pkg/client/applyconfiguration/autoscaling/v1alpha1/podautoscalerspec.go index 1d0c362e..f3ce72d0 100644 --- a/pkg/client/applyconfiguration/autoscaling/v1alpha1/podautoscalerspec.go +++ b/pkg/client/applyconfiguration/autoscaling/v1alpha1/podautoscalerspec.go @@ -18,7 +18,7 @@ limitations under the License. package v1alpha1 import ( - autoscalingv1alpha1 "github.com/aibrix/aibrix/api/autoscaling/v1alpha1" + autoscalingv1alpha1 "github.com/vllm-project/aibrix/api/autoscaling/v1alpha1" v1 "k8s.io/api/core/v1" ) diff --git a/pkg/client/applyconfiguration/model/v1alpha1/modeladapterstatus.go b/pkg/client/applyconfiguration/model/v1alpha1/modeladapterstatus.go index b280a9fc..e567f7da 100644 --- a/pkg/client/applyconfiguration/model/v1alpha1/modeladapterstatus.go +++ b/pkg/client/applyconfiguration/model/v1alpha1/modeladapterstatus.go @@ -18,7 +18,7 @@ limitations under the License. package v1alpha1 import ( - v1alpha1 "github.com/aibrix/aibrix/api/model/v1alpha1" + v1alpha1 "github.com/vllm-project/aibrix/api/model/v1alpha1" v1 "k8s.io/apimachinery/pkg/apis/meta/v1" ) diff --git a/pkg/client/applyconfiguration/orchestration/v1alpha1/rayclusterfleetcondition.go b/pkg/client/applyconfiguration/orchestration/v1alpha1/rayclusterfleetcondition.go index f579bd4f..2954a634 100644 --- a/pkg/client/applyconfiguration/orchestration/v1alpha1/rayclusterfleetcondition.go +++ b/pkg/client/applyconfiguration/orchestration/v1alpha1/rayclusterfleetcondition.go @@ -18,7 +18,7 @@ limitations under the License. package v1alpha1 import ( - v1alpha1 "github.com/aibrix/aibrix/api/orchestration/v1alpha1" + v1alpha1 "github.com/vllm-project/aibrix/api/orchestration/v1alpha1" v1 "k8s.io/api/core/v1" metav1 "k8s.io/apimachinery/pkg/apis/meta/v1" ) diff --git a/pkg/client/applyconfiguration/utils.go b/pkg/client/applyconfiguration/utils.go index 69071c37..cbf6e318 100644 --- a/pkg/client/applyconfiguration/utils.go +++ b/pkg/client/applyconfiguration/utils.go @@ -18,12 +18,12 @@ limitations under the License. package applyconfiguration import ( - v1alpha1 "github.com/aibrix/aibrix/api/autoscaling/v1alpha1" - modelv1alpha1 "github.com/aibrix/aibrix/api/model/v1alpha1" - orchestrationv1alpha1 "github.com/aibrix/aibrix/api/orchestration/v1alpha1" - autoscalingv1alpha1 "github.com/aibrix/aibrix/pkg/client/applyconfiguration/autoscaling/v1alpha1" - applyconfigurationmodelv1alpha1 "github.com/aibrix/aibrix/pkg/client/applyconfiguration/model/v1alpha1" - applyconfigurationorchestrationv1alpha1 "github.com/aibrix/aibrix/pkg/client/applyconfiguration/orchestration/v1alpha1" + v1alpha1 "github.com/vllm-project/aibrix/api/autoscaling/v1alpha1" + modelv1alpha1 "github.com/vllm-project/aibrix/api/model/v1alpha1" + orchestrationv1alpha1 "github.com/vllm-project/aibrix/api/orchestration/v1alpha1" + autoscalingv1alpha1 "github.com/vllm-project/aibrix/pkg/client/applyconfiguration/autoscaling/v1alpha1" + applyconfigurationmodelv1alpha1 "github.com/vllm-project/aibrix/pkg/client/applyconfiguration/model/v1alpha1" + applyconfigurationorchestrationv1alpha1 "github.com/vllm-project/aibrix/pkg/client/applyconfiguration/orchestration/v1alpha1" schema "k8s.io/apimachinery/pkg/runtime/schema" ) diff --git a/pkg/client/clientset/versioned/clientset.go b/pkg/client/clientset/versioned/clientset.go index b64bcab6..1c827faa 100644 --- a/pkg/client/clientset/versioned/clientset.go +++ b/pkg/client/clientset/versioned/clientset.go @@ -21,9 +21,9 @@ import ( "fmt" "net/http" - autoscalingv1alpha1 "github.com/aibrix/aibrix/pkg/client/clientset/versioned/typed/autoscaling/v1alpha1" - modelv1alpha1 "github.com/aibrix/aibrix/pkg/client/clientset/versioned/typed/model/v1alpha1" - orchestrationv1alpha1 "github.com/aibrix/aibrix/pkg/client/clientset/versioned/typed/orchestration/v1alpha1" + autoscalingv1alpha1 "github.com/vllm-project/aibrix/pkg/client/clientset/versioned/typed/autoscaling/v1alpha1" + modelv1alpha1 "github.com/vllm-project/aibrix/pkg/client/clientset/versioned/typed/model/v1alpha1" + orchestrationv1alpha1 "github.com/vllm-project/aibrix/pkg/client/clientset/versioned/typed/orchestration/v1alpha1" discovery "k8s.io/client-go/discovery" rest "k8s.io/client-go/rest" flowcontrol "k8s.io/client-go/util/flowcontrol" diff --git a/pkg/client/clientset/versioned/fake/clientset_generated.go b/pkg/client/clientset/versioned/fake/clientset_generated.go index 978b9f09..0b2c40e9 100644 --- a/pkg/client/clientset/versioned/fake/clientset_generated.go +++ b/pkg/client/clientset/versioned/fake/clientset_generated.go @@ -18,13 +18,13 @@ limitations under the License. package fake import ( - clientset "github.com/aibrix/aibrix/pkg/client/clientset/versioned" - autoscalingv1alpha1 "github.com/aibrix/aibrix/pkg/client/clientset/versioned/typed/autoscaling/v1alpha1" - fakeautoscalingv1alpha1 "github.com/aibrix/aibrix/pkg/client/clientset/versioned/typed/autoscaling/v1alpha1/fake" - modelv1alpha1 "github.com/aibrix/aibrix/pkg/client/clientset/versioned/typed/model/v1alpha1" - fakemodelv1alpha1 "github.com/aibrix/aibrix/pkg/client/clientset/versioned/typed/model/v1alpha1/fake" - orchestrationv1alpha1 "github.com/aibrix/aibrix/pkg/client/clientset/versioned/typed/orchestration/v1alpha1" - fakeorchestrationv1alpha1 "github.com/aibrix/aibrix/pkg/client/clientset/versioned/typed/orchestration/v1alpha1/fake" + clientset "github.com/vllm-project/aibrix/pkg/client/clientset/versioned" + autoscalingv1alpha1 "github.com/vllm-project/aibrix/pkg/client/clientset/versioned/typed/autoscaling/v1alpha1" + fakeautoscalingv1alpha1 "github.com/vllm-project/aibrix/pkg/client/clientset/versioned/typed/autoscaling/v1alpha1/fake" + modelv1alpha1 "github.com/vllm-project/aibrix/pkg/client/clientset/versioned/typed/model/v1alpha1" + fakemodelv1alpha1 "github.com/vllm-project/aibrix/pkg/client/clientset/versioned/typed/model/v1alpha1/fake" + orchestrationv1alpha1 "github.com/vllm-project/aibrix/pkg/client/clientset/versioned/typed/orchestration/v1alpha1" + fakeorchestrationv1alpha1 "github.com/vllm-project/aibrix/pkg/client/clientset/versioned/typed/orchestration/v1alpha1/fake" "k8s.io/apimachinery/pkg/runtime" "k8s.io/apimachinery/pkg/watch" "k8s.io/client-go/discovery" diff --git a/pkg/client/clientset/versioned/fake/register.go b/pkg/client/clientset/versioned/fake/register.go index f3e17e8b..73813e63 100644 --- a/pkg/client/clientset/versioned/fake/register.go +++ b/pkg/client/clientset/versioned/fake/register.go @@ -18,9 +18,9 @@ limitations under the License. package fake import ( - autoscalingv1alpha1 "github.com/aibrix/aibrix/api/autoscaling/v1alpha1" - modelv1alpha1 "github.com/aibrix/aibrix/api/model/v1alpha1" - orchestrationv1alpha1 "github.com/aibrix/aibrix/api/orchestration/v1alpha1" + autoscalingv1alpha1 "github.com/vllm-project/aibrix/api/autoscaling/v1alpha1" + modelv1alpha1 "github.com/vllm-project/aibrix/api/model/v1alpha1" + orchestrationv1alpha1 "github.com/vllm-project/aibrix/api/orchestration/v1alpha1" v1 "k8s.io/apimachinery/pkg/apis/meta/v1" runtime "k8s.io/apimachinery/pkg/runtime" schema "k8s.io/apimachinery/pkg/runtime/schema" diff --git a/pkg/client/clientset/versioned/scheme/register.go b/pkg/client/clientset/versioned/scheme/register.go index 071fac13..7f1e5707 100644 --- a/pkg/client/clientset/versioned/scheme/register.go +++ b/pkg/client/clientset/versioned/scheme/register.go @@ -18,9 +18,9 @@ limitations under the License. package scheme import ( - autoscalingv1alpha1 "github.com/aibrix/aibrix/api/autoscaling/v1alpha1" - modelv1alpha1 "github.com/aibrix/aibrix/api/model/v1alpha1" - orchestrationv1alpha1 "github.com/aibrix/aibrix/api/orchestration/v1alpha1" + autoscalingv1alpha1 "github.com/vllm-project/aibrix/api/autoscaling/v1alpha1" + modelv1alpha1 "github.com/vllm-project/aibrix/api/model/v1alpha1" + orchestrationv1alpha1 "github.com/vllm-project/aibrix/api/orchestration/v1alpha1" v1 "k8s.io/apimachinery/pkg/apis/meta/v1" runtime "k8s.io/apimachinery/pkg/runtime" schema "k8s.io/apimachinery/pkg/runtime/schema" diff --git a/pkg/client/clientset/versioned/typed/autoscaling/v1alpha1/autoscaling_client.go b/pkg/client/clientset/versioned/typed/autoscaling/v1alpha1/autoscaling_client.go index 1291ae1a..37d4e341 100644 --- a/pkg/client/clientset/versioned/typed/autoscaling/v1alpha1/autoscaling_client.go +++ b/pkg/client/clientset/versioned/typed/autoscaling/v1alpha1/autoscaling_client.go @@ -20,8 +20,8 @@ package v1alpha1 import ( "net/http" - v1alpha1 "github.com/aibrix/aibrix/api/autoscaling/v1alpha1" - "github.com/aibrix/aibrix/pkg/client/clientset/versioned/scheme" + v1alpha1 "github.com/vllm-project/aibrix/api/autoscaling/v1alpha1" + "github.com/vllm-project/aibrix/pkg/client/clientset/versioned/scheme" rest "k8s.io/client-go/rest" ) diff --git a/pkg/client/clientset/versioned/typed/autoscaling/v1alpha1/fake/fake_autoscaling_client.go b/pkg/client/clientset/versioned/typed/autoscaling/v1alpha1/fake/fake_autoscaling_client.go index 0a5f2b73..b47bd547 100644 --- a/pkg/client/clientset/versioned/typed/autoscaling/v1alpha1/fake/fake_autoscaling_client.go +++ b/pkg/client/clientset/versioned/typed/autoscaling/v1alpha1/fake/fake_autoscaling_client.go @@ -18,7 +18,7 @@ limitations under the License. package fake import ( - v1alpha1 "github.com/aibrix/aibrix/pkg/client/clientset/versioned/typed/autoscaling/v1alpha1" + v1alpha1 "github.com/vllm-project/aibrix/pkg/client/clientset/versioned/typed/autoscaling/v1alpha1" rest "k8s.io/client-go/rest" testing "k8s.io/client-go/testing" ) diff --git a/pkg/client/clientset/versioned/typed/autoscaling/v1alpha1/fake/fake_podautoscaler.go b/pkg/client/clientset/versioned/typed/autoscaling/v1alpha1/fake/fake_podautoscaler.go index 59726832..e5e1f9c8 100644 --- a/pkg/client/clientset/versioned/typed/autoscaling/v1alpha1/fake/fake_podautoscaler.go +++ b/pkg/client/clientset/versioned/typed/autoscaling/v1alpha1/fake/fake_podautoscaler.go @@ -22,8 +22,8 @@ import ( json "encoding/json" "fmt" - v1alpha1 "github.com/aibrix/aibrix/api/autoscaling/v1alpha1" - autoscalingv1alpha1 "github.com/aibrix/aibrix/pkg/client/applyconfiguration/autoscaling/v1alpha1" + v1alpha1 "github.com/vllm-project/aibrix/api/autoscaling/v1alpha1" + autoscalingv1alpha1 "github.com/vllm-project/aibrix/pkg/client/applyconfiguration/autoscaling/v1alpha1" v1 "k8s.io/apimachinery/pkg/apis/meta/v1" labels "k8s.io/apimachinery/pkg/labels" types "k8s.io/apimachinery/pkg/types" diff --git a/pkg/client/clientset/versioned/typed/autoscaling/v1alpha1/podautoscaler.go b/pkg/client/clientset/versioned/typed/autoscaling/v1alpha1/podautoscaler.go index af7b2343..601e982b 100644 --- a/pkg/client/clientset/versioned/typed/autoscaling/v1alpha1/podautoscaler.go +++ b/pkg/client/clientset/versioned/typed/autoscaling/v1alpha1/podautoscaler.go @@ -23,9 +23,9 @@ import ( "fmt" "time" - v1alpha1 "github.com/aibrix/aibrix/api/autoscaling/v1alpha1" - autoscalingv1alpha1 "github.com/aibrix/aibrix/pkg/client/applyconfiguration/autoscaling/v1alpha1" - scheme "github.com/aibrix/aibrix/pkg/client/clientset/versioned/scheme" + v1alpha1 "github.com/vllm-project/aibrix/api/autoscaling/v1alpha1" + autoscalingv1alpha1 "github.com/vllm-project/aibrix/pkg/client/applyconfiguration/autoscaling/v1alpha1" + scheme "github.com/vllm-project/aibrix/pkg/client/clientset/versioned/scheme" v1 "k8s.io/apimachinery/pkg/apis/meta/v1" types "k8s.io/apimachinery/pkg/types" watch "k8s.io/apimachinery/pkg/watch" diff --git a/pkg/client/clientset/versioned/typed/model/v1alpha1/fake/fake_model_client.go b/pkg/client/clientset/versioned/typed/model/v1alpha1/fake/fake_model_client.go index 5891e94e..4d8cc4c3 100644 --- a/pkg/client/clientset/versioned/typed/model/v1alpha1/fake/fake_model_client.go +++ b/pkg/client/clientset/versioned/typed/model/v1alpha1/fake/fake_model_client.go @@ -18,7 +18,7 @@ limitations under the License. package fake import ( - v1alpha1 "github.com/aibrix/aibrix/pkg/client/clientset/versioned/typed/model/v1alpha1" + v1alpha1 "github.com/vllm-project/aibrix/pkg/client/clientset/versioned/typed/model/v1alpha1" rest "k8s.io/client-go/rest" testing "k8s.io/client-go/testing" ) diff --git a/pkg/client/clientset/versioned/typed/model/v1alpha1/fake/fake_modeladapter.go b/pkg/client/clientset/versioned/typed/model/v1alpha1/fake/fake_modeladapter.go index 6da26bf0..370dce0d 100644 --- a/pkg/client/clientset/versioned/typed/model/v1alpha1/fake/fake_modeladapter.go +++ b/pkg/client/clientset/versioned/typed/model/v1alpha1/fake/fake_modeladapter.go @@ -22,8 +22,8 @@ import ( json "encoding/json" "fmt" - v1alpha1 "github.com/aibrix/aibrix/api/model/v1alpha1" - modelv1alpha1 "github.com/aibrix/aibrix/pkg/client/applyconfiguration/model/v1alpha1" + v1alpha1 "github.com/vllm-project/aibrix/api/model/v1alpha1" + modelv1alpha1 "github.com/vllm-project/aibrix/pkg/client/applyconfiguration/model/v1alpha1" v1 "k8s.io/apimachinery/pkg/apis/meta/v1" labels "k8s.io/apimachinery/pkg/labels" types "k8s.io/apimachinery/pkg/types" diff --git a/pkg/client/clientset/versioned/typed/model/v1alpha1/model_client.go b/pkg/client/clientset/versioned/typed/model/v1alpha1/model_client.go index e3fcd4af..c334fd2e 100644 --- a/pkg/client/clientset/versioned/typed/model/v1alpha1/model_client.go +++ b/pkg/client/clientset/versioned/typed/model/v1alpha1/model_client.go @@ -20,8 +20,8 @@ package v1alpha1 import ( "net/http" - v1alpha1 "github.com/aibrix/aibrix/api/model/v1alpha1" - "github.com/aibrix/aibrix/pkg/client/clientset/versioned/scheme" + v1alpha1 "github.com/vllm-project/aibrix/api/model/v1alpha1" + "github.com/vllm-project/aibrix/pkg/client/clientset/versioned/scheme" rest "k8s.io/client-go/rest" ) diff --git a/pkg/client/clientset/versioned/typed/model/v1alpha1/modeladapter.go b/pkg/client/clientset/versioned/typed/model/v1alpha1/modeladapter.go index a08310b8..20c28b69 100644 --- a/pkg/client/clientset/versioned/typed/model/v1alpha1/modeladapter.go +++ b/pkg/client/clientset/versioned/typed/model/v1alpha1/modeladapter.go @@ -23,9 +23,9 @@ import ( "fmt" "time" - v1alpha1 "github.com/aibrix/aibrix/api/model/v1alpha1" - modelv1alpha1 "github.com/aibrix/aibrix/pkg/client/applyconfiguration/model/v1alpha1" - scheme "github.com/aibrix/aibrix/pkg/client/clientset/versioned/scheme" + v1alpha1 "github.com/vllm-project/aibrix/api/model/v1alpha1" + modelv1alpha1 "github.com/vllm-project/aibrix/pkg/client/applyconfiguration/model/v1alpha1" + scheme "github.com/vllm-project/aibrix/pkg/client/clientset/versioned/scheme" v1 "k8s.io/apimachinery/pkg/apis/meta/v1" types "k8s.io/apimachinery/pkg/types" watch "k8s.io/apimachinery/pkg/watch" diff --git a/pkg/client/clientset/versioned/typed/orchestration/v1alpha1/fake/fake_orchestration_client.go b/pkg/client/clientset/versioned/typed/orchestration/v1alpha1/fake/fake_orchestration_client.go index 2b7fbf60..c0e94fd7 100644 --- a/pkg/client/clientset/versioned/typed/orchestration/v1alpha1/fake/fake_orchestration_client.go +++ b/pkg/client/clientset/versioned/typed/orchestration/v1alpha1/fake/fake_orchestration_client.go @@ -18,7 +18,7 @@ limitations under the License. package fake import ( - v1alpha1 "github.com/aibrix/aibrix/pkg/client/clientset/versioned/typed/orchestration/v1alpha1" + v1alpha1 "github.com/vllm-project/aibrix/pkg/client/clientset/versioned/typed/orchestration/v1alpha1" rest "k8s.io/client-go/rest" testing "k8s.io/client-go/testing" ) diff --git a/pkg/client/clientset/versioned/typed/orchestration/v1alpha1/fake/fake_rayclusterfleet.go b/pkg/client/clientset/versioned/typed/orchestration/v1alpha1/fake/fake_rayclusterfleet.go index ec61070c..5a465016 100644 --- a/pkg/client/clientset/versioned/typed/orchestration/v1alpha1/fake/fake_rayclusterfleet.go +++ b/pkg/client/clientset/versioned/typed/orchestration/v1alpha1/fake/fake_rayclusterfleet.go @@ -22,8 +22,8 @@ import ( json "encoding/json" "fmt" - v1alpha1 "github.com/aibrix/aibrix/api/orchestration/v1alpha1" - orchestrationv1alpha1 "github.com/aibrix/aibrix/pkg/client/applyconfiguration/orchestration/v1alpha1" + v1alpha1 "github.com/vllm-project/aibrix/api/orchestration/v1alpha1" + orchestrationv1alpha1 "github.com/vllm-project/aibrix/pkg/client/applyconfiguration/orchestration/v1alpha1" v1 "k8s.io/apimachinery/pkg/apis/meta/v1" labels "k8s.io/apimachinery/pkg/labels" types "k8s.io/apimachinery/pkg/types" diff --git a/pkg/client/clientset/versioned/typed/orchestration/v1alpha1/fake/fake_rayclusterreplicaset.go b/pkg/client/clientset/versioned/typed/orchestration/v1alpha1/fake/fake_rayclusterreplicaset.go index 812ad4ae..7ef32888 100644 --- a/pkg/client/clientset/versioned/typed/orchestration/v1alpha1/fake/fake_rayclusterreplicaset.go +++ b/pkg/client/clientset/versioned/typed/orchestration/v1alpha1/fake/fake_rayclusterreplicaset.go @@ -22,8 +22,8 @@ import ( json "encoding/json" "fmt" - v1alpha1 "github.com/aibrix/aibrix/api/orchestration/v1alpha1" - orchestrationv1alpha1 "github.com/aibrix/aibrix/pkg/client/applyconfiguration/orchestration/v1alpha1" + v1alpha1 "github.com/vllm-project/aibrix/api/orchestration/v1alpha1" + orchestrationv1alpha1 "github.com/vllm-project/aibrix/pkg/client/applyconfiguration/orchestration/v1alpha1" v1 "k8s.io/apimachinery/pkg/apis/meta/v1" labels "k8s.io/apimachinery/pkg/labels" types "k8s.io/apimachinery/pkg/types" diff --git a/pkg/client/clientset/versioned/typed/orchestration/v1alpha1/orchestration_client.go b/pkg/client/clientset/versioned/typed/orchestration/v1alpha1/orchestration_client.go index 65490429..2ef56617 100644 --- a/pkg/client/clientset/versioned/typed/orchestration/v1alpha1/orchestration_client.go +++ b/pkg/client/clientset/versioned/typed/orchestration/v1alpha1/orchestration_client.go @@ -20,8 +20,8 @@ package v1alpha1 import ( "net/http" - v1alpha1 "github.com/aibrix/aibrix/api/orchestration/v1alpha1" - "github.com/aibrix/aibrix/pkg/client/clientset/versioned/scheme" + v1alpha1 "github.com/vllm-project/aibrix/api/orchestration/v1alpha1" + "github.com/vllm-project/aibrix/pkg/client/clientset/versioned/scheme" rest "k8s.io/client-go/rest" ) diff --git a/pkg/client/clientset/versioned/typed/orchestration/v1alpha1/rayclusterfleet.go b/pkg/client/clientset/versioned/typed/orchestration/v1alpha1/rayclusterfleet.go index 6c1db393..4a3250d3 100644 --- a/pkg/client/clientset/versioned/typed/orchestration/v1alpha1/rayclusterfleet.go +++ b/pkg/client/clientset/versioned/typed/orchestration/v1alpha1/rayclusterfleet.go @@ -23,9 +23,9 @@ import ( "fmt" "time" - v1alpha1 "github.com/aibrix/aibrix/api/orchestration/v1alpha1" - orchestrationv1alpha1 "github.com/aibrix/aibrix/pkg/client/applyconfiguration/orchestration/v1alpha1" - scheme "github.com/aibrix/aibrix/pkg/client/clientset/versioned/scheme" + v1alpha1 "github.com/vllm-project/aibrix/api/orchestration/v1alpha1" + orchestrationv1alpha1 "github.com/vllm-project/aibrix/pkg/client/applyconfiguration/orchestration/v1alpha1" + scheme "github.com/vllm-project/aibrix/pkg/client/clientset/versioned/scheme" v1 "k8s.io/apimachinery/pkg/apis/meta/v1" types "k8s.io/apimachinery/pkg/types" watch "k8s.io/apimachinery/pkg/watch" diff --git a/pkg/client/clientset/versioned/typed/orchestration/v1alpha1/rayclusterreplicaset.go b/pkg/client/clientset/versioned/typed/orchestration/v1alpha1/rayclusterreplicaset.go index acdd007d..ae434681 100644 --- a/pkg/client/clientset/versioned/typed/orchestration/v1alpha1/rayclusterreplicaset.go +++ b/pkg/client/clientset/versioned/typed/orchestration/v1alpha1/rayclusterreplicaset.go @@ -23,9 +23,9 @@ import ( "fmt" "time" - v1alpha1 "github.com/aibrix/aibrix/api/orchestration/v1alpha1" - orchestrationv1alpha1 "github.com/aibrix/aibrix/pkg/client/applyconfiguration/orchestration/v1alpha1" - scheme "github.com/aibrix/aibrix/pkg/client/clientset/versioned/scheme" + v1alpha1 "github.com/vllm-project/aibrix/api/orchestration/v1alpha1" + orchestrationv1alpha1 "github.com/vllm-project/aibrix/pkg/client/applyconfiguration/orchestration/v1alpha1" + scheme "github.com/vllm-project/aibrix/pkg/client/clientset/versioned/scheme" v1 "k8s.io/apimachinery/pkg/apis/meta/v1" types "k8s.io/apimachinery/pkg/types" watch "k8s.io/apimachinery/pkg/watch" diff --git a/pkg/client/informers/externalversions/autoscaling/interface.go b/pkg/client/informers/externalversions/autoscaling/interface.go index 7ea864f5..a0a4a7d5 100644 --- a/pkg/client/informers/externalversions/autoscaling/interface.go +++ b/pkg/client/informers/externalversions/autoscaling/interface.go @@ -18,8 +18,8 @@ limitations under the License. package autoscaling import ( - v1alpha1 "github.com/aibrix/aibrix/pkg/client/informers/externalversions/autoscaling/v1alpha1" - internalinterfaces "github.com/aibrix/aibrix/pkg/client/informers/externalversions/internalinterfaces" + v1alpha1 "github.com/vllm-project/aibrix/pkg/client/informers/externalversions/autoscaling/v1alpha1" + internalinterfaces "github.com/vllm-project/aibrix/pkg/client/informers/externalversions/internalinterfaces" ) // Interface provides access to each of this group's versions. diff --git a/pkg/client/informers/externalversions/autoscaling/v1alpha1/interface.go b/pkg/client/informers/externalversions/autoscaling/v1alpha1/interface.go index a9c00d5d..7e743a5f 100644 --- a/pkg/client/informers/externalversions/autoscaling/v1alpha1/interface.go +++ b/pkg/client/informers/externalversions/autoscaling/v1alpha1/interface.go @@ -18,7 +18,7 @@ limitations under the License. package v1alpha1 import ( - internalinterfaces "github.com/aibrix/aibrix/pkg/client/informers/externalversions/internalinterfaces" + internalinterfaces "github.com/vllm-project/aibrix/pkg/client/informers/externalversions/internalinterfaces" ) // Interface provides access to all the informers in this group version. diff --git a/pkg/client/informers/externalversions/autoscaling/v1alpha1/podautoscaler.go b/pkg/client/informers/externalversions/autoscaling/v1alpha1/podautoscaler.go index 14680eed..5c13bd0a 100644 --- a/pkg/client/informers/externalversions/autoscaling/v1alpha1/podautoscaler.go +++ b/pkg/client/informers/externalversions/autoscaling/v1alpha1/podautoscaler.go @@ -21,10 +21,10 @@ import ( "context" time "time" - autoscalingv1alpha1 "github.com/aibrix/aibrix/api/autoscaling/v1alpha1" - versioned "github.com/aibrix/aibrix/pkg/client/clientset/versioned" - internalinterfaces "github.com/aibrix/aibrix/pkg/client/informers/externalversions/internalinterfaces" - v1alpha1 "github.com/aibrix/aibrix/pkg/client/listers/autoscaling/v1alpha1" + autoscalingv1alpha1 "github.com/vllm-project/aibrix/api/autoscaling/v1alpha1" + versioned "github.com/vllm-project/aibrix/pkg/client/clientset/versioned" + internalinterfaces "github.com/vllm-project/aibrix/pkg/client/informers/externalversions/internalinterfaces" + v1alpha1 "github.com/vllm-project/aibrix/pkg/client/listers/autoscaling/v1alpha1" v1 "k8s.io/apimachinery/pkg/apis/meta/v1" runtime "k8s.io/apimachinery/pkg/runtime" watch "k8s.io/apimachinery/pkg/watch" diff --git a/pkg/client/informers/externalversions/factory.go b/pkg/client/informers/externalversions/factory.go index 3edd09e6..98dcbfb6 100644 --- a/pkg/client/informers/externalversions/factory.go +++ b/pkg/client/informers/externalversions/factory.go @@ -22,11 +22,11 @@ import ( sync "sync" time "time" - versioned "github.com/aibrix/aibrix/pkg/client/clientset/versioned" - autoscaling "github.com/aibrix/aibrix/pkg/client/informers/externalversions/autoscaling" - internalinterfaces "github.com/aibrix/aibrix/pkg/client/informers/externalversions/internalinterfaces" - model "github.com/aibrix/aibrix/pkg/client/informers/externalversions/model" - orchestration "github.com/aibrix/aibrix/pkg/client/informers/externalversions/orchestration" + versioned "github.com/vllm-project/aibrix/pkg/client/clientset/versioned" + autoscaling "github.com/vllm-project/aibrix/pkg/client/informers/externalversions/autoscaling" + internalinterfaces "github.com/vllm-project/aibrix/pkg/client/informers/externalversions/internalinterfaces" + model "github.com/vllm-project/aibrix/pkg/client/informers/externalversions/model" + orchestration "github.com/vllm-project/aibrix/pkg/client/informers/externalversions/orchestration" v1 "k8s.io/apimachinery/pkg/apis/meta/v1" runtime "k8s.io/apimachinery/pkg/runtime" schema "k8s.io/apimachinery/pkg/runtime/schema" diff --git a/pkg/client/informers/externalversions/generic.go b/pkg/client/informers/externalversions/generic.go index a07b2ea9..4a0e2bc0 100644 --- a/pkg/client/informers/externalversions/generic.go +++ b/pkg/client/informers/externalversions/generic.go @@ -20,9 +20,9 @@ package externalversions import ( "fmt" - v1alpha1 "github.com/aibrix/aibrix/api/autoscaling/v1alpha1" - modelv1alpha1 "github.com/aibrix/aibrix/api/model/v1alpha1" - orchestrationv1alpha1 "github.com/aibrix/aibrix/api/orchestration/v1alpha1" + v1alpha1 "github.com/vllm-project/aibrix/api/autoscaling/v1alpha1" + modelv1alpha1 "github.com/vllm-project/aibrix/api/model/v1alpha1" + orchestrationv1alpha1 "github.com/vllm-project/aibrix/api/orchestration/v1alpha1" schema "k8s.io/apimachinery/pkg/runtime/schema" cache "k8s.io/client-go/tools/cache" ) diff --git a/pkg/client/informers/externalversions/internalinterfaces/factory_interfaces.go b/pkg/client/informers/externalversions/internalinterfaces/factory_interfaces.go index 18136973..8f075349 100644 --- a/pkg/client/informers/externalversions/internalinterfaces/factory_interfaces.go +++ b/pkg/client/informers/externalversions/internalinterfaces/factory_interfaces.go @@ -20,7 +20,7 @@ package internalinterfaces import ( time "time" - versioned "github.com/aibrix/aibrix/pkg/client/clientset/versioned" + versioned "github.com/vllm-project/aibrix/pkg/client/clientset/versioned" v1 "k8s.io/apimachinery/pkg/apis/meta/v1" runtime "k8s.io/apimachinery/pkg/runtime" cache "k8s.io/client-go/tools/cache" diff --git a/pkg/client/informers/externalversions/model/interface.go b/pkg/client/informers/externalversions/model/interface.go index 091aab53..d2b2d9cf 100644 --- a/pkg/client/informers/externalversions/model/interface.go +++ b/pkg/client/informers/externalversions/model/interface.go @@ -18,8 +18,8 @@ limitations under the License. package model import ( - internalinterfaces "github.com/aibrix/aibrix/pkg/client/informers/externalversions/internalinterfaces" - v1alpha1 "github.com/aibrix/aibrix/pkg/client/informers/externalversions/model/v1alpha1" + internalinterfaces "github.com/vllm-project/aibrix/pkg/client/informers/externalversions/internalinterfaces" + v1alpha1 "github.com/vllm-project/aibrix/pkg/client/informers/externalversions/model/v1alpha1" ) // Interface provides access to each of this group's versions. diff --git a/pkg/client/informers/externalversions/model/v1alpha1/interface.go b/pkg/client/informers/externalversions/model/v1alpha1/interface.go index 64590085..d41e27d2 100644 --- a/pkg/client/informers/externalversions/model/v1alpha1/interface.go +++ b/pkg/client/informers/externalversions/model/v1alpha1/interface.go @@ -18,7 +18,7 @@ limitations under the License. package v1alpha1 import ( - internalinterfaces "github.com/aibrix/aibrix/pkg/client/informers/externalversions/internalinterfaces" + internalinterfaces "github.com/vllm-project/aibrix/pkg/client/informers/externalversions/internalinterfaces" ) // Interface provides access to all the informers in this group version. diff --git a/pkg/client/informers/externalversions/model/v1alpha1/modeladapter.go b/pkg/client/informers/externalversions/model/v1alpha1/modeladapter.go index 4ecfd377..b10a8fb0 100644 --- a/pkg/client/informers/externalversions/model/v1alpha1/modeladapter.go +++ b/pkg/client/informers/externalversions/model/v1alpha1/modeladapter.go @@ -21,10 +21,10 @@ import ( "context" time "time" - modelv1alpha1 "github.com/aibrix/aibrix/api/model/v1alpha1" - versioned "github.com/aibrix/aibrix/pkg/client/clientset/versioned" - internalinterfaces "github.com/aibrix/aibrix/pkg/client/informers/externalversions/internalinterfaces" - v1alpha1 "github.com/aibrix/aibrix/pkg/client/listers/model/v1alpha1" + modelv1alpha1 "github.com/vllm-project/aibrix/api/model/v1alpha1" + versioned "github.com/vllm-project/aibrix/pkg/client/clientset/versioned" + internalinterfaces "github.com/vllm-project/aibrix/pkg/client/informers/externalversions/internalinterfaces" + v1alpha1 "github.com/vllm-project/aibrix/pkg/client/listers/model/v1alpha1" v1 "k8s.io/apimachinery/pkg/apis/meta/v1" runtime "k8s.io/apimachinery/pkg/runtime" watch "k8s.io/apimachinery/pkg/watch" diff --git a/pkg/client/informers/externalversions/orchestration/interface.go b/pkg/client/informers/externalversions/orchestration/interface.go index b785f529..753d1582 100644 --- a/pkg/client/informers/externalversions/orchestration/interface.go +++ b/pkg/client/informers/externalversions/orchestration/interface.go @@ -18,8 +18,8 @@ limitations under the License. package orchestration import ( - internalinterfaces "github.com/aibrix/aibrix/pkg/client/informers/externalversions/internalinterfaces" - v1alpha1 "github.com/aibrix/aibrix/pkg/client/informers/externalversions/orchestration/v1alpha1" + internalinterfaces "github.com/vllm-project/aibrix/pkg/client/informers/externalversions/internalinterfaces" + v1alpha1 "github.com/vllm-project/aibrix/pkg/client/informers/externalversions/orchestration/v1alpha1" ) // Interface provides access to each of this group's versions. diff --git a/pkg/client/informers/externalversions/orchestration/v1alpha1/interface.go b/pkg/client/informers/externalversions/orchestration/v1alpha1/interface.go index 6e976a38..8f03913d 100644 --- a/pkg/client/informers/externalversions/orchestration/v1alpha1/interface.go +++ b/pkg/client/informers/externalversions/orchestration/v1alpha1/interface.go @@ -18,7 +18,7 @@ limitations under the License. package v1alpha1 import ( - internalinterfaces "github.com/aibrix/aibrix/pkg/client/informers/externalversions/internalinterfaces" + internalinterfaces "github.com/vllm-project/aibrix/pkg/client/informers/externalversions/internalinterfaces" ) // Interface provides access to all the informers in this group version. diff --git a/pkg/client/informers/externalversions/orchestration/v1alpha1/rayclusterfleet.go b/pkg/client/informers/externalversions/orchestration/v1alpha1/rayclusterfleet.go index d3b54707..62d8c1d8 100644 --- a/pkg/client/informers/externalversions/orchestration/v1alpha1/rayclusterfleet.go +++ b/pkg/client/informers/externalversions/orchestration/v1alpha1/rayclusterfleet.go @@ -21,10 +21,10 @@ import ( "context" time "time" - orchestrationv1alpha1 "github.com/aibrix/aibrix/api/orchestration/v1alpha1" - versioned "github.com/aibrix/aibrix/pkg/client/clientset/versioned" - internalinterfaces "github.com/aibrix/aibrix/pkg/client/informers/externalversions/internalinterfaces" - v1alpha1 "github.com/aibrix/aibrix/pkg/client/listers/orchestration/v1alpha1" + orchestrationv1alpha1 "github.com/vllm-project/aibrix/api/orchestration/v1alpha1" + versioned "github.com/vllm-project/aibrix/pkg/client/clientset/versioned" + internalinterfaces "github.com/vllm-project/aibrix/pkg/client/informers/externalversions/internalinterfaces" + v1alpha1 "github.com/vllm-project/aibrix/pkg/client/listers/orchestration/v1alpha1" v1 "k8s.io/apimachinery/pkg/apis/meta/v1" runtime "k8s.io/apimachinery/pkg/runtime" watch "k8s.io/apimachinery/pkg/watch" diff --git a/pkg/client/informers/externalversions/orchestration/v1alpha1/rayclusterreplicaset.go b/pkg/client/informers/externalversions/orchestration/v1alpha1/rayclusterreplicaset.go index 46b052ff..d20b8d7f 100644 --- a/pkg/client/informers/externalversions/orchestration/v1alpha1/rayclusterreplicaset.go +++ b/pkg/client/informers/externalversions/orchestration/v1alpha1/rayclusterreplicaset.go @@ -21,10 +21,10 @@ import ( "context" time "time" - orchestrationv1alpha1 "github.com/aibrix/aibrix/api/orchestration/v1alpha1" - versioned "github.com/aibrix/aibrix/pkg/client/clientset/versioned" - internalinterfaces "github.com/aibrix/aibrix/pkg/client/informers/externalversions/internalinterfaces" - v1alpha1 "github.com/aibrix/aibrix/pkg/client/listers/orchestration/v1alpha1" + orchestrationv1alpha1 "github.com/vllm-project/aibrix/api/orchestration/v1alpha1" + versioned "github.com/vllm-project/aibrix/pkg/client/clientset/versioned" + internalinterfaces "github.com/vllm-project/aibrix/pkg/client/informers/externalversions/internalinterfaces" + v1alpha1 "github.com/vllm-project/aibrix/pkg/client/listers/orchestration/v1alpha1" v1 "k8s.io/apimachinery/pkg/apis/meta/v1" runtime "k8s.io/apimachinery/pkg/runtime" watch "k8s.io/apimachinery/pkg/watch" diff --git a/pkg/client/listers/autoscaling/v1alpha1/podautoscaler.go b/pkg/client/listers/autoscaling/v1alpha1/podautoscaler.go index ff35e2bf..df223f5c 100644 --- a/pkg/client/listers/autoscaling/v1alpha1/podautoscaler.go +++ b/pkg/client/listers/autoscaling/v1alpha1/podautoscaler.go @@ -18,7 +18,7 @@ limitations under the License. package v1alpha1 import ( - v1alpha1 "github.com/aibrix/aibrix/api/autoscaling/v1alpha1" + v1alpha1 "github.com/vllm-project/aibrix/api/autoscaling/v1alpha1" "k8s.io/apimachinery/pkg/api/errors" "k8s.io/apimachinery/pkg/labels" "k8s.io/client-go/tools/cache" diff --git a/pkg/client/listers/model/v1alpha1/modeladapter.go b/pkg/client/listers/model/v1alpha1/modeladapter.go index d2c7c93c..cacd9782 100644 --- a/pkg/client/listers/model/v1alpha1/modeladapter.go +++ b/pkg/client/listers/model/v1alpha1/modeladapter.go @@ -18,7 +18,7 @@ limitations under the License. package v1alpha1 import ( - v1alpha1 "github.com/aibrix/aibrix/api/model/v1alpha1" + v1alpha1 "github.com/vllm-project/aibrix/api/model/v1alpha1" "k8s.io/apimachinery/pkg/api/errors" "k8s.io/apimachinery/pkg/labels" "k8s.io/client-go/tools/cache" diff --git a/pkg/client/listers/orchestration/v1alpha1/rayclusterfleet.go b/pkg/client/listers/orchestration/v1alpha1/rayclusterfleet.go index 6c333ec9..f01d4851 100644 --- a/pkg/client/listers/orchestration/v1alpha1/rayclusterfleet.go +++ b/pkg/client/listers/orchestration/v1alpha1/rayclusterfleet.go @@ -18,7 +18,7 @@ limitations under the License. package v1alpha1 import ( - v1alpha1 "github.com/aibrix/aibrix/api/orchestration/v1alpha1" + v1alpha1 "github.com/vllm-project/aibrix/api/orchestration/v1alpha1" "k8s.io/apimachinery/pkg/api/errors" "k8s.io/apimachinery/pkg/labels" "k8s.io/client-go/tools/cache" diff --git a/pkg/client/listers/orchestration/v1alpha1/rayclusterreplicaset.go b/pkg/client/listers/orchestration/v1alpha1/rayclusterreplicaset.go index 6633d9c8..3447097f 100644 --- a/pkg/client/listers/orchestration/v1alpha1/rayclusterreplicaset.go +++ b/pkg/client/listers/orchestration/v1alpha1/rayclusterreplicaset.go @@ -18,7 +18,7 @@ limitations under the License. package v1alpha1 import ( - v1alpha1 "github.com/aibrix/aibrix/api/orchestration/v1alpha1" + v1alpha1 "github.com/vllm-project/aibrix/api/orchestration/v1alpha1" "k8s.io/apimachinery/pkg/api/errors" "k8s.io/apimachinery/pkg/labels" "k8s.io/client-go/tools/cache" diff --git a/pkg/controller/controller.go b/pkg/controller/controller.go index 79b6565f..9ec7f487 100644 --- a/pkg/controller/controller.go +++ b/pkg/controller/controller.go @@ -17,14 +17,14 @@ limitations under the License. package controller import ( - "github.com/aibrix/aibrix/pkg/config" - "github.com/aibrix/aibrix/pkg/controller/kvcache" - "github.com/aibrix/aibrix/pkg/controller/modeladapter" - "github.com/aibrix/aibrix/pkg/controller/modelrouter" - "github.com/aibrix/aibrix/pkg/controller/podautoscaler" - "github.com/aibrix/aibrix/pkg/controller/rayclusterfleet" - "github.com/aibrix/aibrix/pkg/controller/rayclusterreplicaset" - "github.com/aibrix/aibrix/pkg/features" + "github.com/vllm-project/aibrix/pkg/config" + "github.com/vllm-project/aibrix/pkg/controller/kvcache" + "github.com/vllm-project/aibrix/pkg/controller/modeladapter" + "github.com/vllm-project/aibrix/pkg/controller/modelrouter" + "github.com/vllm-project/aibrix/pkg/controller/podautoscaler" + "github.com/vllm-project/aibrix/pkg/controller/rayclusterfleet" + "github.com/vllm-project/aibrix/pkg/controller/rayclusterreplicaset" + "github.com/vllm-project/aibrix/pkg/features" "k8s.io/apimachinery/pkg/api/meta" "k8s.io/klog/v2" "sigs.k8s.io/controller-runtime/pkg/manager" diff --git a/pkg/controller/kvcache/kvcache_controller.go b/pkg/controller/kvcache/kvcache_controller.go index 107f9310..a211087e 100644 --- a/pkg/controller/kvcache/kvcache_controller.go +++ b/pkg/controller/kvcache/kvcache_controller.go @@ -21,8 +21,8 @@ import ( "fmt" "reflect" - orchestrationv1alpha1 "github.com/aibrix/aibrix/api/orchestration/v1alpha1" - "github.com/aibrix/aibrix/pkg/config" + orchestrationv1alpha1 "github.com/vllm-project/aibrix/api/orchestration/v1alpha1" + "github.com/vllm-project/aibrix/pkg/config" appsv1 "k8s.io/api/apps/v1" corev1 "k8s.io/api/core/v1" diff --git a/pkg/controller/kvcache/kvcache_controller_test.go b/pkg/controller/kvcache/kvcache_controller_test.go index 569988ae..38b50d2d 100644 --- a/pkg/controller/kvcache/kvcache_controller_test.go +++ b/pkg/controller/kvcache/kvcache_controller_test.go @@ -27,7 +27,7 @@ import ( metav1 "k8s.io/apimachinery/pkg/apis/meta/v1" - orchestrationv1alpha1 "github.com/aibrix/aibrix/api/orchestration/v1alpha1" + orchestrationv1alpha1 "github.com/vllm-project/aibrix/api/orchestration/v1alpha1" ) var _ = Describe("KVCache Controller", func() { diff --git a/pkg/controller/kvcache/suite_test.go b/pkg/controller/kvcache/suite_test.go index a80ad413..09aed8a3 100644 --- a/pkg/controller/kvcache/suite_test.go +++ b/pkg/controller/kvcache/suite_test.go @@ -32,7 +32,7 @@ import ( logf "sigs.k8s.io/controller-runtime/pkg/log" "sigs.k8s.io/controller-runtime/pkg/log/zap" - orchestrationv1alpha1 "github.com/aibrix/aibrix/api/orchestration/v1alpha1" + orchestrationv1alpha1 "github.com/vllm-project/aibrix/api/orchestration/v1alpha1" //+kubebuilder:scaffold:imports ) diff --git a/pkg/controller/modeladapter/modeladapter_controller.go b/pkg/controller/modeladapter/modeladapter_controller.go index 79fa0ec7..87d6621e 100644 --- a/pkg/controller/modeladapter/modeladapter_controller.go +++ b/pkg/controller/modeladapter/modeladapter_controller.go @@ -27,11 +27,11 @@ import ( "strings" "time" - modelv1alpha1 "github.com/aibrix/aibrix/api/model/v1alpha1" - "github.com/aibrix/aibrix/pkg/cache" - "github.com/aibrix/aibrix/pkg/config" - "github.com/aibrix/aibrix/pkg/controller/modeladapter/scheduling" - "github.com/aibrix/aibrix/pkg/utils" + modelv1alpha1 "github.com/vllm-project/aibrix/api/model/v1alpha1" + "github.com/vllm-project/aibrix/pkg/cache" + "github.com/vllm-project/aibrix/pkg/config" + "github.com/vllm-project/aibrix/pkg/controller/modeladapter/scheduling" + "github.com/vllm-project/aibrix/pkg/utils" corev1 "k8s.io/api/core/v1" discoveryv1 "k8s.io/api/discovery/v1" apierrors "k8s.io/apimachinery/pkg/api/errors" diff --git a/pkg/controller/modeladapter/modeladapter_controller_test.go b/pkg/controller/modeladapter/modeladapter_controller_test.go index bd094a37..8ca9cbeb 100644 --- a/pkg/controller/modeladapter/modeladapter_controller_test.go +++ b/pkg/controller/modeladapter/modeladapter_controller_test.go @@ -25,7 +25,7 @@ import ( metav1 "k8s.io/apimachinery/pkg/apis/meta/v1" "k8s.io/apimachinery/pkg/types" - modelv1alpha1 "github.com/aibrix/aibrix/api/model/v1alpha1" + modelv1alpha1 "github.com/vllm-project/aibrix/api/model/v1alpha1" ) var _ = Describe("ModelAdapter Controller", func() { diff --git a/pkg/controller/modeladapter/resources.go b/pkg/controller/modeladapter/resources.go index 4e921ff9..23704d63 100644 --- a/pkg/controller/modeladapter/resources.go +++ b/pkg/controller/modeladapter/resources.go @@ -17,7 +17,7 @@ limitations under the License. package modeladapter import ( - modelv1alpha1 "github.com/aibrix/aibrix/api/model/v1alpha1" + modelv1alpha1 "github.com/vllm-project/aibrix/api/model/v1alpha1" corev1 "k8s.io/api/core/v1" discoveryv1 "k8s.io/api/discovery/v1" metav1 "k8s.io/apimachinery/pkg/apis/meta/v1" diff --git a/pkg/controller/modeladapter/resources_test.go b/pkg/controller/modeladapter/resources_test.go index 459eec89..80599392 100644 --- a/pkg/controller/modeladapter/resources_test.go +++ b/pkg/controller/modeladapter/resources_test.go @@ -19,8 +19,8 @@ package modeladapter import ( "testing" - modelv1alpha1 "github.com/aibrix/aibrix/api/model/v1alpha1" "github.com/stretchr/testify/assert" + modelv1alpha1 "github.com/vllm-project/aibrix/api/model/v1alpha1" corev1 "k8s.io/api/core/v1" metav1 "k8s.io/apimachinery/pkg/apis/meta/v1" "k8s.io/apimachinery/pkg/util/intstr" diff --git a/pkg/controller/modeladapter/scheduling/bin_pack.go b/pkg/controller/modeladapter/scheduling/bin_pack.go index 51db339b..fecf3008 100644 --- a/pkg/controller/modeladapter/scheduling/bin_pack.go +++ b/pkg/controller/modeladapter/scheduling/bin_pack.go @@ -20,7 +20,7 @@ import ( "context" "math" - "github.com/aibrix/aibrix/pkg/cache" + "github.com/vllm-project/aibrix/pkg/cache" v1 "k8s.io/api/core/v1" "k8s.io/klog/v2" ) diff --git a/pkg/controller/modeladapter/scheduling/least_adapters.go b/pkg/controller/modeladapter/scheduling/least_adapters.go index 0f9eef6c..637232c6 100644 --- a/pkg/controller/modeladapter/scheduling/least_adapters.go +++ b/pkg/controller/modeladapter/scheduling/least_adapters.go @@ -20,7 +20,7 @@ import ( "context" "math" - "github.com/aibrix/aibrix/pkg/cache" + "github.com/vllm-project/aibrix/pkg/cache" v1 "k8s.io/api/core/v1" "k8s.io/klog/v2" ) diff --git a/pkg/controller/modeladapter/scheduling/least_latency.go b/pkg/controller/modeladapter/scheduling/least_latency.go index 1a05a756..cfa6b9fe 100644 --- a/pkg/controller/modeladapter/scheduling/least_latency.go +++ b/pkg/controller/modeladapter/scheduling/least_latency.go @@ -20,8 +20,8 @@ import ( "context" "math" - "github.com/aibrix/aibrix/pkg/cache" - "github.com/aibrix/aibrix/pkg/metrics" + "github.com/vllm-project/aibrix/pkg/cache" + "github.com/vllm-project/aibrix/pkg/metrics" v1 "k8s.io/api/core/v1" "k8s.io/klog/v2" ) diff --git a/pkg/controller/modeladapter/scheduling/least_throughput.go b/pkg/controller/modeladapter/scheduling/least_throughput.go index e11fdb72..9e6fe113 100644 --- a/pkg/controller/modeladapter/scheduling/least_throughput.go +++ b/pkg/controller/modeladapter/scheduling/least_throughput.go @@ -20,8 +20,8 @@ import ( "context" "math" - "github.com/aibrix/aibrix/pkg/cache" - "github.com/aibrix/aibrix/pkg/metrics" + "github.com/vllm-project/aibrix/pkg/cache" + "github.com/vllm-project/aibrix/pkg/metrics" v1 "k8s.io/api/core/v1" "k8s.io/klog/v2" ) diff --git a/pkg/controller/modeladapter/scheduling/random.go b/pkg/controller/modeladapter/scheduling/random.go index 2ae71c52..4966936c 100644 --- a/pkg/controller/modeladapter/scheduling/random.go +++ b/pkg/controller/modeladapter/scheduling/random.go @@ -21,7 +21,7 @@ import ( "errors" "math/rand" - "github.com/aibrix/aibrix/pkg/cache" + "github.com/vllm-project/aibrix/pkg/cache" v1 "k8s.io/api/core/v1" "k8s.io/klog/v2" ) diff --git a/pkg/controller/modeladapter/scheduling/scheduler.go b/pkg/controller/modeladapter/scheduling/scheduler.go index 2944d1b5..9d2f2a1d 100644 --- a/pkg/controller/modeladapter/scheduling/scheduler.go +++ b/pkg/controller/modeladapter/scheduling/scheduler.go @@ -22,7 +22,7 @@ import ( v1 "k8s.io/api/core/v1" - "github.com/aibrix/aibrix/pkg/cache" + "github.com/vllm-project/aibrix/pkg/cache" ) type Scheduler interface { diff --git a/pkg/controller/modeladapter/suite_test.go b/pkg/controller/modeladapter/suite_test.go index ddd2aea7..37a71adf 100644 --- a/pkg/controller/modeladapter/suite_test.go +++ b/pkg/controller/modeladapter/suite_test.go @@ -32,7 +32,7 @@ import ( logf "sigs.k8s.io/controller-runtime/pkg/log" "sigs.k8s.io/controller-runtime/pkg/log/zap" - modelv1alpha1 "github.com/aibrix/aibrix/api/model/v1alpha1" + modelv1alpha1 "github.com/vllm-project/aibrix/api/model/v1alpha1" //+kubebuilder:scaffold:imports ) diff --git a/pkg/controller/modeladapter/utils.go b/pkg/controller/modeladapter/utils.go index cc54ba1f..48ddfde9 100644 --- a/pkg/controller/modeladapter/utils.go +++ b/pkg/controller/modeladapter/utils.go @@ -23,8 +23,8 @@ import ( "os" "strings" - modelv1alpha1 "github.com/aibrix/aibrix/api/model/v1alpha1" - "github.com/aibrix/aibrix/pkg/config" + modelv1alpha1 "github.com/vllm-project/aibrix/api/model/v1alpha1" + "github.com/vllm-project/aibrix/pkg/config" metav1 "k8s.io/apimachinery/pkg/apis/meta/v1" ) diff --git a/pkg/controller/modeladapter/utils_test.go b/pkg/controller/modeladapter/utils_test.go index eee8580b..426d92e9 100644 --- a/pkg/controller/modeladapter/utils_test.go +++ b/pkg/controller/modeladapter/utils_test.go @@ -25,8 +25,8 @@ import ( metav1 "k8s.io/apimachinery/pkg/apis/meta/v1" "k8s.io/utils/ptr" - modelv1alpha1 "github.com/aibrix/aibrix/api/model/v1alpha1" - "github.com/aibrix/aibrix/pkg/config" + modelv1alpha1 "github.com/vllm-project/aibrix/api/model/v1alpha1" + "github.com/vllm-project/aibrix/pkg/config" ) // Test for validateModelAdapter function diff --git a/pkg/controller/modelrouter/modelrouter_controller.go b/pkg/controller/modelrouter/modelrouter_controller.go index 271c2ad9..ddfc9fe1 100644 --- a/pkg/controller/modelrouter/modelrouter_controller.go +++ b/pkg/controller/modelrouter/modelrouter_controller.go @@ -32,8 +32,8 @@ import ( "sigs.k8s.io/controller-runtime/pkg/client" "sigs.k8s.io/controller-runtime/pkg/manager" - modelv1alpha1 "github.com/aibrix/aibrix/api/model/v1alpha1" - "github.com/aibrix/aibrix/pkg/config" + modelv1alpha1 "github.com/vllm-project/aibrix/api/model/v1alpha1" + "github.com/vllm-project/aibrix/pkg/config" gatewayv1 "sigs.k8s.io/gateway-api/apis/v1" gatewayv1beta1 "sigs.k8s.io/gateway-api/apis/v1beta1" ) diff --git a/pkg/controller/podautoscaler/algorithm/algorithm.go b/pkg/controller/podautoscaler/algorithm/algorithm.go index 98f5aff4..07e158e1 100644 --- a/pkg/controller/podautoscaler/algorithm/algorithm.go +++ b/pkg/controller/podautoscaler/algorithm/algorithm.go @@ -16,7 +16,7 @@ limitations under the License. package algorithm -import "github.com/aibrix/aibrix/pkg/controller/podautoscaler/common" +import "github.com/vllm-project/aibrix/pkg/controller/podautoscaler/common" type ScalingAlgorithm interface { // ComputeTargetReplicas calculates the number of replicas needed based on current metrics diff --git a/pkg/controller/podautoscaler/algorithm/apa.go b/pkg/controller/podautoscaler/algorithm/apa.go index 6713b963..b73bb065 100644 --- a/pkg/controller/podautoscaler/algorithm/apa.go +++ b/pkg/controller/podautoscaler/algorithm/apa.go @@ -21,7 +21,7 @@ import ( "k8s.io/klog/v2" - "github.com/aibrix/aibrix/pkg/controller/podautoscaler/common" + "github.com/vllm-project/aibrix/pkg/controller/podautoscaler/common" ) type ApaScalingAlgorithm struct{} diff --git a/pkg/controller/podautoscaler/algorithm/hpa.go b/pkg/controller/podautoscaler/algorithm/hpa.go index 13a112d3..ca0dec0a 100644 --- a/pkg/controller/podautoscaler/algorithm/hpa.go +++ b/pkg/controller/podautoscaler/algorithm/hpa.go @@ -16,7 +16,7 @@ limitations under the License. package algorithm -import "github.com/aibrix/aibrix/pkg/controller/podautoscaler/common" +import "github.com/vllm-project/aibrix/pkg/controller/podautoscaler/common" // HpaScalingAlgorithm can be used by any scaler without customized algorithms type HpaScalingAlgorithm struct{} diff --git a/pkg/controller/podautoscaler/algorithm/kpa.go b/pkg/controller/podautoscaler/algorithm/kpa.go index 7432ecf8..d39952d7 100644 --- a/pkg/controller/podautoscaler/algorithm/kpa.go +++ b/pkg/controller/podautoscaler/algorithm/kpa.go @@ -19,7 +19,7 @@ package algorithm import ( "math" - "github.com/aibrix/aibrix/pkg/controller/podautoscaler/common" + "github.com/vllm-project/aibrix/pkg/controller/podautoscaler/common" ) type KpaScalingAlgorithm struct{} diff --git a/pkg/controller/podautoscaler/common/context.go b/pkg/controller/podautoscaler/common/context.go index 09d20984..acd33f1c 100644 --- a/pkg/controller/podautoscaler/common/context.go +++ b/pkg/controller/podautoscaler/common/context.go @@ -19,7 +19,7 @@ package common import ( "strconv" - autoscalingv1alpha1 "github.com/aibrix/aibrix/api/autoscaling/v1alpha1" + autoscalingv1alpha1 "github.com/vllm-project/aibrix/api/autoscaling/v1alpha1" "k8s.io/klog/v2" ) diff --git a/pkg/controller/podautoscaler/hpa_resources.go b/pkg/controller/podautoscaler/hpa_resources.go index 919a547a..59799c88 100644 --- a/pkg/controller/podautoscaler/hpa_resources.go +++ b/pkg/controller/podautoscaler/hpa_resources.go @@ -22,7 +22,7 @@ import ( "strconv" "strings" - pav1 "github.com/aibrix/aibrix/api/autoscaling/v1alpha1" + pav1 "github.com/vllm-project/aibrix/api/autoscaling/v1alpha1" corev1 "k8s.io/api/core/v1" "k8s.io/apimachinery/pkg/api/resource" metav1 "k8s.io/apimachinery/pkg/apis/meta/v1" diff --git a/pkg/controller/podautoscaler/metrics/client.go b/pkg/controller/podautoscaler/metrics/client.go index f38798d1..853b222a 100644 --- a/pkg/controller/podautoscaler/metrics/client.go +++ b/pkg/controller/podautoscaler/metrics/client.go @@ -22,10 +22,10 @@ import ( corev1 "k8s.io/api/core/v1" - "github.com/aibrix/aibrix/pkg/controller/podautoscaler/aggregation" + "github.com/vllm-project/aibrix/pkg/controller/podautoscaler/aggregation" "k8s.io/klog/v2" - autoscalingv1alpha1 "github.com/aibrix/aibrix/api/autoscaling/v1alpha1" + autoscalingv1alpha1 "github.com/vllm-project/aibrix/api/autoscaling/v1alpha1" "time" ) diff --git a/pkg/controller/podautoscaler/metrics/client_test.go b/pkg/controller/podautoscaler/metrics/client_test.go index 222d4226..d38e0144 100644 --- a/pkg/controller/podautoscaler/metrics/client_test.go +++ b/pkg/controller/podautoscaler/metrics/client_test.go @@ -20,9 +20,9 @@ import ( "context" "testing" - autoscalingv1alpha1 "github.com/aibrix/aibrix/api/autoscaling/v1alpha1" . "github.com/onsi/ginkgo" . "github.com/onsi/gomega" + autoscalingv1alpha1 "github.com/vllm-project/aibrix/api/autoscaling/v1alpha1" ) func TestProxy(t *testing.T) { diff --git a/pkg/controller/podautoscaler/metrics/fetcher.go b/pkg/controller/podautoscaler/metrics/fetcher.go index f7eb1523..b0376218 100644 --- a/pkg/controller/podautoscaler/metrics/fetcher.go +++ b/pkg/controller/podautoscaler/metrics/fetcher.go @@ -24,7 +24,7 @@ import ( "net/http" "strings" - autoscalingv1alpha1 "github.com/aibrix/aibrix/api/autoscaling/v1alpha1" + autoscalingv1alpha1 "github.com/vllm-project/aibrix/api/autoscaling/v1alpha1" v1 "k8s.io/api/core/v1" metav1 "k8s.io/apimachinery/pkg/apis/meta/v1" diff --git a/pkg/controller/podautoscaler/metrics/interface.go b/pkg/controller/podautoscaler/metrics/interface.go index 3e5b6b49..81fcd78c 100644 --- a/pkg/controller/podautoscaler/metrics/interface.go +++ b/pkg/controller/podautoscaler/metrics/interface.go @@ -25,7 +25,7 @@ import ( v1 "k8s.io/api/core/v1" - autoscalingv1alpha1 "github.com/aibrix/aibrix/api/autoscaling/v1alpha1" + autoscalingv1alpha1 "github.com/vllm-project/aibrix/api/autoscaling/v1alpha1" ) // NamespaceNameMetric contains the namespace, name and the metric name diff --git a/pkg/controller/podautoscaler/metrics/utils.go b/pkg/controller/podautoscaler/metrics/utils.go index 5554ccd4..187fe758 100644 --- a/pkg/controller/podautoscaler/metrics/utils.go +++ b/pkg/controller/podautoscaler/metrics/utils.go @@ -25,7 +25,7 @@ import ( "strings" "time" - autoscalingv1alpha1 "github.com/aibrix/aibrix/api/autoscaling/v1alpha1" + autoscalingv1alpha1 "github.com/vllm-project/aibrix/api/autoscaling/v1alpha1" corev1 "k8s.io/api/core/v1" ) diff --git a/pkg/controller/podautoscaler/podautoscaler_controller.go b/pkg/controller/podautoscaler/podautoscaler_controller.go index 76c53a94..2e455a48 100644 --- a/pkg/controller/podautoscaler/podautoscaler_controller.go +++ b/pkg/controller/podautoscaler/podautoscaler_controller.go @@ -21,13 +21,13 @@ import ( "fmt" "time" - autoscalingv1alpha1 "github.com/aibrix/aibrix/api/autoscaling/v1alpha1" - "github.com/aibrix/aibrix/pkg/config" - "github.com/aibrix/aibrix/pkg/controller/podautoscaler/metrics" + autoscalingv1alpha1 "github.com/vllm-project/aibrix/api/autoscaling/v1alpha1" + "github.com/vllm-project/aibrix/pkg/config" + "github.com/vllm-project/aibrix/pkg/controller/podautoscaler/metrics" - "github.com/aibrix/aibrix/pkg/controller/podautoscaler/scaler" - podutil "github.com/aibrix/aibrix/pkg/utils" - podutils "github.com/aibrix/aibrix/pkg/utils" + "github.com/vllm-project/aibrix/pkg/controller/podautoscaler/scaler" + podutil "github.com/vllm-project/aibrix/pkg/utils" + podutils "github.com/vllm-project/aibrix/pkg/utils" autoscalingv2 "k8s.io/api/autoscaling/v2" corev1 "k8s.io/api/core/v1" diff --git a/pkg/controller/podautoscaler/podautoscaler_controller_test.go b/pkg/controller/podautoscaler/podautoscaler_controller_test.go index 2ef56e4c..16e5192b 100644 --- a/pkg/controller/podautoscaler/podautoscaler_controller_test.go +++ b/pkg/controller/podautoscaler/podautoscaler_controller_test.go @@ -29,7 +29,7 @@ import ( metav1 "k8s.io/apimachinery/pkg/apis/meta/v1" - autoscalingv1alpha1 "github.com/aibrix/aibrix/api/autoscaling/v1alpha1" + autoscalingv1alpha1 "github.com/vllm-project/aibrix/api/autoscaling/v1alpha1" ) func IntToPtr(num int32) *int32 { diff --git a/pkg/controller/podautoscaler/scaler/apa.go b/pkg/controller/podautoscaler/scaler/apa.go index 04ec9546..e0528834 100644 --- a/pkg/controller/podautoscaler/scaler/apa.go +++ b/pkg/controller/podautoscaler/scaler/apa.go @@ -24,12 +24,12 @@ import ( "sigs.k8s.io/controller-runtime/pkg/client" - autoscalingv1alpha1 "github.com/aibrix/aibrix/api/autoscaling/v1alpha1" - "github.com/aibrix/aibrix/pkg/controller/podautoscaler/algorithm" - "github.com/aibrix/aibrix/pkg/controller/podautoscaler/common" - scalingcontext "github.com/aibrix/aibrix/pkg/controller/podautoscaler/common" - "github.com/aibrix/aibrix/pkg/controller/podautoscaler/metrics" - "github.com/aibrix/aibrix/pkg/utils" + autoscalingv1alpha1 "github.com/vllm-project/aibrix/api/autoscaling/v1alpha1" + "github.com/vllm-project/aibrix/pkg/controller/podautoscaler/algorithm" + "github.com/vllm-project/aibrix/pkg/controller/podautoscaler/common" + scalingcontext "github.com/vllm-project/aibrix/pkg/controller/podautoscaler/common" + "github.com/vllm-project/aibrix/pkg/controller/podautoscaler/metrics" + "github.com/vllm-project/aibrix/pkg/utils" v1 "k8s.io/api/core/v1" "k8s.io/klog/v2" diff --git a/pkg/controller/podautoscaler/scaler/apa_test.go b/pkg/controller/podautoscaler/scaler/apa_test.go index 75762f50..1ad8f667 100644 --- a/pkg/controller/podautoscaler/scaler/apa_test.go +++ b/pkg/controller/podautoscaler/scaler/apa_test.go @@ -21,13 +21,13 @@ import ( "testing" "time" - "github.com/aibrix/aibrix/pkg/controller/podautoscaler/algorithm" + "github.com/vllm-project/aibrix/pkg/controller/podautoscaler/algorithm" - autoscalingv1alpha1 "github.com/aibrix/aibrix/api/autoscaling/v1alpha1" + autoscalingv1alpha1 "github.com/vllm-project/aibrix/api/autoscaling/v1alpha1" corev1 "k8s.io/api/core/v1" metav1 "k8s.io/apimachinery/pkg/apis/meta/v1" - "github.com/aibrix/aibrix/pkg/controller/podautoscaler/metrics" + "github.com/vllm-project/aibrix/pkg/controller/podautoscaler/metrics" ) // TestHcpaScale tests the APA behavior. For now, APA implements HCPA algorithm. diff --git a/pkg/controller/podautoscaler/scaler/interface.go b/pkg/controller/podautoscaler/scaler/interface.go index 291c71a8..f6e4875f 100644 --- a/pkg/controller/podautoscaler/scaler/interface.go +++ b/pkg/controller/podautoscaler/scaler/interface.go @@ -20,9 +20,9 @@ import ( "context" "time" - autoscalingv1alpha1 "github.com/aibrix/aibrix/api/autoscaling/v1alpha1" - "github.com/aibrix/aibrix/pkg/controller/podautoscaler/common" - "github.com/aibrix/aibrix/pkg/controller/podautoscaler/metrics" + autoscalingv1alpha1 "github.com/vllm-project/aibrix/api/autoscaling/v1alpha1" + "github.com/vllm-project/aibrix/pkg/controller/podautoscaler/common" + "github.com/vllm-project/aibrix/pkg/controller/podautoscaler/metrics" corev1 "k8s.io/api/core/v1" ) diff --git a/pkg/controller/podautoscaler/scaler/kpa.go b/pkg/controller/podautoscaler/scaler/kpa.go index 5873ff2a..d2fbfd36 100644 --- a/pkg/controller/podautoscaler/scaler/kpa.go +++ b/pkg/controller/podautoscaler/scaler/kpa.go @@ -25,16 +25,16 @@ import ( "sync" "time" - "github.com/aibrix/aibrix/pkg/controller/podautoscaler/algorithm" + "github.com/vllm-project/aibrix/pkg/controller/podautoscaler/algorithm" "sigs.k8s.io/controller-runtime/pkg/client" - autoscalingv1alpha1 "github.com/aibrix/aibrix/api/autoscaling/v1alpha1" - scalingcontext "github.com/aibrix/aibrix/pkg/controller/podautoscaler/common" - "github.com/aibrix/aibrix/pkg/controller/podautoscaler/metrics" - "github.com/aibrix/aibrix/pkg/utils" + autoscalingv1alpha1 "github.com/vllm-project/aibrix/api/autoscaling/v1alpha1" + scalingcontext "github.com/vllm-project/aibrix/pkg/controller/podautoscaler/common" + "github.com/vllm-project/aibrix/pkg/controller/podautoscaler/metrics" + "github.com/vllm-project/aibrix/pkg/utils" v1 "k8s.io/api/core/v1" - "github.com/aibrix/aibrix/pkg/controller/podautoscaler/aggregation" + "github.com/vllm-project/aibrix/pkg/controller/podautoscaler/aggregation" "k8s.io/klog/v2" ) diff --git a/pkg/controller/podautoscaler/scaler/kpa_test.go b/pkg/controller/podautoscaler/scaler/kpa_test.go index 102803be..b8a5dd23 100644 --- a/pkg/controller/podautoscaler/scaler/kpa_test.go +++ b/pkg/controller/podautoscaler/scaler/kpa_test.go @@ -24,15 +24,15 @@ import ( "k8s.io/klog/v2" - "github.com/aibrix/aibrix/pkg/controller/podautoscaler/aggregation" - "github.com/aibrix/aibrix/pkg/controller/podautoscaler/algorithm" - scalingcontext "github.com/aibrix/aibrix/pkg/controller/podautoscaler/common" + "github.com/vllm-project/aibrix/pkg/controller/podautoscaler/aggregation" + "github.com/vllm-project/aibrix/pkg/controller/podautoscaler/algorithm" + scalingcontext "github.com/vllm-project/aibrix/pkg/controller/podautoscaler/common" - v1alpha1 "github.com/aibrix/aibrix/api/autoscaling/v1alpha1" + v1alpha1 "github.com/vllm-project/aibrix/api/autoscaling/v1alpha1" corev1 "k8s.io/api/core/v1" metav1 "k8s.io/apimachinery/pkg/apis/meta/v1" - "github.com/aibrix/aibrix/pkg/controller/podautoscaler/metrics" + "github.com/vllm-project/aibrix/pkg/controller/podautoscaler/metrics" ) // TestKpaScale tests the KPA behavior under high traffic rising condition. diff --git a/pkg/controller/podautoscaler/scaler/scaler.go b/pkg/controller/podautoscaler/scaler/scaler.go index 6cf1855d..cc566349 100644 --- a/pkg/controller/podautoscaler/scaler/scaler.go +++ b/pkg/controller/podautoscaler/scaler/scaler.go @@ -21,8 +21,8 @@ import ( "fmt" "time" - "github.com/aibrix/aibrix/pkg/controller/podautoscaler/metrics" - podutil "github.com/aibrix/aibrix/pkg/utils" + "github.com/vllm-project/aibrix/pkg/controller/podautoscaler/metrics" + podutil "github.com/vllm-project/aibrix/pkg/utils" v1 "k8s.io/api/core/v1" "k8s.io/apimachinery/pkg/labels" "k8s.io/apimachinery/pkg/util/sets" diff --git a/pkg/controller/podautoscaler/scaler/scaler_factory.go b/pkg/controller/podautoscaler/scaler/scaler_factory.go index fd606f55..4c778ed4 100644 --- a/pkg/controller/podautoscaler/scaler/scaler_factory.go +++ b/pkg/controller/podautoscaler/scaler/scaler_factory.go @@ -20,7 +20,7 @@ import ( "fmt" "time" - autoscalingv1alpha1 "github.com/aibrix/aibrix/api/autoscaling/v1alpha1" + autoscalingv1alpha1 "github.com/vllm-project/aibrix/api/autoscaling/v1alpha1" ) // NewAutoscalerFactory creates an Autoscaler based on the given ScalingStrategy diff --git a/pkg/controller/podautoscaler/suite_test.go b/pkg/controller/podautoscaler/suite_test.go index 0482c2fa..a38021fa 100644 --- a/pkg/controller/podautoscaler/suite_test.go +++ b/pkg/controller/podautoscaler/suite_test.go @@ -32,7 +32,7 @@ import ( logf "sigs.k8s.io/controller-runtime/pkg/log" "sigs.k8s.io/controller-runtime/pkg/log/zap" - autoscalingv1alpha1 "github.com/aibrix/aibrix/api/autoscaling/v1alpha1" + autoscalingv1alpha1 "github.com/vllm-project/aibrix/api/autoscaling/v1alpha1" //+kubebuilder:scaffold:imports ) diff --git a/pkg/controller/rayclusterfleet/progress.go b/pkg/controller/rayclusterfleet/progress.go index d053fe02..a3aaf9ff 100644 --- a/pkg/controller/rayclusterfleet/progress.go +++ b/pkg/controller/rayclusterfleet/progress.go @@ -22,8 +22,8 @@ import ( "reflect" "time" - orchestrationv1alpha1 "github.com/aibrix/aibrix/api/orchestration/v1alpha1" - "github.com/aibrix/aibrix/pkg/controller/rayclusterfleet/util" + orchestrationv1alpha1 "github.com/vllm-project/aibrix/api/orchestration/v1alpha1" + "github.com/vllm-project/aibrix/pkg/controller/rayclusterfleet/util" corev1 "k8s.io/api/core/v1" "k8s.io/klog/v2" ) diff --git a/pkg/controller/rayclusterfleet/rayclusterfleet_controller.go b/pkg/controller/rayclusterfleet/rayclusterfleet_controller.go index a06d960c..6359e6f5 100644 --- a/pkg/controller/rayclusterfleet/rayclusterfleet_controller.go +++ b/pkg/controller/rayclusterfleet/rayclusterfleet_controller.go @@ -22,8 +22,8 @@ import ( "reflect" "time" - "github.com/aibrix/aibrix/pkg/config" - "github.com/aibrix/aibrix/pkg/controller/util/expectation" + "github.com/vllm-project/aibrix/pkg/config" + "github.com/vllm-project/aibrix/pkg/controller/util/expectation" appsv1 "k8s.io/api/apps/v1" v1 "k8s.io/api/core/v1" "k8s.io/apimachinery/pkg/api/errors" @@ -36,7 +36,7 @@ import ( "sigs.k8s.io/controller-runtime/pkg/manager" "sigs.k8s.io/controller-runtime/pkg/reconcile" - orchestrationv1alpha1 "github.com/aibrix/aibrix/api/orchestration/v1alpha1" + orchestrationv1alpha1 "github.com/vllm-project/aibrix/api/orchestration/v1alpha1" "k8s.io/apimachinery/pkg/runtime" ctrl "sigs.k8s.io/controller-runtime" "sigs.k8s.io/controller-runtime/pkg/client" diff --git a/pkg/controller/rayclusterfleet/rayclusterfleet_controller_test.go b/pkg/controller/rayclusterfleet/rayclusterfleet_controller_test.go index e11a6d94..2d6d89ea 100644 --- a/pkg/controller/rayclusterfleet/rayclusterfleet_controller_test.go +++ b/pkg/controller/rayclusterfleet/rayclusterfleet_controller_test.go @@ -31,7 +31,7 @@ import ( "k8s.io/utils/ptr" "sigs.k8s.io/controller-runtime/pkg/reconcile" - orchestrationv1alpha1 "github.com/aibrix/aibrix/api/orchestration/v1alpha1" + orchestrationv1alpha1 "github.com/vllm-project/aibrix/api/orchestration/v1alpha1" ) var _ = Describe("RayClusterFleet Controller", func() { diff --git a/pkg/controller/rayclusterfleet/recreate.go b/pkg/controller/rayclusterfleet/recreate.go index 8e0d7749..3cab8f4b 100644 --- a/pkg/controller/rayclusterfleet/recreate.go +++ b/pkg/controller/rayclusterfleet/recreate.go @@ -19,11 +19,11 @@ package rayclusterfleet import ( "context" - orchestrationv1alpha1 "github.com/aibrix/aibrix/api/orchestration/v1alpha1" - rayclusterutil "github.com/aibrix/aibrix/pkg/utils" rayclusterv1 "github.com/ray-project/kuberay/ray-operator/apis/ray/v1" + orchestrationv1alpha1 "github.com/vllm-project/aibrix/api/orchestration/v1alpha1" + rayclusterutil "github.com/vllm-project/aibrix/pkg/utils" - "github.com/aibrix/aibrix/pkg/controller/rayclusterfleet/util" + "github.com/vllm-project/aibrix/pkg/controller/rayclusterfleet/util" "k8s.io/apimachinery/pkg/types" ) diff --git a/pkg/controller/rayclusterfleet/rollback.go b/pkg/controller/rayclusterfleet/rollback.go index 13695a58..ad683eb6 100644 --- a/pkg/controller/rayclusterfleet/rollback.go +++ b/pkg/controller/rayclusterfleet/rollback.go @@ -21,8 +21,8 @@ import ( "fmt" "strconv" - orchestrationv1alpha1 "github.com/aibrix/aibrix/api/orchestration/v1alpha1" - "github.com/aibrix/aibrix/pkg/controller/rayclusterfleet/util" + orchestrationv1alpha1 "github.com/vllm-project/aibrix/api/orchestration/v1alpha1" + "github.com/vllm-project/aibrix/pkg/controller/rayclusterfleet/util" appsv1 "k8s.io/api/apps/v1" v1 "k8s.io/api/core/v1" extensions "k8s.io/api/extensions/v1beta1" diff --git a/pkg/controller/rayclusterfleet/rolling.go b/pkg/controller/rayclusterfleet/rolling.go index 20012b7f..77a17064 100644 --- a/pkg/controller/rayclusterfleet/rolling.go +++ b/pkg/controller/rayclusterfleet/rolling.go @@ -21,8 +21,8 @@ import ( "fmt" "sort" - orchestrationv1alpha1 "github.com/aibrix/aibrix/api/orchestration/v1alpha1" - "github.com/aibrix/aibrix/pkg/controller/rayclusterfleet/util" + orchestrationv1alpha1 "github.com/vllm-project/aibrix/api/orchestration/v1alpha1" + "github.com/vllm-project/aibrix/pkg/controller/rayclusterfleet/util" "k8s.io/klog/v2" ) diff --git a/pkg/controller/rayclusterfleet/suite_test.go b/pkg/controller/rayclusterfleet/suite_test.go index 981abba6..6546592e 100644 --- a/pkg/controller/rayclusterfleet/suite_test.go +++ b/pkg/controller/rayclusterfleet/suite_test.go @@ -33,7 +33,7 @@ import ( logf "sigs.k8s.io/controller-runtime/pkg/log" "sigs.k8s.io/controller-runtime/pkg/log/zap" - orchestrationv1alpha1 "github.com/aibrix/aibrix/api/orchestration/v1alpha1" + orchestrationv1alpha1 "github.com/vllm-project/aibrix/api/orchestration/v1alpha1" //+kubebuilder:scaffold:imports ) diff --git a/pkg/controller/rayclusterfleet/sync.go b/pkg/controller/rayclusterfleet/sync.go index a5642a83..2047e087 100644 --- a/pkg/controller/rayclusterfleet/sync.go +++ b/pkg/controller/rayclusterfleet/sync.go @@ -23,9 +23,9 @@ import ( "sort" "strconv" - orchestrationv1alpha1 "github.com/aibrix/aibrix/api/orchestration/v1alpha1" - "github.com/aibrix/aibrix/pkg/controller/rayclusterfleet/util" - labelsutil "github.com/aibrix/aibrix/pkg/utils" + orchestrationv1alpha1 "github.com/vllm-project/aibrix/api/orchestration/v1alpha1" + "github.com/vllm-project/aibrix/pkg/controller/rayclusterfleet/util" + labelsutil "github.com/vllm-project/aibrix/pkg/utils" appsv1 "k8s.io/api/apps/v1" v1 "k8s.io/api/core/v1" "k8s.io/apimachinery/pkg/api/errors" diff --git a/pkg/controller/rayclusterfleet/util/fleet.go b/pkg/controller/rayclusterfleet/util/fleet.go index 27bf1f72..1119094c 100644 --- a/pkg/controller/rayclusterfleet/util/fleet.go +++ b/pkg/controller/rayclusterfleet/util/fleet.go @@ -28,8 +28,8 @@ import ( "strings" "time" - orchestrationv1alpha1 "github.com/aibrix/aibrix/api/orchestration/v1alpha1" - labelsutil "github.com/aibrix/aibrix/pkg/utils" + orchestrationv1alpha1 "github.com/vllm-project/aibrix/api/orchestration/v1alpha1" + labelsutil "github.com/vllm-project/aibrix/pkg/utils" appsv1 "k8s.io/api/apps/v1" v1 "k8s.io/api/core/v1" apiequality "k8s.io/apimachinery/pkg/api/equality" diff --git a/pkg/controller/rayclusterreplicaset/rayclusterreplicaset_controller.go b/pkg/controller/rayclusterreplicaset/rayclusterreplicaset_controller.go index a9330999..eaa7c543 100644 --- a/pkg/controller/rayclusterreplicaset/rayclusterreplicaset_controller.go +++ b/pkg/controller/rayclusterreplicaset/rayclusterreplicaset_controller.go @@ -22,17 +22,17 @@ import ( "sync" "time" - "github.com/aibrix/aibrix/pkg/controller/util/expectation" + "github.com/vllm-project/aibrix/pkg/controller/util/expectation" apierrors "k8s.io/apimachinery/pkg/api/errors" - "github.com/aibrix/aibrix/pkg/config" rayclusterv1 "github.com/ray-project/kuberay/ray-operator/apis/ray/v1" + "github.com/vllm-project/aibrix/pkg/config" "k8s.io/client-go/tools/record" "k8s.io/klog/v2" "sigs.k8s.io/controller-runtime/pkg/manager" "sigs.k8s.io/controller-runtime/pkg/reconcile" - orchestrationv1alpha1 "github.com/aibrix/aibrix/api/orchestration/v1alpha1" + orchestrationv1alpha1 "github.com/vllm-project/aibrix/api/orchestration/v1alpha1" "k8s.io/apimachinery/pkg/runtime" "k8s.io/apimachinery/pkg/types" ctrl "sigs.k8s.io/controller-runtime" diff --git a/pkg/controller/rayclusterreplicaset/rayclusterreplicaset_controller_test.go b/pkg/controller/rayclusterreplicaset/rayclusterreplicaset_controller_test.go index 4a550b74..691acb5e 100644 --- a/pkg/controller/rayclusterreplicaset/rayclusterreplicaset_controller_test.go +++ b/pkg/controller/rayclusterreplicaset/rayclusterreplicaset_controller_test.go @@ -30,8 +30,8 @@ import ( "k8s.io/utils/ptr" "sigs.k8s.io/controller-runtime/pkg/reconcile" - orchestrationv1alpha1 "github.com/aibrix/aibrix/api/orchestration/v1alpha1" - "github.com/aibrix/aibrix/pkg/controller/util/expectation" + orchestrationv1alpha1 "github.com/vllm-project/aibrix/api/orchestration/v1alpha1" + "github.com/vllm-project/aibrix/pkg/controller/util/expectation" ) var _ = Describe("RayClusterReplicaSet Controller", func() { diff --git a/pkg/controller/rayclusterreplicaset/rayclusterreplicaset_utils.go b/pkg/controller/rayclusterreplicaset/rayclusterreplicaset_utils.go index ecfadbb9..340dd186 100644 --- a/pkg/controller/rayclusterreplicaset/rayclusterreplicaset_utils.go +++ b/pkg/controller/rayclusterreplicaset/rayclusterreplicaset_utils.go @@ -19,9 +19,9 @@ package rayclusterreplicaset import ( "reflect" - orchestrationv1alpha1 "github.com/aibrix/aibrix/api/orchestration/v1alpha1" - rayclusterutil "github.com/aibrix/aibrix/pkg/utils" rayclusterv1 "github.com/ray-project/kuberay/ray-operator/apis/ray/v1" + orchestrationv1alpha1 "github.com/vllm-project/aibrix/api/orchestration/v1alpha1" + rayclusterutil "github.com/vllm-project/aibrix/pkg/utils" "k8s.io/apimachinery/pkg/api/meta" metav1 "k8s.io/apimachinery/pkg/apis/meta/v1" "k8s.io/apimachinery/pkg/labels" diff --git a/pkg/controller/rayclusterreplicaset/suite_test.go b/pkg/controller/rayclusterreplicaset/suite_test.go index 0350e0ed..cb05e400 100644 --- a/pkg/controller/rayclusterreplicaset/suite_test.go +++ b/pkg/controller/rayclusterreplicaset/suite_test.go @@ -33,7 +33,7 @@ import ( logf "sigs.k8s.io/controller-runtime/pkg/log" "sigs.k8s.io/controller-runtime/pkg/log/zap" - orchestrationv1alpha1 "github.com/aibrix/aibrix/api/orchestration/v1alpha1" + orchestrationv1alpha1 "github.com/vllm-project/aibrix/api/orchestration/v1alpha1" //+kubebuilder:scaffold:imports ) diff --git a/pkg/metadata/helpers.go b/pkg/metadata/helpers.go index 042fe014..6c4474af 100644 --- a/pkg/metadata/helpers.go +++ b/pkg/metadata/helpers.go @@ -24,8 +24,8 @@ import ( "net/http" "strings" - "github.com/aibrix/aibrix/pkg/utils" "github.com/go-playground/validator/v10" + "github.com/vllm-project/aibrix/pkg/utils" ) type malformedRequest struct { diff --git a/pkg/metadata/users.go b/pkg/metadata/users.go index 2b4f3142..09b9ac93 100644 --- a/pkg/metadata/users.go +++ b/pkg/metadata/users.go @@ -21,9 +21,9 @@ import ( "fmt" "net/http" - "github.com/aibrix/aibrix/pkg/utils" "github.com/gorilla/mux" "github.com/redis/go-redis/v9" + "github.com/vllm-project/aibrix/pkg/utils" "k8s.io/klog/v2" ) diff --git a/pkg/plugins/gateway/algorithms/least_busy_time.go b/pkg/plugins/gateway/algorithms/least_busy_time.go index c2383c67..c04e9816 100644 --- a/pkg/plugins/gateway/algorithms/least_busy_time.go +++ b/pkg/plugins/gateway/algorithms/least_busy_time.go @@ -22,7 +22,7 @@ import ( "math" "math/rand" - "github.com/aibrix/aibrix/pkg/cache" + "github.com/vllm-project/aibrix/pkg/cache" v1 "k8s.io/api/core/v1" "k8s.io/klog/v2" ) diff --git a/pkg/plugins/gateway/algorithms/least_kv_cache.go b/pkg/plugins/gateway/algorithms/least_kv_cache.go index 70c7d9df..07fac272 100644 --- a/pkg/plugins/gateway/algorithms/least_kv_cache.go +++ b/pkg/plugins/gateway/algorithms/least_kv_cache.go @@ -22,8 +22,8 @@ import ( "math" "math/rand" - "github.com/aibrix/aibrix/pkg/cache" - metrics "github.com/aibrix/aibrix/pkg/metrics" + "github.com/vllm-project/aibrix/pkg/cache" + metrics "github.com/vllm-project/aibrix/pkg/metrics" v1 "k8s.io/api/core/v1" "k8s.io/klog/v2" ) diff --git a/pkg/plugins/gateway/algorithms/least_latency.go b/pkg/plugins/gateway/algorithms/least_latency.go index f5807554..f90268e0 100644 --- a/pkg/plugins/gateway/algorithms/least_latency.go +++ b/pkg/plugins/gateway/algorithms/least_latency.go @@ -22,8 +22,8 @@ import ( "math" "math/rand" - "github.com/aibrix/aibrix/pkg/cache" - "github.com/aibrix/aibrix/pkg/metrics" + "github.com/vllm-project/aibrix/pkg/cache" + "github.com/vllm-project/aibrix/pkg/metrics" v1 "k8s.io/api/core/v1" "k8s.io/klog/v2" ) diff --git a/pkg/plugins/gateway/algorithms/least_request.go b/pkg/plugins/gateway/algorithms/least_request.go index 4cd693fe..e4cf8b2e 100644 --- a/pkg/plugins/gateway/algorithms/least_request.go +++ b/pkg/plugins/gateway/algorithms/least_request.go @@ -22,9 +22,9 @@ import ( "math" "math/rand" - "github.com/aibrix/aibrix/pkg/cache" - "github.com/aibrix/aibrix/pkg/metrics" - "github.com/aibrix/aibrix/pkg/utils" + "github.com/vllm-project/aibrix/pkg/cache" + "github.com/vllm-project/aibrix/pkg/metrics" + "github.com/vllm-project/aibrix/pkg/utils" v1 "k8s.io/api/core/v1" "k8s.io/klog/v2" ) diff --git a/pkg/plugins/gateway/algorithms/prefix_cache.go b/pkg/plugins/gateway/algorithms/prefix_cache.go index ce2c5b50..5db592e2 100644 --- a/pkg/plugins/gateway/algorithms/prefix_cache.go +++ b/pkg/plugins/gateway/algorithms/prefix_cache.go @@ -22,8 +22,8 @@ import ( "math/rand" "strconv" - "github.com/aibrix/aibrix/pkg/plugins/gateway/prefixcacheindexer" - "github.com/aibrix/aibrix/pkg/utils" + "github.com/vllm-project/aibrix/pkg/plugins/gateway/prefixcacheindexer" + "github.com/vllm-project/aibrix/pkg/utils" v1 "k8s.io/api/core/v1" "k8s.io/klog/v2" ) diff --git a/pkg/plugins/gateway/algorithms/router.go b/pkg/plugins/gateway/algorithms/router.go index 72d745a9..ab273aae 100644 --- a/pkg/plugins/gateway/algorithms/router.go +++ b/pkg/plugins/gateway/algorithms/router.go @@ -20,7 +20,7 @@ import ( "context" "fmt" - "github.com/aibrix/aibrix/pkg/utils" + "github.com/vllm-project/aibrix/pkg/utils" v1 "k8s.io/api/core/v1" ) diff --git a/pkg/plugins/gateway/algorithms/router_test.go b/pkg/plugins/gateway/algorithms/router_test.go index efb0e265..5fb2fd95 100644 --- a/pkg/plugins/gateway/algorithms/router_test.go +++ b/pkg/plugins/gateway/algorithms/router_test.go @@ -22,9 +22,9 @@ import ( "testing" "time" - "github.com/aibrix/aibrix/pkg/cache" - "github.com/aibrix/aibrix/pkg/metrics" "github.com/stretchr/testify/assert" + "github.com/vllm-project/aibrix/pkg/cache" + "github.com/vllm-project/aibrix/pkg/metrics" v1 "k8s.io/api/core/v1" metav1 "k8s.io/apimachinery/pkg/apis/meta/v1" ) diff --git a/pkg/plugins/gateway/algorithms/throughput.go b/pkg/plugins/gateway/algorithms/throughput.go index b1c1d1c4..f1824e31 100644 --- a/pkg/plugins/gateway/algorithms/throughput.go +++ b/pkg/plugins/gateway/algorithms/throughput.go @@ -22,9 +22,9 @@ import ( "math" "math/rand" - "github.com/aibrix/aibrix/pkg/cache" - "github.com/aibrix/aibrix/pkg/metrics" - "github.com/aibrix/aibrix/pkg/utils" + "github.com/vllm-project/aibrix/pkg/cache" + "github.com/vllm-project/aibrix/pkg/metrics" + "github.com/vllm-project/aibrix/pkg/utils" v1 "k8s.io/api/core/v1" "k8s.io/klog/v2" ) diff --git a/pkg/plugins/gateway/gateway.go b/pkg/plugins/gateway/gateway.go index e51a1f88..ad7e68ab 100644 --- a/pkg/plugins/gateway/gateway.go +++ b/pkg/plugins/gateway/gateway.go @@ -40,13 +40,13 @@ import ( "k8s.io/client-go/kubernetes" "k8s.io/klog/v2" - "github.com/aibrix/aibrix/pkg/cache" - routing "github.com/aibrix/aibrix/pkg/plugins/gateway/algorithms" - ratelimiter "github.com/aibrix/aibrix/pkg/plugins/gateway/ratelimiter" - "github.com/aibrix/aibrix/pkg/utils" configPb "github.com/envoyproxy/go-control-plane/envoy/config/core/v3" extProcPb "github.com/envoyproxy/go-control-plane/envoy/service/ext_proc/v3" envoyTypePb "github.com/envoyproxy/go-control-plane/envoy/type/v3" + "github.com/vllm-project/aibrix/pkg/cache" + routing "github.com/vllm-project/aibrix/pkg/plugins/gateway/algorithms" + ratelimiter "github.com/vllm-project/aibrix/pkg/plugins/gateway/ratelimiter" + "github.com/vllm-project/aibrix/pkg/utils" healthPb "google.golang.org/grpc/health/grpc_health_v1" ) diff --git a/pkg/plugins/gateway/prefixcacheindexer/hash.go b/pkg/plugins/gateway/prefixcacheindexer/hash.go index 2897e5f6..cbac85e4 100644 --- a/pkg/plugins/gateway/prefixcacheindexer/hash.go +++ b/pkg/plugins/gateway/prefixcacheindexer/hash.go @@ -23,8 +23,8 @@ import ( "sync" "time" - "github.com/aibrix/aibrix/pkg/utils" "github.com/cespare/xxhash" + "github.com/vllm-project/aibrix/pkg/utils" v1 "k8s.io/api/core/v1" "k8s.io/klog/v2" ) diff --git a/pkg/plugins/gateway/prefixcacheindexer/hash_test.go b/pkg/plugins/gateway/prefixcacheindexer/hash_test.go index 4ac252bc..64cbe16f 100644 --- a/pkg/plugins/gateway/prefixcacheindexer/hash_test.go +++ b/pkg/plugins/gateway/prefixcacheindexer/hash_test.go @@ -21,8 +21,8 @@ import ( "testing" "time" - "github.com/aibrix/aibrix/pkg/utils" "github.com/stretchr/testify/assert" + "github.com/vllm-project/aibrix/pkg/utils" v1 "k8s.io/api/core/v1" metav1 "k8s.io/apimachinery/pkg/apis/meta/v1" ) diff --git a/python/aibrix/README.md b/python/aibrix/README.md index 6e93129e..aa9f0689 100644 --- a/python/aibrix/README.md +++ b/python/aibrix/README.md @@ -15,17 +15,17 @@ The AI Runtime supports model downloading from the following storage backends: * S3 * TOS -For more details on model downloading, please refer to our [Runtime docs](https://github.com/aibrix/aibrix/blob/main/docs/source/features/runtime.rst#model-downloading). +For more details on model downloading, please refer to our [Runtime docs](https://github.com/vllm-project/aibrix/blob/main/docs/source/features/runtime.rst#model-downloading). ### Integrate with inference engines The AI Runtime hides various implementation details on the inference engine side, providing a universal method to guide model management, as well as expose inference monitoring metrics. At present, `vLLM` engine is supported, and in the future, `SGLang` and other inference engines will be supported. -For more details on integrate with `vLLM`, please refer to our [Runtime docs](https://github.com/aibrix/aibrix/blob/main/docs/source/features/runtime.rst#metric-standardization). +For more details on integrate with `vLLM`, please refer to our [Runtime docs](https://github.com/vllm-project/aibrix/blob/main/docs/source/features/runtime.rst#metric-standardization). ## Contributing -We welcome contributions from the community! Check out our [contributing guidelines](https://github.com/aibrix/aibrix/blob/main/CONTRIBUTING.md) to see how you can make a difference. +We welcome contributions from the community! Check out our [contributing guidelines](https://github.com/vllm-project/aibrix/blob/main/CONTRIBUTING.md) to see how you can make a difference. ### Build from source @@ -48,4 +48,4 @@ bash ./scripts/format.sh ## License -AI Runtime is licensed under the [APACHE License](https://github.com/aibrix/aibrix/LICENSE.md). \ No newline at end of file +AI Runtime is licensed under the [APACHE License](https://github.com/vllm-project/aibrix/LICENSE.md). \ No newline at end of file diff --git a/python/aibrix/pyproject.toml b/python/aibrix/pyproject.toml index 93bbce94..526443e6 100644 --- a/python/aibrix/pyproject.toml +++ b/python/aibrix/pyproject.toml @@ -7,7 +7,7 @@ authors = [ ] readme = "README.md" license = "Apache-2.0" -repository = "https://github.com/aibrix/aibrix/tree/main/python/aibrix" +repository = "https://github.com/vllm-project/aibrix/tree/main/python/aibrix" classifiers = [ "Intended Audience :: Developers", "Intended Audience :: Education", diff --git a/test/e2e/util.go b/test/e2e/util.go index 0ffbd04f..0f75500a 100644 --- a/test/e2e/util.go +++ b/test/e2e/util.go @@ -22,10 +22,10 @@ import ( "os" "testing" - v1alpha1 "github.com/aibrix/aibrix/pkg/client/clientset/versioned" - crdinformers "github.com/aibrix/aibrix/pkg/client/informers/externalversions" "github.com/openai/openai-go" "github.com/openai/openai-go/option" + v1alpha1 "github.com/vllm-project/aibrix/pkg/client/clientset/versioned" + crdinformers "github.com/vllm-project/aibrix/pkg/client/informers/externalversions" "k8s.io/apimachinery/pkg/util/runtime" "k8s.io/client-go/informers" "k8s.io/client-go/kubernetes"