litellm/openai-proxy/utils.py

7 lines
266 B
Python

import os, litellm
import dotenv
dotenv.load_dotenv() # load env variables
def set_callbacks():
if ("LANGUFSE_PUBLIC_KEY" in os.environ and "LANGUFSE_SECRET_KEY" in os.environ) or "LANGFUSE_HOST" in os.environ:
litellm.success_callback = ["langfuse"]