add colab badges

wandb · Aug 12, 2024 · b58812b · b58812b
1 parent 2197c36
commit b58812b
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 3 deletions.
diff --git a/docs/docs/guides/cookbooks/summarization/chain-of-density/index.md b/docs/docs/guides/cookbooks/summarization/chain-of-density/index.md
@@ -1,5 +1,7 @@
 # Summarization using Chain of Density
 
+[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/wandb/weave/blob/anish/add-summarization-example/examples/cookbooks/summarization/chain_of_density.ipynb)
+
 Summarizing complex technical documents while preserving crucial details is a challenging task. The Chain of Density (CoD) summarization technique offers a solution by iteratively refining summaries to be more concise and information-dense. This guide demonstrates how to implement CoD using Weave, a powerful framework for building, tracking, and evaluating LLM applications. By combining CoD's effectiveness with Weave's robust tooling, you'll learn to create a summarization pipeline that produces high-quality, entity-rich summaries of technical content while gaining insights into the summarization process.
 
 ![Final Evaluation](./media/eval_comparison.gif)

diff --git a/examples/cookbooks/summarization/chain_of_density.ipynb b/examples/cookbooks/summarization/chain_of_density.ipynb
@@ -1,5 +1,23 @@
 {
  "cells": [
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/wandb/weave/blob/anish/add-summarization-example/examples/cookbooks/summarization/chain_of_density.ipynb)\n",
+    "\n",
+    "<!--- @wandbcode{summarization-chain-of-density-cookbook} -->"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "!pip install -qU anthropic weave pydantic requests PyPDF2 set-env-colab-kaggle-dotenv"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,
@@ -10,14 +28,15 @@
     "import anthropic\n",
     "import weave\n",
     "from datetime import datetime, timezone\n",
-    "from dotenv import load_dotenv\n",
     "from pydantic import BaseModel\n",
     "import requests\n",
     "import io\n",
     "from PyPDF2 import PdfReader\n",
+    "from set_env import set_env\n",
+    "\n",
+    "set_env(\"WANDB_API_KEY\")\n",
+    "set_env(\"ANTHROPIC_API_KEY\")\n",
     "\n",
-    "load_dotenv()\n",
-    "# Setup\n",
     "weave.init(\"summarization-chain-of-density-cookbook\")\n",
     "anthropic_client = anthropic.Anthropic(api_key=os.getenv(\"ANTHROPIC_API_KEY\"))"
    ]