phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

(Feat) 273% improvement GCS Bucket Logger - use Batched Logging (#6679)

Browse source 
* use CustomBatchLogger for GCS

* add GCS bucket logging type

* use batch logging for GCs bucket

* add gcs_bucket

* allow setting flush_interval on CustomBatchLogger

* set GCS_FLUSH_INTERVAL to 1s

* fix test_key_logging

* fix test_key_logging

* add docs on new env vars
This commit is contained in:
Ishaan Jaff 2024-11-10 22:05:34 -08:00 • committed by GitHub
parent 70aa85af1f
commit eb92ed4156
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
9 changed files with 128 additions and 72 deletions
Download patch file Download diff file Expand all files Collapse all files

1
tests/local_testing/test_gcs_bucket.py
View file

@ -28,6 +28,7 @@ verbose_logger.setLevel(logging.DEBUG)
def load_vertex_ai_credentials():
# Define the path to the vertex_key.json file
print("loading vertex ai credentials")
os.environ["GCS_FLUSH_INTERVAL"] = "1"
filepath = os.path.dirname(os.path.abspath(__file__))
vertex_key_path = filepath + "/adroit-crow-413218-bc47f303efc9.json"