phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 03:04:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat(databricks/chat/transformation.py): add tools and 'tool_choice' param support (#8076)
All checks were successful
Read Version from pyproject.toml / read-version (push) Successful in 38s
Details

Browse source 
* feat(databricks/chat/transformation.py): add tools and 'tool_choice' param support

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

* refactor: cleanup redundant file

* test: mark flaky test

* test: mark all parallel request tests as flaky
This commit is contained in:
Krish Dholakia 2025-01-29 21:09:07 -08:00 • committed by GitHub
parent 9fa44a4fbe
commit ba8ba9eddb
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 92 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

8
tests/local_testing/test_parallel_request_limiter.py
View file

@ -454,6 +454,7 @@ Test with Router
"""
@pytest.mark.flaky(retries=6, delay=2)
@pytest.mark.asyncio
async def test_normal_router_call():
model_list = [
@ -528,6 +529,7 @@ async def test_normal_router_call():
)
@pytest.mark.flaky(retries=6, delay=2)
@pytest.mark.asyncio
async def test_normal_router_tpm_limit():
import logging
@ -615,6 +617,7 @@ async def test_normal_router_tpm_limit():
assert e.status_code == 429
@pytest.mark.flaky(retries=6, delay=2)
@pytest.mark.asyncio
async def test_streaming_router_call():
model_list = [
@ -690,6 +693,7 @@ async def test_streaming_router_call():
)
@pytest.mark.flaky(retries=6, delay=2)
@pytest.mark.asyncio
async def test_streaming_router_tpm_limit():
litellm.set_verbose = True
@ -845,6 +849,7 @@ async def test_bad_router_call():
)
@pytest.mark.flaky(retries=6, delay=2)
@pytest.mark.asyncio
async def test_bad_router_tpm_limit():
model_list = [
@ -923,6 +928,7 @@ async def test_bad_router_tpm_limit():
)
@pytest.mark.flaky(retries=6, delay=2)
@pytest.mark.asyncio
async def test_bad_router_tpm_limit_per_model():
model_list = [
@ -1023,6 +1029,7 @@ async def test_bad_router_tpm_limit_per_model():
)
@pytest.mark.flaky(retries=6, delay=2)
@pytest.mark.asyncio
async def test_pre_call_hook_rpm_limits_per_model():
"""
@ -1101,6 +1108,7 @@ async def test_pre_call_hook_rpm_limits_per_model():
)
@pytest.mark.flaky(retries=6, delay=2)
@pytest.mark.asyncio
async def test_pre_call_hook_tpm_limits_per_model():
"""