feat: add MiniMax as alternative LLM provider for prompt optimization#62
Open
octo-patch wants to merge 1 commit intozai-org:mainfrom
Open
feat: add MiniMax as alternative LLM provider for prompt optimization#62octo-patch wants to merge 1 commit intozai-org:mainfrom
octo-patch wants to merge 1 commit intozai-org:mainfrom
Conversation
Add PROVIDER_PRESETS system to prompt_optimize.py with support for MiniMax (M2.7), Zhipu AI, and OpenAI as selectable LLM backends for prompt enhancement. Update Gradio demo with provider dropdown and LLM_PROVIDER env var. Includes 18 unit tests and 3 integration tests.
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Summary
Add MiniMax as an alternative LLM provider for prompt optimization alongside the existing Zhipu AI backend.
prompt_optimize.py:--provider minimax|zhipu|openaifor easy switching, withMINIMAX_API_KEYenv var auto-detectionLLM_PROVIDERenv varChanged files (7 files, 444 additions)
inference/prompt_optimize.pyPROVIDER_PRESETSdict,--providerarg,MINIMAX_API_KEYenv var supportinference/gradio_web_demo.pyLLM_PROVIDERenv var, importedPROVIDER_PRESETSREADME.mdREADME_zh.mdtests/__init__.pytests/test_prompt_optimize.pytests/test_integration_minimax.pyTest plan
python -m unittest tests.test_prompt_optimize -v)MINIMAX_API_KEY=... python -m unittest tests.test_integration_minimax -v)prompt_optimize.pyCLI behavior unchanged (defaults to zhipu provider)LLM_PROVIDER=minimax(requires GPU for diffusion model)This PR adds MiniMax as an alternative — the default behavior (Zhipu AI) is fully preserved and no existing functionality is changed.