(docs) add wandb callback

This commit is contained in:
ishaan-jaff 2023-10-17 18:15:58 -07:00
parent da3625e2f3
commit 0aa3563ccf
4 changed files with 55 additions and 0 deletions

View file

@ -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"]` |

View file

@ -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
<Image img={require('../../img/wandb.png')} />
:::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

Binary file not shown.

After

Width:  |  Height:  |  Size: 242 KiB

View file

@ -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",