Skip to content

Feat: Support running AIBrix in local mode#2039

Open
Jeffwan wants to merge 1 commit intovllm-project:mainfrom
Jeffwan:jiaxin/aibrix-local-run
Open

Feat: Support running AIBrix in local mode#2039
Jeffwan wants to merge 1 commit intovllm-project:mainfrom
Jeffwan:jiaxin/aibrix-local-run

Conversation

@Jeffwan
Copy link
Collaborator

@Jeffwan Jeffwan commented Mar 21, 2026

Pull Request Description

Support local mode - easier to debug and connect to engine for testing purpose.

Related Issues

Resolves: #1951 #2040

not all contributors have cluster environment, they can use such ways to test against aibrix router & kv cache or even benchmark later.
This PR is to unblock their development.

Important: Before submitting, please complete the description above and review the checklist below.


Contribution Guidelines (Expand for Details)

We appreciate your contribution to aibrix! To ensure a smooth review process and maintain high code quality, please adhere to the following guidelines:

Pull Request Title Format

Your PR title should start with one of these prefixes to indicate the nature of the change:

  • [Bug]: Corrections to existing functionality
  • [CI]: Changes to build process or CI pipeline
  • [Docs]: Updates or additions to documentation
  • [API]: Modifications to aibrix's API or interface
  • [CLI]: Changes or additions to the Command Line Interface
  • [Misc]: For changes not covered above (use sparingly)

Note: For changes spanning multiple categories, use multiple prefixes in order of importance.

Submission Checklist

  • PR title includes appropriate prefix(es)
  • Changes are clearly explained in the PR description
  • New and existing tests pass successfully
  • Code adheres to project style and best practices
  • Documentation updated to reflect changes (if applicable)
  • Thorough testing completed, no regressions introduced

By submitting this PR, you confirm that you've read these guidelines and your changes align with the project's contribution standards.

@gemini-code-assist
Copy link
Contributor

Summary of Changes

Hello, I'm Gemini Code Assist1! I'm currently reviewing this pull request and will post my feedback shortly. In the meantime, here's a summary to help you and other reviewers quickly get up to speed!

This pull request significantly enhances the developer experience by introducing a local mode for AIBrix. This new capability allows engineers to run the core gateway components directly on their machines, bypassing the need for containerization or orchestration tools like Kubernetes. This change streamlines the debugging process, accelerates local testing cycles, and provides a more accessible environment for validating gateway functionality, ultimately improving development efficiency.

Highlights

  • Local Mode Support: Introduced a new 'local mode' for AIBrix, allowing the gateway (Envoy + gateway-plugin) to run as bare processes without Docker or Kubernetes.
  • Simplified Development & Debugging: This mode is designed to facilitate local development, debugging of routing algorithms, single-node testing, and quick validation of gateway behavior.
  • Direct vLLM Engine Discovery: The local setup uses static configuration to directly discover and route requests to vLLM engine instances.
  • Comprehensive Documentation & Scripts: Provided detailed README.md documentation, example configuration files (endpoints.yaml, envoy.yaml), and convenience scripts (run-local.sh, stop-local.sh) for easy setup and management of the local environment.

🧠 New Feature in Public Preview: You can now enable Memory to help Gemini Code Assist learn from your team's feedback. This makes future code reviews more consistent and personalized to your project's style. Click here to enable Memory in your admin console.

Using Gemini Code Assist

The full guide for Gemini Code Assist can be found on our documentation page, here are some quick tips.

Invoking Gemini

You can request assistance from Gemini at any point by creating a comment using either /gemini <command> or @gemini-code-assist <command>. Below is a summary of the supported commands on the current page.

Feature Command Description
Code Review /gemini review Performs a code review for the current pull request in its current state.
Pull Request Summary /gemini summary Provides a summary of the current pull request in its current state.
Comment @gemini-code-assist Responds in comments when explicitly tagged, both in pull request comments and review comments.
Help /gemini help Displays a list of available commands.

Customization

To customize Gemini Code Assist for GitHub experience, repository maintainers can create a configuration file and/or provide a custom code review style guide (such as PEP-8 for Python) by creating and adding files to a .gemini/ folder in the base of the repository. Detailed instructions can be found here.

Limitations & Feedback

Gemini Code Assist may make mistakes. Please leave feedback on any instances where its feedback is incorrect or counter productive. You can react with 👍 and 👎 on @gemini-code-assist comments. If you're interested in giving your feedback about your experience with Gemini Code Assist for GitHub and other Google products, sign up here.

Footnotes

  1. Review the Privacy Notices, Generative AI Prohibited Use Policy, Terms of Service, and learn how to configure Gemini Code Assist in GitHub here. Gemini can make mistakes, so double check it and use code with caution.

Copy link
Contributor

@gemini-code-assist gemini-code-assist bot left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Code Review

This pull request introduces a valuable local mode for running AIBrix, which will greatly simplify development and testing. The implementation is well-structured, with clear documentation, configuration files, and helper scripts. My review focuses on improving the user experience by ensuring consistency in the documentation and increasing the robustness of the startup script. I've suggested aligning the example model names across all files to prevent user errors and replacing a fixed sleep with a more reliable polling mechanism in the run-local.sh script.

Signed-off-by: Jiaxin Shan <seedjeffwan@gmail.com>
@Jeffwan Jeffwan force-pushed the jiaxin/aibrix-local-run branch from 7560db6 to 448063a Compare March 21, 2026 23:37
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

1 participant