Commit graph

38 commits

Author SHA1 Message Date
Krish Dholakia
dd602753c0
Litellm fix router testing (#5748)
* test: fix testing - azure changed content policy error logic

* test: fix tests to use mock responses

* test(test_image_generation.py): handle api instability

* test(test_image_generation.py): handle azure api instability

* fix(utils.py): fix unbounded variable error

* fix(utils.py): fix unbounded variable error

* test: refactor test to use mock response

* test: mark flaky azure tests
2024-09-17 18:02:23 -07:00
Krish Dholakia
60709a0753
LiteLLM Minor Fixes and Improvements (09/13/2024) (#5689)
* refactor: cleanup unused variables + fix pyright errors

* feat(health_check.py): Closes https://github.com/BerriAI/litellm/issues/5686

* fix(o1_reasoning.py): add stricter check for o-1 reasoning model

* refactor(mistral/): make it easier to see mistral transformation logic

* fix(openai.py): fix openai o-1 model param mapping

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

* feat(main.py): infer finetuned gemini model from base model

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

* docs(vertex.md): update docs to call finetuned gemini models

* feat(proxy_server.py): allow admin to hide proxy model aliases

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

* docs(load_balancing.md): add docs on hiding alias models from proxy config

* fix(base.py): don't raise notimplemented error

* fix(user_api_key_auth.py): fix model max budget check

* fix(router.py): fix elif

* fix(user_api_key_auth.py): don't set team_id to empty str

* fix(team_endpoints.py): fix response type

* test(test_completion.py): handle predibase error

* test(test_proxy_server.py): fix test

* fix(o1_transformation.py): fix max_completion_token mapping

* test(test_image_generation.py): mark flaky test
2024-09-14 10:02:55 -07:00
Ishaan Jaff
5f025390bb test test_image_generation_azure_dall_e_3 2024-08-27 22:52:11 -07:00
Krrish Dholakia
fd3a6f0870 test(test_image_generation.py): handle azure api error 2024-08-21 21:46:00 -07:00
Ishaan Jaff
62c1595497 testing skip internal server errors 2024-08-08 19:09:46 -07:00
Krish Dholakia
bca71019ad
Merge branch 'main' into litellm_anthropic_api_streaming 2024-08-03 21:16:50 -07:00
Krrish Dholakia
23cad65256 test: fix test 2024-08-03 20:28:11 -07:00
Ishaan Jaff
1c1631222c fix test test_aimage_generation_vertex_ai 2024-08-03 18:32:52 -07:00
Krrish Dholakia
c982ec88d8 fix(bedrock.py): fix response format for bedrock image generation response
Fixes https://github.com/BerriAI/litellm/issues/5010
2024-08-03 09:46:49 -07:00
Krrish Dholakia
a1986fab60 fix(vertex_httpx.py): add sync vertex image gen support
Fixes https://github.com/BerriAI/litellm/issues/4623
2024-07-09 13:33:54 -07:00
Krrish Dholakia
ac00639301 fix(azure.py): fix httpx call for dall e 3 2024-07-02 18:30:21 -07:00
Krrish Dholakia
cf5334fe8a refactor(azure.py): refactor sync azure calls to httpx 2024-07-02 17:06:48 -07:00
Krrish Dholakia
589c1c6280 refactor(azure.py): replaces the custom transport logic for just using our httpx client
Done to fix all the http/https proxy issues people are facing with proxy.
2024-07-02 15:32:53 -07:00
Krrish Dholakia
2e49614230 fix(utils.py): exception map vertex ai 500 internal server errors 2024-06-10 21:37:54 -07:00
Krrish Dholakia
6e9bca59b0 fix(utils.py): fix exception mapping for azure internal server error 2024-06-06 17:12:30 -07:00
Krrish Dholakia
8625280c32 test(test_image_generation.py): fix azure dall e test 2024-06-03 23:32:19 -07:00
Ishaan Jaff
f20184cd49 (ci/cd) run again 2024-05-20 18:50:17 -07:00
Ishaan Jaff
518db13982 add parameter mapping with vertex ai 2024-05-20 13:28:20 -07:00
Ishaan Jaff
571d4cf569 test - fix test_aimage_generation_vertex_ai 2024-05-20 12:10:28 -07:00
Ishaan Jaff
6ddc9873e5 test vertex image gen test 2024-05-20 11:14:16 -07:00
Ishaan Jaff
2da89a0c8e fix vertex test 2024-05-20 10:51:25 -07:00
Ishaan Jaff
2519879e67 add ImageObject 2024-05-20 10:45:37 -07:00
Ishaan Jaff
24951d44a4 feat - working httpx requests vertex ai image gen 2024-05-20 09:51:15 -07:00
Ishaan Jaff
5ba5f15b56 test - test_aimage_generation_vertex_ai 2024-05-20 08:14:43 -07:00
Krrish Dholakia
285a3733a9 test(test_image_generation.py): fix test 2024-04-30 12:14:29 -07:00
Krrish Dholakia
00d1440d0d test(test_image_generation.py): change img model for test - bedrock EOL 2024-04-30 08:55:40 -07:00
Ishaan Jaff
365497e860 (fix) OpenAI img gen endpoints unstable 2024-03-28 12:42:04 -07:00
Krrish Dholakia
478307d4cf fix(bedrock.py): support anthropic messages api on bedrock (claude-3) 2024-03-04 17:15:47 -08:00
Krrish Dholakia
4c951d20bc test: removing aws tests - account suspended - pending their approval 2024-02-28 13:46:20 -08:00
Krrish Dholakia
36416360c4 feat(bedrock.py): add stable diffusion image generation support 2024-02-03 12:08:38 -08:00
Krrish Dholakia
f3dd0c0563 test(test_image_generation.py): ignore content violation errors for image gen test 2024-02-02 18:05:39 -08:00
ishaan-jaff
c0b56b6575 (test) catch litellm.ContentPolicyViolationError 2024-01-09 17:04:04 +05:30
Krrish Dholakia
e99a41307a test: testing fixes 2024-01-09 10:23:34 +05:30
ishaan-jaff
08525ce200 (ci/cd) use 3 retries for image generation 2024-01-09 10:07:09 +05:30
Krrish Dholakia
4905929de3 refactor: add black formatting 2023-12-25 14:11:20 +05:30
Krrish Dholakia
f59b9436be feat(main.py): add async image generation support 2023-12-20 16:58:15 +05:30
Krrish Dholakia
b3962e483f feat(azure.py): add support for azure image generations endpoint 2023-12-20 16:37:21 +05:30
Krrish Dholakia
13d088b72e feat(main.py): add support for image generation endpoint 2023-12-16 21:07:29 -08:00