add colab badge

wandb · Aug 12, 2024 · 858e6d7 · 858e6d7
1 parent 8a64310
commit 858e6d7
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/docs/docs/guides/cookbooks/codegen/index.md b/docs/docs/guides/cookbooks/codegen/index.md
@@ -4,6 +4,8 @@ hide_table_of_contents: true
 
 # Code Generation using Weave and OpenAI
 
+[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/wandb/weave/blob/anish/codegen-testing-coobook/examples/cookbooks/codegen/codegen.ipynb)
+
 Generating high-quality code with proper structure, documentation, and tests is a challenging task. This guide demonstrates how to implement a code generation pipeline using Weave, a powerful framework for building, tracking, and evaluating LLM applications. By combining OpenAI's language models with Weave's robust tooling, you'll learn to create a code generation pipeline that produces high-quality Python functions, complete with main program runners and unit tests, while gaining insights into the generation process.
 
 Our pipeline will:
@@ -21,7 +23,7 @@ For a visual demonstration of the code generation pipeline using Weave, Groq, an
 
 <iframe width="560" height="315" src="https://www.youtube.com/embed/B70jJYPVAzE?si=75Z4Fg_DBxAiu9_6&amp" title="YouTube video player" frameborder="0" allow="accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture; web-share" referrerpolicy="strict-origin-when-cross-origin" allowfullscreen></iframe>
 
-This video provides a step-by-step walkthrough of the process, showcasing how Weave integrates with Groq to create a powerful code generation tool and then running the code in E2B, to validate the code.
+This video provides a step-by-step walkthrough of the process, showcasing how Weave integrates with Groq to create a powerful code generation tool and then running the code in E2B, to validate the code. We use OpenAI in the following example, but you can use any LLM provider with Weave.
 
 ## Why use Weave?
 

diff --git a/examples/cookbooks/codegen/codegen.ipynb b/examples/cookbooks/codegen/codegen.ipynb
@@ -1,5 +1,14 @@
 {
  "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/master/examples/cookbooks/codegen/codegen.ipynb)\n",
+    "\n",
+    "<!--- @wandbcode{codegen-cookbook} -->"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,