forked from phoenix/litellm-mirror
(docs) add wandb callback
This commit is contained in:
parent
da3625e2f3
commit
0aa3563ccf
4 changed files with 55 additions and 0 deletions
|
@ -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"]` |
|
||||
|
|
53
docs/my-website/docs/observability/wandb_integration.md
Normal file
53
docs/my-website/docs/observability/wandb_integration.md
Normal 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
|
BIN
docs/my-website/img/wandb.png
Normal file
BIN
docs/my-website/img/wandb.png
Normal file
Binary file not shown.
After Width: | Height: | Size: 242 KiB |
|
@ -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",
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue