diff --git a/docs/docs/guides/cookbooks/codegen/index.md b/docs/docs/guides/cookbooks/codegen/index.md
index 4fadc4ca75c..1baee991b04 100644
--- 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
index 85440ff2ce1..9281de9094f 100644
--- 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,