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

..

.litellm_cache

refactor: move all testing to top-level of repo

2024-09-28 21:08:14 -07:00

example_config_yaml

(feat) - allow using os.environ/ vars for any value on config.yaml (#6276)

2024-10-19 09:00:27 +05:30

test_configs

Litellm ruff linting enforcement (#5992)

2024-10-01 19:44:20 -04:00

test_model_response_typing

LiteLLM Minor Fixes & Improvements (11/05/2024) (#6590)

2024-11-07 04:17:05 +05:30

adroit-crow-413218-bc47f303efc9.json

vertex testing use pathrise-convert-1606954137718

2025-01-05 14:00:17 -08:00

azure_fine_tune.jsonl

refactor: move all testing to top-level of repo

2024-09-28 21:08:14 -07:00

batch_job_results_furniture.jsonl

refactor: move all testing to top-level of repo

2024-09-28 21:08:14 -07:00

cache_unit_tests.py

(code refactor) - Add BaseRerankConfig. Use BaseRerankConfig for cohere/rerank and azure_ai/rerank (#7319)

2024-12-19 17:03:34 -08:00

conftest.py

ci(conftest.py): reset conftest.py for local_testing/ (#6657)

2024-11-08 19:14:16 +05:30

create_mock_standard_logging_payload.py

[Bug Fix]: Errors in LiteLLM When Using Embeddings Model with Usage-Based Routing (#7390)

2024-12-23 17:42:24 -08:00

data_map.txt

refactor: move all testing to top-level of repo

2024-09-28 21:08:14 -07:00

eagle.wav

refactor: move all testing to top-level of repo

2024-09-28 21:08:14 -07:00

gettysburg.wav

refactor: move all testing to top-level of repo

2024-09-28 21:08:14 -07:00

large_text.py

refactor: move all testing to top-level of repo

2024-09-28 21:08:14 -07:00

messages_with_counts.py

refactor: move all testing to top-level of repo

2024-09-28 21:08:14 -07:00

model_cost.json

refactor: move all testing to top-level of repo

2024-09-28 21:08:14 -07:00

openai_batch_completions_router.jsonl

refactor: move all testing to top-level of repo

2024-09-28 21:08:14 -07:00

openai_batch_completions.jsonl

refactor: move all testing to top-level of repo

2024-09-28 21:08:14 -07:00

speech_vertex.mp3

refactor: move all testing to top-level of repo

2024-09-28 21:08:14 -07:00

stream_chunk_testdata.py

refactor: move all testing to top-level of repo

2024-09-28 21:08:14 -07:00

test_acompletion_fallbacks.py

(core sdk fix) - fix fallbacks stuck in infinite loop (#7751)

2025-01-13 19:34:34 -08:00

test_acompletion.py

Complete o3 model support (#8183)

2025-02-02 22:36:37 -08:00

test_acooldowns_router.py

fix(pattern_match_deployments.py): default to user input if unable to… (#6632)

2024-11-08 00:55:57 +05:30

test_add_function_to_prompt.py

LiteLLM Minor Fixes & Improvements (11/05/2024) (#6590)

2024-11-07 04:17:05 +05:30

test_add_update_models.py

test: fix tests

2025-03-13 19:26:10 -07:00

test_aim_guardrails.py

Support post-call guards for stream and non-stream responses

2025-03-13 08:53:54 +02:00

test_alangfuse.py

all e2e langfuse tests now run on test_langfuse_e2e_test.p

2025-02-04 21:22:51 -08:00

test_amazing_vertex_completion.py

Support litellm.api_base for vertex_ai + gemini/ across completion, embedding, image_generation (#9516)

2025-03-25 23:46:20 -07:00

test_anthropic_prompt_caching.py

LiteLLM Minor Fixes & Improvements (01/16/2025) - p2 (#7828)

2025-02-02 23:17:50 -08:00

test_arize_ai.py

Merge branch 'main' into litellm_arize_dynamic_logging

2025-03-18 22:13:35 -07:00

test_arize_phoenix.py

Litellm contributor prs 02 24 2025 (#8775)

2025-02-24 18:55:48 -08:00

test_assistants.py

test_create_delete_assistants

2025-03-25 22:08:06 -07:00

test_async_fn.py

refactor(sagemaker/): separate chat + completion routes + make them b… (#7151)

2024-12-10 19:40:05 -08:00

test_audio_speech.py

Litellm dev 2024 12 20 p1 (#7335)

2024-12-20 21:22:31 -08:00

test_auth_utils.py

Ollama ssl verify = False + Spend Logs reliability fixes (#7931)

2025-01-23 23:05:41 -08:00

test_azure_content_safety.py

(refactor) caching use LLMCachingHandler for async_get_cache and set_cache (#6208)

2024-10-14 16:34:01 +05:30

test_azure_openai.py

Litellm dev 01 20 2025 p1 (#7884)

2025-01-20 21:45:48 -08:00

test_azure_perf.py

LiteLLM Minor Fixes & Improvements (11/29/2024) (#6965)

2024-12-01 05:24:11 -08:00

test_bad_params.py

(feat) Add cost tracking for /batches requests OpenAI (#7384)

2024-12-23 17:47:26 -08:00

test_basic_python_version.py

Litellm dev 01 10 2025 p2 (#7679)

2025-01-10 21:50:53 -08:00

test_batch_completion_return_exceptions.py

refactor: move all testing to top-level of repo

2024-09-28 21:08:14 -07:00

test_batch_completions.py

Litellm dev contributor prs 01 31 2025 (#8168)

2025-02-01 09:05:20 -08:00

test_blocked_user_list.py

(docs) add docstrings for all /key, /user, /team, /customer endpoints (#6804)

2024-11-18 19:44:06 -08:00

test_braintrust.py

Litellm dev 01 07 2025 p3 (#7635)

2025-01-08 11:46:24 -08:00

test_budget_manager.py

Litellm ruff linting enforcement (#5992)

2024-10-01 19:44:20 -04:00

test_caching_handler.py

(code refactor) - Add BaseRerankConfig. Use BaseRerankConfig for cohere/rerank and azure_ai/rerank (#7319)

2024-12-19 17:03:34 -08:00

test_caching_ssl.py

(redis fix) - fix AbstractConnection.__init__() got an unexpected keyword argument 'ssl' (#6908)

2024-11-25 22:52:44 -08:00

test_caching.py

test: improve flaky test

2025-03-24 23:15:04 -07:00

test_class.py

refactor: move all testing to top-level of repo

2024-09-28 21:08:14 -07:00

test_completion_cost.py

test: skip flaky tests

2025-03-11 19:43:04 -07:00

test_completion_with_retries.py

fix(main.py): fix retries being multiplied when using openai sdk (#7221)

2024-12-14 11:56:55 -08:00

test_completion.py

test_dynamic_azure_params

2025-03-18 17:26:23 -07:00

test_config.py

Litellm dev 01 29 2025 p2 (#8102)

2025-01-29 20:53:37 -08:00

test_cost_calc.py

test(test_cost_calc.py): fix test to handle llm api errors

2024-12-24 16:49:02 -08:00

test_custom_api_logger.py

refactor: move all testing to top-level of repo

2024-09-28 21:08:14 -07:00

test_custom_callback_input.py

test: loosen test

2025-03-17 09:44:22 -07:00

test_custom_callback_router.py

(Fixes) OpenAI Streaming Token Counting + Fixes usage track when litellm.turn_off_message_logging=True (#8156)

2025-01-31 15:06:37 -08:00

test_custom_llm.py

LiteLLM Minor Fixes & Improvements (01/10/2025) - p1 (#7670)

2025-01-10 17:49:05 -08:00

test_custom_logger.py

(Refactor / QA) - Use LoggingCallbackManager to append callbacks and ensure no duplicate callbacks are added (#8112)

2025-01-30 19:35:50 -08:00

test_disk_cache_unit_tests.py

LiteLLM Minor Fixes & Improvements (11/12/2024) (#6705)

2024-11-12 22:50:51 +05:30

test_dual_cache.py

(code refactor) - Add BaseRerankConfig. Use BaseRerankConfig for cohere/rerank and azure_ai/rerank (#7319)

2024-12-19 17:03:34 -08:00

test_dynamic_rate_limit_handler.py

LiteLLM Minor Fixes & Improvements (10/15/2024) (#6242)

2024-10-16 07:32:06 -07:00

test_dynamodb_logs.py

refactor: move all testing to top-level of repo

2024-09-28 21:08:14 -07:00

test_embedding.py

Add cohere v2/rerank support (#8421) (#8605)

2025-02-22 22:25:29 -08:00

test_exceptions.py

feat(azure.py): add azure bad request error support

2025-03-10 15:59:06 -07:00

test_file_types.py

refactor: move all testing to top-level of repo

2024-09-28 21:08:14 -07:00

test_function_call_parsing.py

refactor: move all testing to top-level of repo

2024-09-28 21:08:14 -07:00

test_function_calling.py

Support caching on reasoning content + other fixes (#8973)

2025-03-04 21:12:16 -08:00

test_function_setup.py

refactor: move all testing to top-level of repo

2024-09-28 21:08:14 -07:00

test_gcs_bucket.py

use latest bucket for testing

2025-01-05 14:55:48 -08:00

test_get_llm_provider.py

test(test_get_llm_provider.py): cover scenario where xai not in model name

2025-03-18 11:04:59 -07:00

test_get_model_file.py

LiteLLM Minor Fixes & Improvements (10/05/2024) (#6083)

2024-10-05 18:59:11 -04:00

test_get_model_info.py

add supports_web_search

2025-03-22 13:32:22 -07:00

test_get_optional_params_embeddings.py

refactor: move all testing to top-level of repo

2024-09-28 21:08:14 -07:00

test_get_optional_params_functions_not_supported.py

refactor: move all testing to top-level of repo

2024-09-28 21:08:14 -07:00

test_google_ai_studio_gemini.py

refactor: move all testing to top-level of repo

2024-09-28 21:08:14 -07:00

test_guardrails_ai.py

LiteLLM Minor Fixes & Improvements (10/15/2024) (#6242)

2024-10-16 07:32:06 -07:00

test_guardrails_config.py

feat(guardrails.py): return specific litellm params in /guardrails/list endpoint

2025-02-10 22:13:58 -08:00

test_health_check.py

feat(health_check.py): set upperbound for api when making health check call (#7865)

2025-01-18 19:47:43 -08:00

test_helicone_integration.py

refactor: move all testing to top-level of repo

2024-09-28 21:08:14 -07:00

test_hf_prompt_templates.py

(Refactor) Code Quality improvement - remove /prompt_templates/ , base_aws_llm.py from /llms folder (#7164)

2024-12-11 00:02:46 -08:00

test_http_parsing_utils.py

(Bug fix) - reading /parsing request body when on hypercorn (#8734)

2025-02-25 15:18:04 -08:00

test_img_resize.py

fix: Support WebP image format and avoid token calculation error (#7182)

2024-12-12 14:32:39 -08:00

test_lakera_ai_prompt_injection.py

Merge pull request #9222 from BerriAI/litellm_snowflake_pr_mar_13

2025-03-13 21:35:39 -07:00

test_langchain_ChatLiteLLM.py

refactor: move all testing to top-level of repo

2024-09-28 21:08:14 -07:00

test_langsmith.py

Litellm dev 11 30 2024 (#6974)

2024-12-02 21:03:33 -08:00

test_least_busy_routing.py

build: merge litellm_dev_03_01_2025_p2

2025-03-03 23:05:41 -08:00

test_litellm_max_budget.py

refactor: move all testing to top-level of repo

2024-09-28 21:08:14 -07:00

test_literalai.py

Litellm Minor Fixes & Improvements (10/03/2024) (#6049)

2024-10-03 18:02:28 -04:00

test_llm_guard.py

(refactor) caching use LLMCachingHandler for async_get_cache and set_cache (#6208)

2024-10-14 16:34:01 +05:30

test_load_test_router_s3.py

(refactor) caching use LLMCachingHandler for async_get_cache and set_cache (#6208)

2024-10-14 16:34:01 +05:30

test_loadtest_router.py

refactor: move all testing to top-level of repo

2024-09-28 21:08:14 -07:00

test_logfire.py

refactor: move all testing to top-level of repo

2024-09-28 21:08:14 -07:00

test_logging.py

LiteLLM Minor Fixes & Improvements (11/05/2024) (#6590)

2024-11-07 04:17:05 +05:30

test_longer_context_fallback.py

refactor: move all testing to top-level of repo

2024-09-28 21:08:14 -07:00

test_lowest_cost_routing.py

(refactor) caching use LLMCachingHandler for async_get_cache and set_cache (#6208)

2024-10-14 16:34:01 +05:30

test_lowest_latency_routing.py

(refactor) caching use LLMCachingHandler for async_get_cache and set_cache (#6208)

2024-10-14 16:34:01 +05:30

test_lunary.py

refactor: move all testing to top-level of repo

2024-09-28 21:08:14 -07:00

test_max_tpm_rpm_limiter.py

(refactor) caching use LLMCachingHandler for async_get_cache and set_cache (#6208)

2024-10-14 16:34:01 +05:30

test_mem_leak.py

LiteLLM Minor Fixes & Improvements (10/30/2024) (#6519)

2024-11-02 00:44:32 +05:30

test_mem_usage.py

refactor: move all testing to top-level of repo

2024-09-28 21:08:14 -07:00

test_mock_request.py

Litellm dev 02 12 2025 p1 (#8494)

2025-02-12 22:39:29 -08:00

test_model_alias_map.py

refactor: move all testing to top-level of repo

2024-09-28 21:08:14 -07:00

test_model_max_token_adjust.py

refactor: move all testing to top-level of repo

2024-09-28 21:08:14 -07:00

test_multiple_deployments.py

refactor: move all testing to top-level of repo

2024-09-28 21:08:14 -07:00

test_ollama_local_chat.py

refactor: move all testing to top-level of repo

2024-09-28 21:08:14 -07:00

test_ollama_local.py

refactor: move all testing to top-level of repo

2024-09-28 21:08:14 -07:00

test_ollama.py

Litellm dev 03 08 2025 p3 (#9089)

2025-03-09 18:20:56 -07:00

test_openai_moderations_hook.py

(refactor) caching use LLMCachingHandler for async_get_cache and set_cache (#6208)

2024-10-14 16:34:01 +05:30

test_opik.py

[Feat] Observability integration - Opik by Comet (#6062)

2024-10-10 18:27:50 +05:30

test_parallel_request_limiter.py

Add all /key/generate api params to UI + add metadata fields on team AND org add/update (#8667)

2025-02-19 21:13:06 -08:00

test_pass_through_endpoints.py

oops

2025-03-11 08:27:36 -04:00

test_presidio_masking.py

Litellm dev 12 06 2024 (#7067)

2024-12-06 22:44:18 -08:00

test_profiling_router.py

refactor: move all testing to top-level of repo

2024-09-28 21:08:14 -07:00

test_prometheus_service.py

(fix) PrometheusServicesLogger _get_metric should return metric in Registry (#6486)

2024-10-29 21:29:19 +05:30

test_prometheus.py

done

2025-01-22 20:19:31 +09:00

test_prompt_caching.py

LiteLLM Minor Fixes & Improvements (12/05/2024) (#7037)

2024-12-05 00:02:31 -08:00

test_prompt_injection_detection.py

(refactor) caching use LLMCachingHandler for async_get_cache and set_cache (#6208)

2024-10-14 16:34:01 +05:30

test_promptlayer_integration.py

LiteLLM Minor Fixes & Improvements (11/05/2024) (#6590)

2024-11-07 04:17:05 +05:30

test_provider_specific_config.py

test: update testing

2025-03-17 22:34:08 -07:00

test_pydantic_namespaces.py

refactor: move all testing to top-level of repo

2024-09-28 21:08:14 -07:00

test_pydantic.py

refactor: move all testing to top-level of repo

2024-09-28 21:08:14 -07:00

test_register_model.py

refactor: move all testing to top-level of repo

2024-09-28 21:08:14 -07:00

test_router_batch_completion.py

(code quality) run ruff rule to ban unused imports (#7313)

2024-12-19 12:33:42 -08:00

test_router_budget_limiter.py

Support budget/rate limit tiers for keys (#7429)

2024-12-26 19:05:27 -08:00

test_router_caching.py

Fix: Create RedisClusterCache when startup nodes provided in cache args of router (#9010)

2025-03-06 17:14:32 -08:00

test_router_client_init.py

test: fix test - delete env var before running

2025-03-11 20:57:57 -07:00

test_router_cooldowns.py

fix(route_llm_request.py): move to using common router, even for clie… (#8966)

2025-03-03 22:57:08 -08:00

test_router_custom_routing.py

refactor: move all testing to top-level of repo

2024-09-28 21:08:14 -07:00

test_router_debug_logs.py

refactor: move all testing to top-level of repo

2024-09-28 21:08:14 -07:00

test_router_fallback_handlers.py

(Feat) - return x-litellm-attempted-fallbacks in responses from litellm proxy (#8558)

2025-02-15 14:54:23 -08:00

test_router_fallbacks.py

(Feat) - return x-litellm-attempted-fallbacks in responses from litellm proxy (#8558)

2025-02-15 14:54:23 -08:00

test_router_get_deployments.py

build: merge litellm_dev_03_01_2025_p2

2025-03-03 23:05:41 -08:00

test_router_init.py

fix: remove client init tests for router - dup behaviour - provider caching already exists

2025-03-10 21:17:36 -07:00

test_router_max_parallel_requests.py

fix(lowest_tpm_rpm_routing.py): fix parallel rate limit check (#6577)

2024-11-05 22:03:44 +05:30

test_router_pattern_matching.py

(code quality) run ruff rule to ban unused imports (#7313)

2024-12-19 12:33:42 -08:00

test_router_policy_violation.py

refactor: move all testing to top-level of repo

2024-09-28 21:08:14 -07:00

test_router_retries.py

Litellm dev 12 14 2024 p1 (#7231)

2024-12-14 22:22:29 -08:00

test_router_tag_routing.py

build: merge litellm_dev_03_01_2025_p2

2025-03-03 23:05:41 -08:00

test_router_timeout.py

Litellm dev 01 20 2025 p3 (#7890)

2025-01-20 21:46:36 -08:00

test_router_utils.py

fix(router.py): add new test

2025-03-14 14:23:45 -07:00

test_router_with_fallbacks.py

refactor: move all testing to top-level of repo

2024-09-28 21:08:14 -07:00

test_router.py

test: remove redundant tests

2025-03-11 17:52:05 -07:00

test_rules.py

Litellm ruff linting enforcement (#5992)

2024-10-01 19:44:20 -04:00

test_sagemaker.py

test: mock sagemaker tests

2025-03-21 16:21:18 -07:00

test_scheduler.py

refactor: move all testing to top-level of repo

2024-09-28 21:08:14 -07:00

test_secret_detect_hook.py

test(base_llm_unit_tests.py): add test to ensure drop params is respe… (#8224)

2025-02-03 16:04:44 -08:00

test_simple_shuffle.py

refactor: move all testing to top-level of repo

2024-09-28 21:08:14 -07:00

test_spend_calculate_endpoint.py

refactor: move all testing to top-level of repo

2024-09-28 21:08:14 -07:00

test_stream_chunk_builder.py

Add anthropic thinking + reasoning content support (#8778)

2025-02-24 21:54:30 -08:00

test_streaming.py

Support openrouter reasoning_content on streaming (#9094)

2025-03-09 20:03:59 -07:00

test_supabase_integration.py

Litellm ruff linting enforcement (#5992)

2024-10-01 19:44:20 -04:00

test_team_config.py

refactor: move all testing to top-level of repo

2024-09-28 21:08:14 -07:00

test_text_completion.py

fix(ollama/completions/transformation.py): pass prompt, untemplated on /completions request

2025-03-17 18:35:44 -07:00

test_timeout.py

refactor: move all testing to top-level of repo

2024-09-28 21:08:14 -07:00

test_together_ai.py

refactor: move all testing to top-level of repo

2024-09-28 21:08:14 -07:00

test_token_counter.py

fix(utils.py): handle token counter error when invalid message passed in (#8670)

2025-02-19 22:21:34 -08:00

test_tpm_rpm_routing_v2.py

fix(response_metadata.py): log the litellm_model_name

2025-03-18 17:46:33 -07:00

test_traceloop.py

test: skip redundant test

2025-02-10 22:13:58 -08:00

test_ui_sso_helper_utils.py

LiteLLM Minor Fixes & Improvements (10/17/2024) (#6293)

2024-10-17 22:09:11 -07:00

test_unit_test_caching.py

(Bug fix) - don't log messages in model_parameters in StandardLoggingPayload (#8932)

2025-03-01 13:39:45 -08:00

test_update_spend.py

(docs) add docstrings for all /key, /user, /team, /customer endpoints (#6804)

2024-11-18 19:44:06 -08:00

test_validate_environment.py

refactor: move all testing to top-level of repo

2024-09-28 21:08:14 -07:00

test_wandb.py

LiteLLM Minor Fixes & Improvements (11/05/2024) (#6590)

2024-11-07 04:17:05 +05:30

test_whisper.py

Litellm dev 12 17 2024 p2 (#7277)

2024-12-17 14:05:14 -08:00

user_cost.json

refactor: move all testing to top-level of repo

2024-09-28 21:08:14 -07:00

vertex_ai.jsonl

refactor: move all testing to top-level of repo

2024-09-28 21:08:14 -07:00

vertex_batch_completions.jsonl

(feat) add Vertex Batches API support in OpenAI format (#7032)

2024-12-04 19:40:28 -08:00

vertex_key.json

ci/cd update vertex acct

2025-01-05 13:43:32 -08:00

whitelisted_bedrock_models.txt

Add anthropic3-7-sonnet (#8766)

2025-02-24 12:59:00 -08:00