Krish Dholakia ac4f32fb1e Cost tracking for gemini-2.5-pro (#9837) ...

* build(model_prices_and_context_window.json): add google/gemini-2.0-flash-lite-001 versioned pricing

Closes https://github.com/BerriAI/litellm/issues/9829

* build(model_prices_and_context_window.json): add initial support for 'supported_output_modalities' param

* build(model_prices_and_context_window.json): add initial support for 'supported_output_modalities' param

* build(model_prices_and_context_window.json): add supported endpoints to gemini-2.5-pro

* build(model_prices_and_context_window.json): add gemini 200k+ pricing

* feat(utils.py): support cost calculation for gemini-2.5-pro above 200k tokens

Fixes https://github.com/BerriAI/litellm/issues/9807

* build: test dockerfile change

* build: revert apk change

* ci(config.yml): pip install wheel

* ci: test problematic package first

* ci(config.yml): pip install only binary

* ci: try more things

* ci: test different ml_dtypes version

* ci(config.yml): check ml_dtypes==0.4.0

* ci: test

* ci: cleanup config.yml

* ci: specify ml dtypes in requirements.txt

* ci: remove redisvl depedency (temporary)

* fix: fix linting errors

* test: update test

* test: fix test

2025-04-09 18:48:43 -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

example.jsonl

VertexAI non-jsonl file storage support (#9781)

2025-04-09 14:01:48 -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

Allow team admins to add/update/delete models on UI + show api base and model id on request logs (#9572)

2025-03-27 12:06:31 -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

Fix VertexAI Credential Caching issue (#9756)

2025-04-04 16:38:08 -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

test_completion_invalid_param_cohere

2025-04-02 06:49:11 -07: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

test(test_caching_handler.py): move to in-memory cache - prevent redis flakiness from impacting ci/cd

2025-03-28 13:32:04 -07: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_redis_caching_llm_caching_ttl

2025-04-02 21:54:35 -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

databricks/databricks-meta-llama-3-3-70b-instruct

2025-04-07 20:16:24 -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

Allow passing thinking param to litellm proxy via client sdk + Code QA Refactor on get_optional_params (get correct values) (#9386)

2025-04-07 21:04:11 -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 fixes - azure deprecated dall-e-2

2025-04-02 20:56:20 -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 inference providers support for Hugging Face (#8258) (#9738) (#9773)

2025-04-05 10:50:15 -07: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

Litellm dev 04 05 2025 p2 (#9774)

2025-04-07 21:02:52 -07: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

VertexAI non-jsonl file storage support (#9781)

2025-04-09 14:01:48 -07: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

Cost tracking for gemini-2.5-pro (#9837)

2025-04-09 18:48:43 -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_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 fixes - azure deprecated dall-e-2

2025-04-02 20:56:20 -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

test_batch_update_spend

2025-04-01 07:12:29 -07: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

Add OpenAI gpt-4o-transcribe support (#9517)

2025-03-26 23:10:25 -07: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