feat: declare explicit media provider capabilities

extensions/comfy/music-generation-provider.test.ts

@@ -12,7 +12,7 @@ describe("comfy music-generation provider", () => {
 
     expect(provider.defaultModel).toBe("workflow");
     expect(provider.models).toEqual(["workflow"]);
-    expect(provider.capabilities.maxInputImages).toBe(1);
+    expect(provider.capabilities.edit?.maxInputImages).toBe(1);
   });
 
   it("runs a music workflow and returns audio outputs", async () => {

extensions/comfy/music-generation-provider.ts

@@ -50,7 +50,11 @@ export function buildComfyMusicGenerationProvider(): MusicGenerationProvider {
         capability: "music",
       }),
     capabilities: {
-      maxInputImages: COMFY_MAX_INPUT_IMAGES,
+      generate: {},
+      edit: {
+        enabled: true,
+        maxInputImages: COMFY_MAX_INPUT_IMAGES,
+      },
     },
     async generateMusic(req) {
       if ((req.inputImages?.length ?? 0) > COMFY_MAX_INPUT_IMAGES) {

extensions/comfy/video-generation-provider.ts

@@ -39,14 +39,27 @@ export function buildComfyVideoGenerationProvider(): VideoGenerationProvider {
         capability: "video",
       }),
     capabilities: {
-      maxVideos: 1,
-      maxInputImages: 1,
-      maxInputVideos: 0,
-      supportsSize: false,
-      supportsAspectRatio: false,
-      supportsResolution: false,
-      supportsAudio: false,
-      supportsWatermark: false,
+      generate: {
+        maxVideos: 1,
+        supportsSize: false,
+        supportsAspectRatio: false,
+        supportsResolution: false,
+        supportsAudio: false,
+        supportsWatermark: false,
+      },
+      imageToVideo: {
+        enabled: true,
+        maxVideos: 1,
+        maxInputImages: 1,
+        supportsSize: false,
+        supportsAspectRatio: false,
+        supportsResolution: false,
+        supportsAudio: false,
+        supportsWatermark: false,
+      },
+      videoToVideo: {
+        enabled: false,
+      },
     },
     async generateVideo(req) {
       if ((req.inputImages?.length ?? 0) > 1) {