diff --git a/api/pyproject.toml b/api/pyproject.toml
index 5aeef81a2a..d7587e12cd 100644
--- a/api/pyproject.toml
+++ b/api/pyproject.toml
@@ -40,7 +40,7 @@ dependencies = [
     "numpy~=1.26.4",
     "openpyxl~=3.1.5",
     "opik~=1.10.37",
-    "litellm==1.82.6",                                    # Pinned to avoid madoka dependency issue
+    "litellm==1.82.6", # Pinned to avoid madoka dependency issue
     "opentelemetry-api==1.28.0",
     "opentelemetry-distro==0.49b0",
     "opentelemetry-exporter-otlp==1.28.0",
@@ -93,6 +93,7 @@ dependencies = [
     "weave>=0.52.25",
     "fastopenapi[flask]>=0.7.0",
     "bleach~=6.3.0",
+    "wandb>=0.25.1",
 ]
 # Before adding new dependency, consider place it in
 # alphabet order (a-z) and suitable group.
diff --git a/api/uv.lock b/api/uv.lock
index f648a39f43..95e8552a07 100644
--- a/api/uv.lock
+++ b/api/uv.lock
@@ -1545,6 +1545,7 @@ dependencies = [
     { name = "tiktoken" },
     { name = "transformers" },
     { name = "unstructured", extra = ["docx", "epub", "md", "ppt", "pptx"] },
+    { name = "wandb" },
     { name = "weave" },
     { name = "weaviate-client" },
     { name = "webvtt-py" },
@@ -1748,6 +1749,7 @@ requires-dist = [
     { name = "tiktoken", specifier = "~=0.12.0" },
     { name = "transformers", specifier = "~=5.3.0" },
     { name = "unstructured", extras = ["docx", "epub", "md", "ppt", "pptx"], specifier = "~=0.21.5" },
+    { name = "wandb", specifier = ">=0.25.1" },
     { name = "weave", specifier = ">=0.52.25" },
     { name = "weaviate-client", specifier = "==4.20.4" },
     { name = "webvtt-py", specifier = "~=0.5.1" },