mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-08 03:08:45 +00:00
(test) test langsmith logger
This commit is contained in:
parent
994979a3f4
commit
c9763789f3
1 changed files with 29 additions and 0 deletions
29
litellm/tests/test_langsmith.py
Normal file
29
litellm/tests/test_langsmith.py
Normal file
|
|
@ -0,0 +1,29 @@
|
|||
import sys
|
||||
import os
|
||||
import io
|
||||
|
||||
sys.path.insert(0, os.path.abspath('../..'))
|
||||
|
||||
from litellm import completion
|
||||
import litellm
|
||||
|
||||
litellm.success_callback = ["langsmith"]
|
||||
# litellm.set_verbose = True
|
||||
import time
|
||||
|
||||
|
||||
def test_langsmith_logging():
|
||||
try:
|
||||
response = completion(model="claude-instant-1.2",
|
||||
messages=[{
|
||||
"role": "user",
|
||||
"content": "Hi 👋 - i'm claude"
|
||||
}],
|
||||
max_tokens=10,
|
||||
temperature=0.2
|
||||
)
|
||||
print(response)
|
||||
except Exception as e:
|
||||
print(e)
|
||||
|
||||
test_langsmith_logging()
|
||||
Loading…
Add table
Reference in a new issue