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

(test) langsmith logger with metadata

Browse source
This commit is contained in:
ishaan-jaff 2023-10-18 12:00:30 -07:00
parent 4258828969
commit 92b56de2db
1 changed files with 22 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

23
litellm/tests/test_langsmith.py
View file

@ -17,7 +17,7 @@ def test_langsmith_logging():
response = completion(model="claude-instant-1.2", response = completion(model="claude-instant-1.2",
messages=[{ messages=[{
"role": "user", "role": "user",
"content": "Hi 👋 - i'm claude" "content": "what llm are u"
}], }],
max_tokens=10, max_tokens=10,
temperature=0.2 temperature=0.2
@ -27,3 +27,24 @@ def test_langsmith_logging():
print(e) print(e)
test_langsmith_logging() test_langsmith_logging()
def test_langsmith_logging_with_metadata():
try:
response = completion(model="gpt-3.5-turbo",
messages=[{
"role": "user",
"content": "what llm are u"
}],
max_tokens=10,
temperature=0.2,
metadata={
"run_name": "litellmRUN",
"project_name": "litellm-completion",
}
)
print(response)
except Exception as e:
print(e)
test_langsmith_logging_with_metadata()