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

test: cleanup tests

Browse source
This commit is contained in:
Krrish Dholakia 2024-06-19 23:02:56 -07:00
parent ee3e3dfe7d
commit 1aaee81ad6
2 changed files with 0 additions and 35 deletions
Download patch file Download diff file Expand all files Collapse all files

34
litellm/tests/test_amazing_langfuse.py
View file

@ -867,37 +867,3 @@ async def test_make_request():
}
},
)
def test_alangfuse_dynamic_logging():
"""
pass in langfuse credentials via completion call
assert call is logged.
Covers the team-logging scenario.
"""
import uuid
import langfuse
trace_id = str(uuid.uuid4())
_ = litellm.completion(
model="gpt-3.5-turbo",
messages=[{"role": "user", "content": "Hey"}],
mock_response="Hey! how's it going?",
langfuse_public_key=os.getenv("LANGFUSE_PROJECT2_PUBLIC"),
langfuse_secret_key=os.getenv("LANGFUSE_PROJECT2_SECRET"),
langfuse_host="https://us.cloud.langfuse.com",
metadata={"trace_id": trace_id},
success_callback=["langfuse"],
)
time.sleep(1)
langfuse_client = langfuse.Langfuse(
public_key=os.getenv("LANGFUSE_PROJECT2_PUBLIC"),
secret_key=os.getenv("LANGFUSE_PROJECT2_SECRET"),
host="https://us.cloud.langfuse.com",
)
langfuse_client.get_trace(id=trace_id)

1
tests/test_team_logging.py
View file

@ -144,7 +144,6 @@ async def test_team_2logging():
langfuse_client = langfuse.Langfuse(
public_key=os.getenv("LANGFUSE_PROJECT2_PUBLIC"),
secret_key=os.getenv("LANGFUSE_PROJECT2_SECRET"),
host="https://us.cloud.langfuse.com",
)
await asyncio.sleep(10)