scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-08 03:08:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 115

test fixes

Browse source
This commit is contained in:
Ishaan Jaffer 2025-10-04 14:23:01 -07:00
parent 4d95241397
commit 3dcacb881b
2 changed files with 26 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
tests/pass_through_unit_tests/test_pass_through_unit_tests.py
View file

@ -45,6 +45,18 @@ def mock_request():
class QueryParams:
def __init__(self):
self._dict = {}
def __iter__(self):
return iter(self._dict.items())
def items(self):
return self._dict.items()
def keys(self):
return self._dict.keys()
def values(self):
return self._dict.values()
class MockRequest:
def __init__(

16
tests/test_litellm/passthrough/test_passthrough_main.py
View file

@ -182,6 +182,18 @@ def mock_request():
class QueryParams:
def __init__(self):
self._dict = {}
def __iter__(self):
return iter(self._dict.items())
def items(self):
return self._dict.items()
def keys(self):
return self._dict.keys()
def values(self):
return self._dict.values()
class MockRequest:
def __init__(
@ -291,7 +303,7 @@ async def test_pass_through_request_stream_param_override(
"POST",
httpx.URL("https://api.anthropic.com/v1/messages"),
json=request_body,
params=None,
params={},
headers={
"Authorization": "Bearer test-key"
},
@ -393,7 +405,7 @@ async def test_pass_through_request_stream_param_no_override(
headers={
"Authorization": "Bearer test-key"
},
params=None,
params={},
json=request_body,
)