litellm/tests/test_litellm/proxy
Cole McIntosh 3b911ba1b2
Enhance proxy CLI with Rich formatting and improved user experience (#11420)
* Enhance proxy CLI with Rich formatting and improved user experience

- Integrated Rich library for better console output in `proxy_cli.py`, including version display, health check results, and test completion responses.
- Updated health check and test completion methods to provide progress indicators and formatted tables.
- Refactored feedback display in `proxy_server.py` to use Rich for a more visually appealing user interface.
- Adjusted tests in `test_proxy_cli.py` to mock console output instead of using print statements, ensuring compatibility with Rich formatting.

* fix linting error

* refactor(proxy_cli.py): simplify DB setup logging

- Removed progress indicators for IAM token generation and environment variable decryption to simplify the code.
- Consolidated the logic for generating the database URL and setting environment variables.
- Enhanced error handling for configuration loading and database setup, ensuring clearer feedback

* Update test-linting workflow to include proxy-dev dependencies in Poetry installation

* Enhance proxy server initialization with Rich console for improved model display. Added support for loading model parameters from environment variables and refined provider identification logic. Fallback to original print formatting if Rich is not available.

* Refactor feedback handling: Moved feedback message generation and custom warning display to utils.py. Enhanced feedback box with rich formatting and fallback to ASCII for environments without rich. Cleaned up proxy_server.py by removing obsolete code.

* fix linting error

* Refactor model initialization display: Moved model initialization logic to a new utility function `display_model_initialization` for improved readability and maintainability. Enhanced model provider extraction with a dedicated function. Fallback to basic logging if Rich console is unavailable.

* Refactor model provider extraction: Replace the `_extract_provider_from_model` function with a more robust approach using `get_llm_provider`. Implement fallback logic for provider identification and improve error handling. Ensure compatibility with Rich console for model initialization display.
2025-06-06 17:16:53 -07:00
..
anthropic_endpoints Litellm fix GitHub action testing (#11163) 2025-05-26 14:41:42 -07:00
auth [Feat] Enforce Vector Store Access Controls on LiteLLM Auth (#11281) 2025-05-30 22:20:11 -07:00
client Litellm fix GitHub action testing (#11163) 2025-05-26 14:41:42 -07:00
common_utils [Bug Fix] Create/Update team member api 500 errror (#10479) 2025-06-03 15:57:33 -07:00
db build(VLLM-Passthrough-with-loadbalancing-support-(enables-using-model-list-for-VLLM-/classify-endpoint)): Closes #11205 2025-05-31 09:00:04 -07:00
experimental/mcp_server Litellm fix GitHub action testing (#11163) 2025-05-26 14:41:42 -07:00
guardrails build(VLLM-Passthrough-with-loadbalancing-support-(enables-using-model-list-for-VLLM-/classify-endpoint)): Closes #11205 2025-05-31 09:00:04 -07:00
health_endpoints Litellm fix GitHub action testing (#11163) 2025-05-26 14:41:42 -07:00
hooks feat(parallel_request_limiter_v2.py): add sliding window logic (#11283) 2025-05-31 10:06:42 -07:00
image_endpoints Feat: add add azure endpoint for image endpoints (#11482) 2025-06-06 10:38:37 -07:00
management_endpoints UI / SSO - Update proxy admin id role in DB + Handle SSO redirects with custom root path (#11384) 2025-06-03 21:16:55 -07:00
middleware Litellm fix GitHub action testing (#11163) 2025-05-26 14:41:42 -07:00
openai_files_endpoint Litellm fix GitHub action testing (#11163) 2025-05-26 14:41:42 -07:00
pass_through_endpoints Fix: Respect user_header_name property for budget selection and user identification (#11419) 2025-06-06 14:21:02 -07:00
spend_tracking Litellm fix GitHub action testing (#11163) 2025-05-26 14:41:42 -07:00
test_configs Litellm fix GitHub action testing (#11163) 2025-05-26 14:41:42 -07:00
types_utils Litellm fix GitHub action testing (#11163) 2025-05-26 14:41:42 -07:00
ui_crud_endpoints Refactor settings response models in proxy_setting_endpoints.py 2025-06-04 15:08:05 -06:00
test_caching_routes.py Litellm fix GitHub action testing (#11163) 2025-05-26 14:41:42 -07:00
test_common_request_processing.py [Feat] DD Trace - Add instrumentation for streaming chunks (#11338) 2025-06-02 16:48:39 -07:00
test_custom_proxy.py Custom Root Path Improvements: don't require reserving /litellm route (#11460) 2025-06-05 16:36:47 -07:00
test_litellm_pre_call_utils.py Litellm fix GitHub action testing (#11163) 2025-05-26 14:41:42 -07:00
test_proxy_cli.py Enhance proxy CLI with Rich formatting and improved user experience (#11420) 2025-06-06 17:16:53 -07:00
test_proxy_server.py Litellm fix GitHub action testing (#11163) 2025-05-26 14:41:42 -07:00
test_proxy_utils.py UI / SSO - Update proxy admin id role in DB + Handle SSO redirects with custom root path (#11384) 2025-06-03 21:16:55 -07:00
test_route_llm_request.py Litellm fix GitHub action testing (#11163) 2025-05-26 14:41:42 -07:00
test_spend_log_cleanup.py [Feat] Make batch size for maximum retention in spend logs a controllable parameter (#11459) 2025-06-05 17:11:51 -07:00
test_team_member_update.py Litellm fix GitHub action testing (#11163) 2025-05-26 14:41:42 -07:00