Skip to content

Conversation

@Jack-Khuu
Copy link
Contributor

@Jack-Khuu Jack-Khuu commented Oct 31, 2025

We currently use argparsers, but as we land more pipeline features, this will grow unwieldy and we'll want a unified config for controlling all the individual agent components.

This PR adds an initial Hydra setup and migrates the existing UI clis (pipeline-ui, fuser-ui, kernel-agent).

Hydra integration for other scripts will be in follow up PRs

Note: We can follow up this PR to use StructuredConfigs once the cli is a bit more stable

Note 2: I'm not 100% sold that I want to use Hydra, but it's a simple onramp/offramp if we want to use something else (Fiddle, manual dataclass factories, etc.) and it's better than manual parsing


Manual tests

scripts/triton_ui.py

kernel-agent
kernel-agent port=8086

python scripts/triton_ui.py
python scripts/triton_ui.py port=8086

scripts/pipeline_ui.py

pipeline-ui
pipeline-ui port=8085

python scripts/pipeline_ui.py
python scripts/pipeline_ui.py port=8085

scripts/fuser_ui.py

fuser-ui
fuser-ui port=8085

python scripts/fuser_ui.py
python scripts/fuser_ui.py port=8085

Example Output (Similar across scripts)

  • Note the change in ports
🚀 Starting Triton Kernel Agent UI...
📝 Provide your OpenAI API key in the UI or configure in .env file
🌐 Opening on Meta devserver: https://devgpu008.eag2.facebook.com:8085/
💡 Make sure you're connected to Meta VPN to access the demo
* Running on local URL:  https://devgpu008.eag2.facebook.com:8085
* To create a public link, set `share=True` in `launch()`.
^CKeyboard interruption in main thread... closing server.

🚀 Starting Triton Kernel Agent UI...
📝 Provide your OpenAI API key in the UI or configure in .env file
🌐 Opening on Meta devserver: https://devgpu008.eag2.facebook.com:8086/
💡 Make sure you're connected to Meta VPN to access the demo
* Running on local URL:  https://devgpu008.eag2.facebook.com:8086
* To create a public link, set `share=True` in `launch()`.
^CKeyboard interruption in main thread... closing server.

@meta-cla meta-cla bot added the CLA Signed This label is managed by the Meta Open Source bot. label Oct 31, 2025
@Jack-Khuu Jack-Khuu changed the title Add Basic Hydra Config support for triton_ui Add Hydra Config support for UI scripts Nov 8, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

CLA Signed This label is managed by the Meta Open Source bot.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants