diff --git a/.speakeasy/workflow.yaml b/.speakeasy/workflow.yaml
index cafc4085..c4be9c1d 100644
--- a/.speakeasy/workflow.yaml
+++ b/.speakeasy/workflow.yaml
@@ -4,6 +4,7 @@ sources:
     livepeer-studio-api:
         inputs:
             - location: https://raw.githubusercontent.com/livepeer/studio/master/packages/api/src/schema/api-schema.yaml
+            - location: https://raw.githubusercontent.com/livepeer/studio/master/packages/api/src/schema/ai-api-schema.yaml
         overlays:
             - location: https://raw.githubusercontent.com/livepeer/livepeer-js/main/codeSamples.yaml
             - location: https://raw.githubusercontent.com/livepeer/livepeer-go/main/codeSamples.yaml