phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-24 18:24:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

(test) promptlayer logger

Browse source
This commit is contained in:
ishaan-jaff 2023-11-02 08:17:59 -07:00
parent 2c18f7d374
commit 76c007d542
1 changed files with 4 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

7
litellm/tests/test_promptlayer_integration.py
View file

@ -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()