mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-27 03:34:10 +00:00
(test) OTEL / traceloop - waiting for async support
This commit is contained in:
parent
11a8713a50
commit
01aa8941a5
1 changed files with 49 additions and 0 deletions
49
litellm/tests/test_traceloop.py
Normal file
49
litellm/tests/test_traceloop.py
Normal file
|
@ -0,0 +1,49 @@
|
|||
# Commented out for now - since traceloop break ci/cd
|
||||
# import sys
|
||||
# import os
|
||||
# import io, asyncio
|
||||
|
||||
# sys.path.insert(0, os.path.abspath('../..'))
|
||||
|
||||
# from litellm import completion
|
||||
# import litellm
|
||||
# litellm.num_retries = 3
|
||||
# litellm.success_callback = [""]
|
||||
# import time
|
||||
# import pytest
|
||||
# from traceloop.sdk import Traceloop
|
||||
# Traceloop.init(app_name="test-litellm", disable_batch=True)
|
||||
|
||||
|
||||
# def test_traceloop_logging():
|
||||
# try:
|
||||
# litellm.set_verbose = True
|
||||
# response = litellm.completion(
|
||||
# model="gpt-3.5-turbo",
|
||||
# messages=[{"role": "user", "content":"This is a test"}],
|
||||
# max_tokens=1000,
|
||||
# temperature=0.7,
|
||||
# timeout=5,
|
||||
# )
|
||||
# print(f"response: {response}")
|
||||
# except Exception as e:
|
||||
# pytest.fail(f"An exception occurred - {e}")
|
||||
# # test_traceloop_logging()
|
||||
|
||||
|
||||
# # def test_traceloop_logging_async():
|
||||
# # try:
|
||||
# # litellm.set_verbose = True
|
||||
# # async def test_acompletion():
|
||||
# # return await litellm.acompletion(
|
||||
# # model="gpt-3.5-turbo",
|
||||
# # messages=[{"role": "user", "content":"This is a test"}],
|
||||
# # max_tokens=1000,
|
||||
# # temperature=0.7,
|
||||
# # timeout=5,
|
||||
# # )
|
||||
# # response = asyncio.run(test_acompletion())
|
||||
# # print(f"response: {response}")
|
||||
# # except Exception as e:
|
||||
# # pytest.fail(f"An exception occurred - {e}")
|
||||
# # test_traceloop_logging_async()
|
Loading…
Add table
Add a link
Reference in a new issue