mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-27 11:43:54 +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