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

(test) comment out traceloop test

Browse source
This commit is contained in:
ishaan-jaff 2023-10-27 11:27:38 -07:00
parent de29df1c7f
commit 05795551f2
1 changed files with 54 additions and 54 deletions
Download patch file Download diff file Expand all files Collapse all files

108
litellm/tests/test_traceloop.py
View file

@ -1,62 +1,62 @@
import sys
import os
import io
#
sys.path.insert(0, os.path.abspath('../..'))
import litellm
from litellm import completion
from traceloop.sdk import Traceloop
Traceloop.init(app_name="test_traceloop", disable_batch=True, traceloop_sync_enabled=False)
litellm.success_callback = ["traceloop"]
# import sys
# import os
# import io
# #
# sys.path.insert(0, os.path.abspath('../..'))
# import litellm
# from litellm import completion
# from traceloop.sdk import Traceloop
# Traceloop.init(app_name="test_traceloop", disable_batch=True, traceloop_sync_enabled=False)
# litellm.success_callback = ["traceloop"]
def test_traceloop_logging():
try:
print('making completion call')
response = completion(
model="claude-instant-1.2",
messages=[
{"role": "user", "content": "Tell me a joke about OpenTelemetry"}
],
max_tokens=10,
temperature=0.2,
)
print(response)
except Exception as e:
print(e)
# def test_traceloop_logging():
# try:
# print('making completion call')
# response = completion(
# model="claude-instant-1.2",
# messages=[
# {"role": "user", "content": "Tell me a joke about OpenTelemetry"}
# ],
# max_tokens=10,
# temperature=0.2,
# )
# print(response)
# except Exception as e:
# print(e)
# test_traceloop_logging()
# # test_traceloop_logging()
def test_traceloop_tracing_function_calling():
function1 = [
{
"name": "get_current_weather",
"description": "Get the current weather in a given location",
"parameters": {
"type": "object",
"properties": {
"location": {
"type": "string",
"description": "The city and state, e.g. San Francisco, CA",
},
"unit": {"type": "string", "enum": ["celsius", "fahrenheit"]},
},
"required": ["location"],
},
}
]
try:
response = completion(
model="gpt-3.5-turbo",
messages=[{"role": "user", "content": "what's the weather in boston"}],
temperature=0.1,
functions=function1,
)
print(response)
except Exception as e:
print(e)
# def test_traceloop_tracing_function_calling():
# function1 = [
# {
# "name": "get_current_weather",
# "description": "Get the current weather in a given location",
# "parameters": {
# "type": "object",
# "properties": {
# "location": {
# "type": "string",
# "description": "The city and state, e.g. San Francisco, CA",
# },
# "unit": {"type": "string", "enum": ["celsius", "fahrenheit"]},
# },
# "required": ["location"],
# },
# }
# ]
# try:
# response = completion(
# model="gpt-3.5-turbo",
# messages=[{"role": "user", "content": "what's the weather in boston"}],
# temperature=0.1,
# functions=function1,
# )
# print(response)
# except Exception as e:
# print(e)
# test_traceloop_tracing_function_calling()
# # test_traceloop_tracing_function_calling()