From 82e93db77419fe3e16df95bd6793d24f6c767a4c Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Thu, 2 Nov 2023 08:17:59 -0700 Subject: [PATCH] (test) promptlayer logger --- litellm/tests/test_promptlayer_integration.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/litellm/tests/test_promptlayer_integration.py b/litellm/tests/test_promptlayer_integration.py index 6d3bbe105c..8c919bb1e8 100644 --- a/litellm/tests/test_promptlayer_integration.py +++ b/litellm/tests/test_promptlayer_integration.py @@ -46,12 +46,13 @@ def test_promptlayer_logging_with_metadata(): old_stdout = sys.stdout sys.stdout = new_stdout = io.StringIO() - - response = completion(model="j2-light", + response = completion(model="gpt-3.5-turbo", messages=[{ "role": "user", "content": "Hi 👋 - i'm ai21" }], + temperature=0.2, + max_tokens=20, metadata={"model": "ai21"}) # Restore stdout @@ -65,7 +66,7 @@ def test_promptlayer_logging_with_metadata(): except Exception as e: print(e) -# test_promptlayer_logging_with_metadata() +test_promptlayer_logging_with_metadata()