litellm/litellm/integrations
Krish Dholakia 1e403a8447
Litellm dev 10 29 2024 (#6502)
* fix(core_helpers.py): return None, instead of raising kwargs is None error

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

* docs(cost_tracking.md): cleanup doc

* fix(vertex_and_google_ai_studio.py): handle function call with no params passed in

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

* test(test_router_timeout.py): add test for router timeout + retry logic

* test: update test to use module level values

* (fix) Prometheus - Log Postgres DB latency, status on prometheus  (#6484)

* fix logging DB fails on prometheus

* unit testing log to otel wrapper

* unit testing for service logger + prometheus

* use LATENCY buckets for service logging

* fix service logging

* docs clarify vertex vs gemini

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

* fix router strat

* use async set / get cache in router_strategy

* add coverage for router strategy

* fix imports

* fix batch_get_cache

* use async methods for least busy

* fix least busy use async methods

* fix test_dual_cache_increment

* test async_get_available_deployment when routing_strategy="least-busy"

* (fix) proxy - fix when `STORE_MODEL_IN_DB` should be set (#6492)

* set store_model_in_db at the top

* correctly use store_model_in_db global

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

* fix logging DB fails on prometheus

* unit testing log to otel wrapper

* unit testing for service logger + prometheus

* use LATENCY buckets for service logging

* fix service logging

* fix _get_metric in prom services logger

* add clear doc string

* unit testing for prom service logger

* bump: version 1.51.0 → 1.51.1

* Add `azure/gpt-4o-mini-2024-07-18` to model_prices_and_context_window.json (#6477)

* Update utils.py (#6468)

Fixed missing keys

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

* docs(exception_mapping.md): add missing exception types

Fixes https://github.com/Aider-AI/aider/issues/2120#issuecomment-2438971183

* fix(main.py): register custom model pricing with specific key

Ensure custom model pricing is registered to the specific model+provider key combination

* test: make testing more robust for custom pricing

* fix(redis_cache.py): instrument otel logging for sync redis calls

ensures complete coverage for all redis cache calls

* refactor: pass parent_otel_span for redis caching calls in router

allows for more observability into what calls are causing latency issues

* test: update tests with new params

* refactor: ensure e2e otel tracing for router

* refactor(router.py): add more otel tracing acrosss router

catch all latency issues for router requests

* fix: fix linting error

* fix(router.py): fix linting error

* fix: fix test

* test: fix tests

* fix(dual_cache.py): pass ttl to redis cache

* fix: fix param

* perf(cooldown_cache.py): improve cooldown cache, to store cache results in memory for 5s, prevents redis call from being made on each request

reduces 100ms latency per call with caching enabled on router

* fix: fix test

* fix(cooldown_cache.py): handle if a result is None

* fix(cooldown_cache.py): add debug statements

* refactor(dual_cache.py): move to using an in-memory check for batch get cache, to prevent redis from being hit for every call

* fix(cooldown_cache.py): fix linting erropr

* refactor(prometheus.py): move to using standard logging payload for reading the remaining request / tokens

Ensures prometheus token tracking works for anthropic as well

* fix: fix linting error

* fix(redis_cache.py): make sure ttl is always int (handle float values)

Fixes issue where redis_client.ex was not working correctly due to float ttl

* fix: fix linting error

* test: update test

* fix: fix linting error

---------

Co-authored-by: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Co-authored-by: Xingyao Wang <xingyao@all-hands.dev>
Co-authored-by: vibhanshu-ob <115142120+vibhanshu-ob@users.noreply.github.com>
2024-10-29 22:04:16 -07:00
..
_types feat - arize ai open inference types 2024-07-22 11:07:48 -07:00
datadog (Feat) New Logging integration - add Datadog LLM Observability support (#6449) 2024-10-28 22:01:32 +05:30
email_templates fix - move email templates 2024-05-31 10:37:56 -07:00
gcs_bucket (refactor) use helper function _assemble_complete_response_from_streaming_chunks to assemble complete responses in caching and logging callbacks (#6220) 2024-10-15 12:45:12 +05:30
langfuse (fix) Langfuse key based logging (#6372) 2024-10-23 18:24:22 +05:30
opik (code quality) add ruff check PLR0915 for too-many-statements (#6309) 2024-10-18 15:36:49 +05:30
prometheus_helpers feat(prometheus_api.py): support querying prometheus metrics for all-up + key-level spend on UI (#5782) 2024-09-18 22:39:15 -07:00
SlackAlerting redis otel tracing + async support for latency routing (#6452) 2024-10-28 21:52:12 -07:00
__init__.py add linting 2023-08-18 11:05:05 -07:00
argilla.py feat(custom_logger.py): expose new async_dataset_hook for modifying… (#6331) 2024-10-20 09:00:04 -07:00
arize_ai.py (feat) Arize - Allow using Arize HTTP endpoint (#6364) 2024-10-23 09:38:35 +05:30
athina.py Litellm ruff linting enforcement (#5992) 2024-10-01 19:44:20 -04:00
braintrust_logging.py (code quality) add ruff check PLR0915 for too-many-statements (#6309) 2024-10-18 15:36:49 +05:30
custom_batch_logger.py Revert "(perf) move s3 logging to Batch logging + async [94% faster p… (#6275) 2024-10-17 16:14:57 +05:30
custom_guardrail.py LiteLLM Minor Fixes & Improvements (10/15/2024) (#6242) 2024-10-16 07:32:06 -07:00
custom_logger.py redis otel tracing + async support for latency routing (#6452) 2024-10-28 21:52:12 -07:00
dynamodb.py Litellm ruff linting enforcement (#5992) 2024-10-01 19:44:20 -04:00
email_alerting.py Add pyright to ci/cd + Fix remaining type-checking errors (#6082) 2024-10-05 17:04:00 -04:00
galileo.py Litellm ruff linting enforcement (#5992) 2024-10-01 19:44:20 -04:00
greenscale.py Litellm ruff linting enforcement (#5992) 2024-10-01 19:44:20 -04:00
helicone.py Litellm ruff linting enforcement (#5992) 2024-10-01 19:44:20 -04:00
lago.py Litellm ruff linting enforcement (#5992) 2024-10-01 19:44:20 -04:00
langsmith.py (code quality) add ruff check PLR0915 for too-many-statements (#6309) 2024-10-18 15:36:49 +05:30
langtrace.py Feat: Add Langtrace integration (#5341) 2024-10-11 19:19:53 +05:30
literal_ai.py fix literal ai typing errors 2024-10-09 15:23:39 +05:30
logfire_logger.py Litellm ruff linting enforcement (#5992) 2024-10-01 19:44:20 -04:00
lunary.py Add pyright to ci/cd + Fix remaining type-checking errors (#6082) 2024-10-05 17:04:00 -04:00
openmeter.py Add pyright to ci/cd + Fix remaining type-checking errors (#6082) 2024-10-05 17:04:00 -04:00
opentelemetry.py redis otel tracing + async support for latency routing (#6452) 2024-10-28 21:52:12 -07:00
prometheus.py Litellm dev 10 29 2024 (#6502) 2024-10-29 22:04:16 -07:00
prometheus_services.py (fix) PrometheusServicesLogger _get_metric should return metric in Registry (#6486) 2024-10-29 21:29:19 +05:30
prompt_layer.py Litellm ruff linting enforcement (#5992) 2024-10-01 19:44:20 -04:00
s3.py Revert "(perf) move s3 logging to Batch logging + async [94% faster p… (#6275) 2024-10-17 16:14:57 +05:30
supabase.py Litellm ruff linting enforcement (#5992) 2024-10-01 19:44:20 -04:00
test_httpx.py fix(utils.py): improved predibase exception mapping 2024-06-08 14:32:43 -07:00
traceloop.py Add pyright to ci/cd + Fix remaining type-checking errors (#6082) 2024-10-05 17:04:00 -04:00
weights_biases.py Litellm ruff linting enforcement (#5992) 2024-10-01 19:44:20 -04:00