diff --git a/litellm/tests/test_gcs_bucket.py b/litellm/tests/test_gcs_bucket.py new file mode 100644 index 00000000000..7ee917c450c --- /dev/null +++ b/litellm/tests/test_gcs_bucket.py @@ -0,0 +1,39 @@ +import io +import os +import sys + +sys.path.insert(0, os.path.abspath("../..")) + +import asyncio +import logging +import uuid + +import pytest + +import litellm +from litellm import completion +from litellm._logging import verbose_logger +from litellm.integrations.gcs_bucket import GCSBucketLogger + +verbose_logger.setLevel(logging.DEBUG) + + +gcs_logger = GCSBucketLogger() +print("GCSBucketLogger", gcs_logger) + + +@pytest.mark.asyncio +async def test_basic_gcs_logger(): + litellm.callbacks = [gcs_logger] + response = await litellm.acompletion( + model="gpt-3.5-turbo", + temperature=0.7, + messages=[{"role": "user", "content": "This is a test"}], + max_tokens=10, + user="ishaan-2", + mock_response="Hi!", + ) + + print("response", response) + + await asyncio.sleep(5)