mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-02 02:11:58 +00:00
test case to check if request params are stripped or sanitized when build request is called.
This commit is contained in:
parent
2c62dc0e3d
commit
d18190ed16
1 changed files with 16 additions and 0 deletions
|
|
@ -136,6 +136,22 @@ class TestTransformFileContent:
|
|||
assert isinstance(result, HttpxBinaryResponseContent)
|
||||
assert result.response.content == b'{"line": 1}\n{"line": 2}\n'
|
||||
|
||||
def test_should_preserve_alt_media_query_when_build_request_has_no_params(self, config):
|
||||
file_id = "gs://my-bucket/path/to/file.jsonl"
|
||||
url, params = config.transform_file_content_request(
|
||||
file_content_request={"file_id": file_id},
|
||||
optional_params={},
|
||||
litellm_params={},
|
||||
)
|
||||
|
||||
assert params == {}
|
||||
|
||||
client = httpx.Client()
|
||||
request = client.build_request("GET", url, headers={})
|
||||
|
||||
assert str(request.url) == url
|
||||
assert request.url.params.get("alt") == "media"
|
||||
|
||||
|
||||
class TestTransformDeleteFile:
|
||||
def test_should_build_correct_gcs_delete_url(self, config):
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue