Ishaan Jaff
87f0201f84
test_handle_db_exception_with_connection_error
2025-03-26 20:44:25 -07:00
Ishaan Jaff
34e58be36d
fix order of _setup_prisma_client
2025-03-26 20:42:01 -07:00
Ishaan Jaff
427580eff5
fix _setup_prisma_client
2025-03-26 20:35:27 -07:00
Ishaan Jaff
15c04da735
refactor tests
2025-03-26 20:00:10 -07:00
Ishaan Jaff
88ef97b9d1
allow proxy to startup on DB unavailable
2025-03-26 19:50:57 -07:00
Ishaan Jaff
497570b2a6
bug fix - allow pods to startup when DB is unavailable
2025-03-26 19:49:51 -07:00
Ishaan Jaff
7142b0b610
refactor PrismaDBExceptionHandler
2025-03-26 19:22:24 -07:00
Ishaan Jaff and GitHub
0155b0eba2
Merge pull request #9533 from BerriAI/litellm_stability_fixes
...
[Reliability Fixes] - Gracefully handle exceptions when DB is having an outage
2025-03-26 18:57:38 -07:00
Ishaan Jaff
485aa87e65
allow_requests_on_db_unavailable
2025-03-26 18:48:18 -07:00
Ishaan Jaff
763f853a9f
docs fix
2025-03-26 18:32:58 -07:00
Ishaan Jaff
5242c5fbab
test - auth exception handler
2025-03-26 18:28:28 -07:00
Ishaan Jaff
15b1a8afb0
test_is_database_connection_error_prisma_errors
2025-03-26 18:27:39 -07:00
Krish Dholakia and GitHub
4351c77253
Support Gemini audio token cost tracking + fix openai audio input token cost tracking ( #9535 )
...
* fix(vertex_and_google_ai_studio_gemini.py): log gemini audio tokens in usage object
enables accurate cost tracking
* refactor(vertex_ai/cost_calculator.py): refactor 128k+ token cost calculation to only run if model info has it
Google has moved away from this for gemini-2.0 models
* refactor(vertex_ai/cost_calculator.py): migrate to usage object for more flexible data passthrough
* fix(llm_cost_calc/utils.py): support audio token cost tracking in generic cost per token
enables vertex ai cost tracking to work with audio tokens
* fix(llm_cost_calc/utils.py): default to total prompt tokens if text tokens field not set
* refactor(llm_cost_calc/utils.py): move openai cost tracking to generic cost per token
more consistent behaviour across providers
* test: add unit test for gemini audio token cost calculation
* ci: bump ci config
* test: fix test
2025-03-26 17:26:25 -07:00
Krrish Dholakia
04490c99d7
test: fix test
2025-03-26 17:12:09 -07:00
Krrish Dholakia
72c0ad419f
build(migration.sql): add migration file for new dailyusertable
...
documents prisma db changes
2025-03-26 17:11:25 -07:00
Krrish Dholakia
d4adc9764b
test(test_db_schema_migration.py): ci/cd test to enforce schema migrations are documented in .sql files
2025-03-26 16:59:50 -07:00
37265c8b1c
Add Daily User Spend Aggregate view - allows UI Usage tab to work > 1m rows ( #9538 )
...
* ci: update github action
* build(schema.prisma): enable daily user spend table
allows storing aggregate view of user's daily spend
* build(schema.prisma): add new daily user spend table
* feat: working daily user spend tracking
maintains an aggregate view for easier querying in high traffic
* setup_google_dns
* ci: update ci yaml
---------
Co-authored-by: Ishaan Jaff <ishaanjaffer0324@gmail.com >
2025-03-26 16:36:36 -07:00
Krrish Dholakia
fb9abd40f3
ci(config.yml): add pytest-postgres to ci/cd
2025-03-26 16:25:08 -07:00
Krish Dholakia and GitHub
132d3f7baa
feat(prisma-migrations): add baseline db migration file ( #9565 )
...
adds initial baseline db migration file
enables future schema changes to be documented via .sql files
2025-03-26 16:22:56 -07:00
Ishaan Jaff and GitHub
b7f4abd13a
Merge pull request #9542 from BerriAI/litellm_fix_vertex_ai_ft_models
...
[Feature]: Support for Fine-Tuned Vertex AI LLMs
2025-03-26 16:19:41 -07:00
Ishaan Jaff
23aa7f81b5
fix ProxyException
2025-03-26 15:53:33 -07:00
Ishaan Jaff
ff33ed020c
fix auth checks
2025-03-26 15:45:58 -07:00
Ishaan Jaff
8bd2081dec
fix get_key_object
2025-03-26 15:41:40 -07:00
Ishaan Jaff
f8caebc54b
is_database_connection_error
2025-03-26 14:55:40 -07:00
Ishaan Jaff
4948673e35
fix test changes
2025-03-26 14:51:33 -07:00
Ishaan Jaff
1812ce4a54
undo config.yml changes
2025-03-26 14:43:03 -07:00
Ishaan Jaff
044c56c439
Merge branch 'main' into litellm_stability_fixes
2025-03-26 14:41:43 -07:00
Ishaan Jaff
0aae9aa24a
rename _is_model_gemini_spec_model
2025-03-26 14:28:26 -07:00
Ishaan Jaff
c38b41f65b
test_get_supports_system_message
2025-03-26 14:26:08 -07:00
Ishaan Jaff
72f08bc6ea
unit tests for VertexGeminiConfig
2025-03-26 14:21:35 -07:00
Ishaan Jaff
8eaf4c55c0
test_gemini_fine_tuned_model_request_consistency
2025-03-26 14:18:11 -07:00
Ishaan Jaff
da9d849348
test_gemini_fine_tuned_model_request_consistency
2025-03-26 14:10:32 -07:00
Ishaan Jaff
82faa49668
doc fix Fine-tuned Models
2025-03-26 14:09:27 -07:00
Ishaan Jaff
28ab8fdccc
docs vertex ft model
2025-03-26 14:08:19 -07:00
Ishaan Jaff
e24a601897
docs verte ft models
2025-03-26 14:03:42 -07:00
Ishaan Jaff
82016eba0a
docs vertex ft models
2025-03-26 13:55:38 -07:00
Ishaan Jaff
12eb77d02d
docs litellm vertex ai ft models
2025-03-26 12:24:49 -07:00
Ishaan Jaff
93daf5cbac
_get_model_name_from_gemini_spec_model
2025-03-26 12:16:18 -07:00
801ecb6517
Nova Canvas complete image generation tasks ( #9177 ) ( #9525 )
...
* Nova Canvas complete image generation tasks (#9177 )
* add initial support for Amazon Nova Canvas model
Signed-off-by: omrishiv <327609+omrishiv@users.noreply.github.com >
* adjust name to AmazonNovaCanvas and map function variables to config
Signed-off-by: omrishiv <327609+omrishiv@users.noreply.github.com >
* tighten model name check
Signed-off-by: omrishiv <327609+omrishiv@users.noreply.github.com >
* fix quality mapping
Signed-off-by: omrishiv <327609+omrishiv@users.noreply.github.com >
* add premium quality in config
Signed-off-by: omrishiv <327609+omrishiv@users.noreply.github.com >
* support all Amazon Nova Canvas tasks
* remove unused import
Signed-off-by: omrishiv <327609+omrishiv@users.noreply.github.com >
* add tests for image generation tasks and fix payload
Signed-off-by: omrishiv <327609+omrishiv@users.noreply.github.com >
* add missing util file
Signed-off-by: omrishiv <327609+omrishiv@users.noreply.github.com >
* update model prices backup file
Signed-off-by: omrishiv <327609+omrishiv@users.noreply.github.com >
* remove image tasks other than text->image
Signed-off-by: omrishiv <327609+omrishiv@users.noreply.github.com >
* add color guided generation task for Nova Canvas
Signed-off-by: omrishiv <327609+omrishiv@users.noreply.github.com >
* fix merge
Signed-off-by: omrishiv <327609+omrishiv@users.noreply.github.com >
* add nova canvas image generation documentation
Signed-off-by: omrishiv <327609+omrishiv@users.noreply.github.com >
* add nova canvas unit tests
Signed-off-by: omrishiv <327609+omrishiv@users.noreply.github.com >
---------
Signed-off-by: omrishiv <327609+omrishiv@users.noreply.github.com >
Co-authored-by: Krish Dholakia <krrishdholakia@gmail.com >
* ci(config.yml): bump ci config
* test: fix test
---------
Signed-off-by: omrishiv <327609+omrishiv@users.noreply.github.com >
Co-authored-by: omrishiv <327609+omrishiv@users.noreply.github.com >
2025-03-26 11:28:20 -07:00
Ishaan Jaff
793a920caa
rename _is_model_gemini_spec_model
2025-03-26 11:14:51 -07:00
Ishaan Jaff
1822c45c0e
test fix test_pick_cheapest_chat_model_from_llm_provider
2025-03-26 11:13:21 -07:00
Ishaan Jaff
baa9b34950
Merge branch 'main' into litellm_fix_vertex_ai_ft_models
2025-03-26 11:11:54 -07:00
Ishaan Jaff
a2ae9be6a2
undo changes to utils
2025-03-26 11:09:29 -07:00
Ishaan Jaff
fee20250a6
pick_cheapest_chat_models_from_llm_provider
2025-03-26 10:59:02 -07:00
Ishaan Jaff
8a72b67b18
undo code changes
2025-03-26 10:57:08 -07:00
Ishaan Jaff
3ee7962f9c
fix llm request utils
2025-03-26 10:55:56 -07:00
Ishaan Jaff
bbe69a47a9
_is_model_gemini_gemini_spec_model
2025-03-26 10:53:23 -07:00
Ishaan Jaff
8c845847cd
fix vertex embedding perf test
2025-03-26 10:25:50 -07:00
Ishaan Jaff
5f51228b72
DNS lookup for Redis host
2025-03-26 09:28:59 -07:00
Ishaan Jaff
d7884114d0
setup_google_dns
2025-03-26 09:26:33 -07:00
Ishaan Jaff