scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-08 22:21:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 78

test gcs bucket

Browse source
This commit is contained in:
Ishaan Jaff 2024-08-05 08:56:10 -07:00
parent a4eb32f4f3
commit a6e4fdf502
1 changed files with 10 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
litellm/tests/test_gcs_bucket.py
View file

@ -15,7 +15,7 @@ import pytest
import litellm
from litellm import completion
from litellm._logging import verbose_logger
from litellm.integrations.gcs_bucket import GCSBucketLogger
from litellm.integrations.gcs_bucket import GCSBucketLogger, GCSBucketPayload
verbose_logger.setLevel(logging.DEBUG)
@ -87,9 +87,15 @@ async def test_basic_gcs_logger():
object_from_gcs = json.loads(object_from_gcs)
print("object_from_gcs", object_from_gcs)
assert object_from_gcs["request_id"] == response.id
assert object_from_gcs["call_type"] == "acompletion"
assert object_from_gcs["model"] == "gpt-3.5-turbo"
gcs_payload = GCSBucketPayload(**object_from_gcs)
print("gcs_payload", gcs_payload)
assert gcs_payload["request_kwargs"]["model"] == "gpt-3.5-turbo"
assert gcs_payload["request_kwargs"]["messages"] == [
{"role": "user", "content": "This is a test"}
]
assert gcs_payload["response_obj"]["choices"][0]["message"]["content"] == "Hi!"
# Delete Object from GCS
print("deleting object from GCS")