diff --git a/docs/my-website/docs/observability/integrations.md b/docs/my-website/docs/observability/integrations.md index c6a158f694..2ebb8b9f2c 100644 --- a/docs/my-website/docs/observability/integrations.md +++ b/docs/my-website/docs/observability/integrations.md @@ -5,6 +5,7 @@ | Promptlayer | `PROMPLAYER_API_KEY` | `litellm.success_callback=["promptlayer"]` | | LLMonitor | `LLMONITOR_APP_ID` | `litellm.success_callback=["llmonitor"]` | | LangFuse | `LANGFUSE_PUBLIC_KEY`, `LANGFUSE_PRIVATE_KEY` | `litellm.success_callback=["langfuse"]` | +| Weights & Biases | `WANDB_API_KEY` | `litellm.success_callback=["wandb"]` | | Sentry | `SENTRY_API_URL` | `litellm.success_callback=["sentry"]` | | Posthog | `POSTHOG_API_KEY`,`POSTHOG_API_URL` | `litellm.success_callback=["posthog"]` | | Slack | `SLACK_API_TOKEN`,`SLACK_API_SECRET`,`SLACK_API_CHANNEL` | `litellm.success_callback=["slack"]` | diff --git a/docs/my-website/docs/observability/wandb_integration.md b/docs/my-website/docs/observability/wandb_integration.md new file mode 100644 index 0000000000..d760e44bd8 --- /dev/null +++ b/docs/my-website/docs/observability/wandb_integration.md @@ -0,0 +1,53 @@ +import Image from '@theme/IdealImage'; + +# Weights & Biases - Logging LLM Input/Output +Weights & Biases helps AI developers build better models faster https://wandb.ai + + + +:::info +We want to learn how we can make the callbacks better! Meet the LiteLLM [founders](https://calendly.com/d/4mp-gd3-k5k/berriai-1-1-onboarding-litellm-hosted-version) or +join our [discord](https://discord.gg/wuPM9dRgDw) +::: + +## Pre-Requisites +Ensure you have run `pip install wandb` for this integration +```shell +pip install wandb litellm +``` + +## Quick Start +Use just 2 lines of code, to instantly log your responses **across all providers** with Langfuse + + +Get your Langfuse API Keys from https://cloud.langfuse.com/ +```python +litellm.success_callback = ["wandb"] +``` +```python +# pip install langfuse +import litellm +import os + +os.environ["WANDB_API_KEY"] = "" +# LLM API Keys +os.environ['OPENAI_API_KEY']="" + +# set langfuse as a callback, litellm will send the data to langfuse +litellm.success_callback = ["wandb"] + +# openai call +response = litellm.completion( + model="gpt-3.5-turbo", + messages=[ + {"role": "user", "content": "Hi 👋 - i'm openai"} + ] +) +``` + +## Support & Talk to Founders + +- [Schedule Demo 👋](https://calendly.com/d/4mp-gd3-k5k/berriai-1-1-onboarding-litellm-hosted-version) +- [Community Discord 💭](https://discord.gg/wuPM9dRgDw) +- Our numbers 📞 +1 (770) 8783-106 / ‭+1 (412) 618-6238‬ +- Our emails ✉️ ishaan@berri.ai / krrish@berri.ai \ No newline at end of file diff --git a/docs/my-website/img/wandb.png b/docs/my-website/img/wandb.png new file mode 100644 index 0000000000..13b610ffe0 Binary files /dev/null and b/docs/my-website/img/wandb.png differ diff --git a/docs/my-website/sidebars.js b/docs/my-website/sidebars.js index ae9a3a094c..addcfe62e8 100644 --- a/docs/my-website/sidebars.js +++ b/docs/my-website/sidebars.js @@ -116,6 +116,7 @@ const sidebars = { "observability/custom_callback", "observability/sentry", "observability/promptlayer_integration", + "observability/wandb_integration", "observability/langfuse_integration", "observability/traceloop_integration", "observability/llmonitor_integration",