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

(fix) failing prompt layer test

Browse source
This commit is contained in:
ishaan-jaff 2024-02-22 13:36:14 -08:00
parent b603d9784d
commit 365bba06e1
2 changed files with 11 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

11
litellm/tests/test_promptlayer_integration.py
View file

@ -9,8 +9,6 @@ import litellm
import pytest
litellm.success_callback = ["promptlayer"]
litellm.set_verbose = True
import time
# def test_promptlayer_logging():
@ -45,6 +43,8 @@ def test_promptlayer_logging_with_metadata():
# Redirect stdout
old_stdout = sys.stdout
sys.stdout = new_stdout = io.StringIO()
litellm.set_verbose = True
litellm.success_callback = ["promptlayer"]
response = completion(
model="gpt-3.5-turbo",
@ -69,6 +69,9 @@ def test_promptlayer_logging_with_metadata():
def test_promptlayer_logging_with_metadata_tags():
try:
# Redirect stdout
litellm.set_verbose = True
litellm.success_callback = ["promptlayer"]
old_stdout = sys.stdout
sys.stdout = new_stdout = io.StringIO()
@ -78,7 +81,7 @@ def test_promptlayer_logging_with_metadata_tags():
temperature=0.2,
max_tokens=20,
metadata={"model": "ai21", "pl_tags": ["env:dev"]},
mock_response="this is a mock response"
mock_response="this is a mock response",
)
# Restore stdout
@ -92,8 +95,6 @@ def test_promptlayer_logging_with_metadata_tags():
except Exception as e:
pytest.fail(f"Error occurred: {e}")
test_promptlayer_logging_with_metadata()
test_promptlayer_logging_with_metadata_tags()
# def test_chat_openai():
# try: