litellm-mirror/litellm/tests/test_promptlayer_integration.py
2023-08-31 11:04:17 -07:00

47 lines
1.1 KiB
Python

# #### What this tests ####
# # This tests if logging to the llmonitor integration actually works
# # Adds the parent directory to the system path
# import sys
# import os
# sys.path.insert(0, os.path.abspath('../..'))
# from litellm import completion, embedding
# import litellm
# litellm.success_callback = ["promptlayer"]
# litellm.set_verbose = True
# def test_chat_openai():
# try:
# response = completion(model="gpt-3.5-turbo",
# messages=[{
# "role": "user",
# "content": "Hi 👋 - i'm openai"
# }])
# print(response)
# except Exception as e:
# print(e)
# def test_chat_openai():
# litellm.success_callback = ["langfuse"]
# try:
# response = completion(model="gpt-3.5-turbo",
# messages=[{
# "role": "user",
# "content": "Hi 👋 - i'm openai"
# }])
# print(response)
# except Exception as e:
# print(e)
# test_chat_openai()