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

cleaning up client, and fixing replicate bug

Browse source
This commit is contained in:
Krrish Dholakia 2023-07-31 13:25:28 -07:00
parent 22374705b0
commit 745252e3bf
6 changed files with 216 additions and 186 deletions
Download patch file Download diff file Expand all files Collapse all files

20
litellm/tests/test_client.py
View file

@ -2,27 +2,19 @@ import sys, os
import traceback
sys.path.append('..') # Adds the parent directory to the system path
import main
from main import litellm_client
client = litellm_client(success_callback=["posthog"], failure_callback=["slack", "sentry", "posthog"], verbose=True)
completion = client.completion
embedding = client.embedding
from main import embedding, completion
main.success_callback = ["posthog"]
main.failure_callback = ["slack", "sentry", "posthog"]
main.set_verbose = True
# main.set_verbose = True
def logger_fn(model_call_object: dict):
print(f"model call details: {model_call_object}")
# print(f"model call details: {model_call_object}")
pass
user_message = "Hello, how are you?"
messages = [{ "content": user_message,"role": "user"}]
# test on openai completion call
try:
response = completion(model="gpt-3.5-turbo", messages=messages, logger_fn=logger_fn)
except:
print(f"error occurred: {traceback.format_exc()}")
pass
# test on openai completion call
try:
response = completion(model="gpt-3.5-turbo", messages=messages, logger_fn=logger_fn)