cleaning up client, and fixing replicate bug

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

View file

@ -2,10 +2,9 @@ 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