Krish Dholakia 179d2f56b7 LiteLLM Minor Fixes & Improvements (12/16/2024) - p1 (#7263) ...

* fix(factory.py): skip empty text blocks for bedrock user messages

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

* Add support for Gemini 2.0 GoogleSearch tool (#7257)

* Add support for google_search tool in gemini 2.0

* Add/modify tests

* Fix grounding check

* Remove 2.0 grounding test; exclude experimental model in VERTEX_MODELS_TO_NOT_TEST

* Swap order of tools

* DFix formatting

* fix(get_api_base.py): return api base in streaming response

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

Closes https://github.com/BerriAI/litellm/pull/7250

* fix(cost_calculator.py): only set base model to model if not none

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

* fix(cost_calculator.py): enforce stricter order when picking model for cost calculation

* fix(cost_calculator.py): fix '_select_model_name_for_cost_calc' to return model name with region name prefix if provided

* fix(utils.py): fix 'get_model_info()' to handle edge case where model name starts with custom llm provider AND custom llm provider is given

* fix(cost_calculator.py): handle `custom_llm_provider-` scenario

* fix(cost_calculator.py): e2e working tts cost tracking

ensures initial message is passed in, to cost calculator

* fix(factory.py): suppress linting errors

* fix(cost_calculator.py): strip llm provider from model name after selecting cost calc model

* fix(litellm_logging.py): store initial request in 'input' field + accept base_model to be passed in litellm_params directly

* test: handle none env var value in flaky test

* fix(litellm_logging.py): fix linting errors

---------

Co-authored-by: Sam B <samlingx@gmail.com>

2024-12-17 15:33:36 -08: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

refactor: move all testing to top-level of repo

2024-09-28 21:08:14 -07: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

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

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

conftest.py

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

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

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

log.txt

LiteLLM Minor Fixes & Improvements (10/02/2024) (#6023)

2024-10-02 22:00:28 -04: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.py

refactor: move all testing to top-level of repo

2024-09-28 21:08:14 -07: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

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

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

test_alangfuse.py

test_langfuse_masked_input_output

2024-11-21 22:59:36 -08:00

test_alerting.py

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

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

test_amazing_s3_logs.py

Revert "(perf) move s3 logging to Batch logging + async [94% faster p… (#6275)

2024-10-17 16:14:57 +05:30

test_amazing_vertex_completion.py

LiteLLM Minor Fixes & Improvements (12/16/2024) - p1 (#7263)

2024-12-17 15:33:36 -08:00

test_anthropic_prompt_caching.py

(fix) latency fix - revert prompt caching check on litellm router (#7211)

2024-12-12 20:50:16 -08:00

test_arize_ai.py

(feat) Arize - Allow using Arize HTTP endpoint (#6364)

2024-10-23 09:38:35 +05:30

test_assistants.py

rename llms/OpenAI/ -> llms/openai/ (#7154)

2024-12-10 20:14:07 -08: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 Minor Fixes & Improvements (12/16/2024) - p1 (#7263)

2024-12-17 15:33:36 -08:00

test_auth_checks.py

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

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

test_auth_utils.py

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

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

test_aws_secret_manager.py

(Feat) Add support for storing virtual keys in AWS SecretManager (#6728)

2024-11-14 09:25:07 -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

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

2024-12-11 00:02:46 -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

Litellm ruff linting enforcement (#5992)

2024-10-01 19:44:20 -04: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 merge pr (#7161)

2024-12-10 22:49:26 -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

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

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

test_budget_manager.py

Litellm ruff linting enforcement (#5992)

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

test_caching_handler.py

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

2024-12-05 00:02:31 -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

Provider Budget Routing - Get Budget, Spend Details (#7063)

2024-12-06 21:14:12 -08: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

LiteLLM Minor Fixes & Improvements (12/16/2024) - p1 (#7263)

2024-12-17 15:33:36 -08: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

LITELLM: Remove requests library usage (#7235)

2024-12-17 12:50:04 -08:00

test_config.py

Litellm dev 12 12 2024 (#7203)

2024-12-13 08:54:03 -08:00

test_cost_calc.py

(feat) add bedrock/stability.stable-image-ultra-v1:0 (#6723)

2024-11-14 14:47:15 -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

LiteLLM Minor Fixes & Improvement (11/14/2024) (#6730)

2024-11-15 01:02:54 +05:30

test_custom_callback_router.py

Litellm ruff linting enforcement (#5992)

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

test_custom_llm.py

Litellm Minor Fixes & Improvements (10/12/2024) (#6179)

2024-10-12 11:48:34 -07:00

test_custom_logger.py

build(config.yml): add codecov to repo (#6172)

2024-10-12 14:48:17 -07: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

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

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

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

feat(databricks/chat): support structured outputs on databricks

2024-12-02 23:08:19 -08:00

test_exceptions.py

Litellm 12 02 2024 (#6994)

2024-12-02 22:00:01 -08:00

test_file_types.py

refactor: move all testing to top-level of repo

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

test_fine_tuning_api.py

Code Quality Improvement - use vertex_ai/ as folder name for vertexAI (#7166)

2024-12-11 00:32:41 -08: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 12 02 2024 (#6994)

2024-12-02 22:00:01 -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

LiteLLM Minor Fixes & Improvements (12/16/2024) - p1 (#7263)

2024-12-17 15:33:36 -08:00

test_get_llm_provider.py

LiteLLM Minor Fixes & Improvement (11/14/2024) (#6730)

2024-11-15 01:02:54 +05:30

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

LiteLLM Minor Fixes & Improvements (12/16/2024) - p1 (#7263)

2024-12-17 15:33:36 -08: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_get_secret.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

refactor: move all testing to top-level of repo

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

test_health_check.py

fix(acompletion): support fallbacks on acompletion (#7184)

2024-12-11 19:20:54 -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

LiteLLM Minor Fixes & Improvements (11/27/2024) (#6943)

2024-11-28 00:32:46 +05:30

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

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

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

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

(router_strategy/) ensure all async functions use async cache methods (#6489)

2024-10-29 21:07:17 +05:30

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 ruff linting enforcement (#5992)

2024-10-01 19:44:20 -04: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 ollama refactor (#7162)

2024-12-10 21:45:35 -08:00

test_openai_batches_and_files.py

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

2024-12-04 19:40:28 -08: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

(minor fix proxy) Clarify Proxy Rate limit errors are showing hash of litellm virtual key (#7210)

2024-12-12 20:13:14 -08:00

test_pass_through_endpoints.py

fix test_aaapass_through_endpoint_pass_through_keys_langfuse

2024-11-21 22:05:00 -08: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

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

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

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

Litellm merge pr (#7161)

2024-12-10 22:49:26 -08: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_python_38.py

Litellm dev 11 08 2024 (#6658)

2024-11-08 22:07:17 +05:30

test_register_model.py

refactor: move all testing to top-level of repo

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

test_rerank.py

LiteLLM Minor Fixes & Improvements (11/23/2024) (#6870)

2024-11-23 15:17:40 +05:30

test_router_batch_completion.py

Litellm dev 11 21 2024 (#6837)

2024-11-22 01:53:52 +05:30

test_router_budget_limiter.py

(feat) Add Tag-based budgets on litellm router / proxy (#7236)

2024-12-14 17:28:36 -08:00

test_router_caching.py

refactor: move all testing to top-level of repo

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

test_router_client_init.py

(fix) litellm router.aspeech (#6962)

2024-12-05 13:39:50 -08:00

test_router_cooldowns.py

Litellm dev 12 07 2024 (#7086)

2024-12-08 00:30:33 -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

Litellm router max depth (#6501)

2024-10-29 22:05:41 -07:00

test_router_fallbacks.py

Litellm dev 12 11 2024 v2 (#7215)

2024-12-13 12:49:57 -08:00

test_router_get_deployments.py

refactor: move all testing to top-level of repo

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

test_router_init.py

LiteLLM Minor Fixes & Improvements (11/27/2024) (#6943)

2024-11-28 00:32:46 +05:30

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

Litellm dev 12 12 2024 (#7203)

2024-12-13 08:54:03 -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

refactor: move all testing to top-level of repo

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

test_router_timeout.py

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

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

test_router_utils.py

LiteLLM Minor Fixes & Improvements (11/26/2024) (#6913)

2024-11-28 00:01:38 +05:30

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

Litellm dev 12 07 2024 (#7086)

2024-12-08 00:30:33 -08:00

test_rules.py

Litellm ruff linting enforcement (#5992)

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

test_sagemaker.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_scheduler.py

refactor: move all testing to top-level of repo

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

test_secret_detect_hook.py

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

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

test_secret_manager.py

Code Quality Improvement - remove tokenizers/ from /llms (#7163)

2024-12-10 23:50:15 -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

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

2024-12-06 14:29:53 -08:00

test_streaming.py

LiteLLM Minor Fixes & Improvements (12/16/2024) - p1 (#7263)

2024-12-17 15:33:36 -08: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

test: update hf test to check if client closed

2024-12-12 11:34:50 -08: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: Support WebP image format and avoid token calculation error (#7182)

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

test_tpm_rpm_routing_v2.py

LiteLLM Minor Fixes & Improvements (11/26/2024) (#6913)

2024-11-28 00:01:38 +05:30

test_traceloop.py

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

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

test_triton.py

build: Squashed commit of https://github.com/BerriAI/litellm/pull/7170

2024-12-11 01:03:57 -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

(perf) Litellm redis router fix - ~100ms improvement (#6483)

2024-10-29 13:58:29 -07: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_utils.py

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

2024-12-12 14:32:39 -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

refactor: move all testing to top-level of repo

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