litellm/tests/litellm
Krish Dholakia 6fd18651d1
Support litellm.api_base for vertex_ai + gemini/ across completion, embedding, image_generation (#9516)
* test(tests): add unit testing for litellm_proxy integration

* fix(cost_calculator.py): fix tracking cost in sdk when calling proxy

* fix(main.py): respect litellm.api_base on `vertex_ai/` and `gemini/` routes

* fix(main.py): consistently support custom api base across gemini + vertexai on embedding + completion

* feat(vertex_ai/): test

* fix: fix linting error

* test: set api base as None before starting loadtest
2025-03-25 23:46:20 -07:00
..
caching update redisvl dependency 2025-03-24 08:42:11 -04:00
experimental_mcp_client test_tools.py 2025-03-21 18:38:24 -07:00
integrations TestCustomPromptManagement 2025-03-19 17:40:15 -07:00
litellm_core_utils test: add unit test 2025-03-24 14:45:20 -07:00
llms fix vertex ai multimodal embedding translation (#9471) 2025-03-24 23:23:28 -07:00
proxy Merge branch 'main' into litellm_exp_mcp_server 2025-03-24 19:03:56 -07:00
rerank_api build: merge squashed commit 2025-02-26 09:39:27 -08:00
responses TestResponseAPILoggingUtils 2025-03-12 12:45:03 -07:00
router_strategy refactor(base_routing_strategy.py): fix function names 2025-03-18 22:41:02 -07:00
log.txt Support format param for specifying image type (#9019) 2025-03-05 19:52:53 -08:00
readme.md (UI + Proxy) Cache Health Check Page - Cleanup/Improvements (#8665) 2025-02-19 19:08:50 -08:00
test_cost_calculator.py Support litellm.api_base for vertex_ai + gemini/ across completion, embedding, image_generation (#9516) 2025-03-25 23:46:20 -07:00
test_main.py Support format param for specifying image type (#9019) 2025-03-05 19:52:53 -08:00

Testing for litellm/

This directory 1:1 maps the the litellm/ directory, and can only contain mocked tests.

The point of this is to:

  1. Increase test coverage of litellm/
  2. Make it easy for contributors to add tests for the litellm/ package and easily run tests without needing LLM API keys.

File name conventions

  • litellm/proxy/test_caching_routes.py maps to litellm/proxy/caching_routes.py
  • test_<filename>.py maps to litellm/<filename>.py