adding berrispend integration

This commit is contained in:
Krrish Dholakia 2023-08-05 16:11:45 -07:00
parent d3a2ac808a
commit a6e236459d
10 changed files with 292 additions and 11 deletions

View file

@ -0,0 +1,25 @@
#### What this tests ####
# This tests if logging to the helicone integration actually works
import sys, os
import traceback
import pytest
sys.path.insert(0, os.path.abspath('../..')) # Adds the parent directory to the system path
import litellm
from litellm import embedding, completion
litellm.success_callback = ["berrispend"]
litellm.failure_callback = ["berrispend"]
litellm.set_verbose = True
user_message = "Hello, how are you?"
messages = [{ "content": user_message,"role": "user"}]
#openai call
response = completion(model="gpt-3.5-turbo", messages=[{"role": "user", "content": "Hi 👋 - i'm openai"}])
#bad request call
response = completion(model="chatgpt-test", messages=[{"role": "user", "content": "Hi 👋 - i'm a bad request"}])