forked from phoenix/litellm-mirror
test: cleanup tests
This commit is contained in:
parent
ee3e3dfe7d
commit
1aaee81ad6
2 changed files with 0 additions and 35 deletions
|
@ -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)
|
|
||||||
|
|
|
@ -144,7 +144,6 @@ async def test_team_2logging():
|
||||||
langfuse_client = langfuse.Langfuse(
|
langfuse_client = langfuse.Langfuse(
|
||||||
public_key=os.getenv("LANGFUSE_PROJECT2_PUBLIC"),
|
public_key=os.getenv("LANGFUSE_PROJECT2_PUBLIC"),
|
||||||
secret_key=os.getenv("LANGFUSE_PROJECT2_SECRET"),
|
secret_key=os.getenv("LANGFUSE_PROJECT2_SECRET"),
|
||||||
host="https://us.cloud.langfuse.com",
|
|
||||||
)
|
)
|
||||||
|
|
||||||
await asyncio.sleep(10)
|
await asyncio.sleep(10)
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue