feat: add integration tests for skill-driven extraction functionality

docs/CONTRIBUTING.md

@@ -73,6 +73,7 @@ npm run test:integration
 ```bash
 # Run just the files integration tests
 npm run test:integration -- tests/integration/client/files.test.ts
+npm run test:integration -- --testPathPatterns="skills" -t "skill-driven extraction"
 
 # Run
 npm run test:e2e -- tests/e2e/client/chat-completions.test.ts

tests/integration/client/skills.test.ts

@@ -2,7 +2,7 @@ import { config } from "dotenv";
 config({ path: ".env.test" });
 
 import { VlmRun } from "../../../src/index";
-import { SkillInfo } from "../../../src/client/types";
+import { SkillInfo, PredictionResponse } from "../../../src/client/types";
 
 jest.setTimeout(60000);
 
@@ -379,4 +379,65 @@ describe("Integration: Skills", () => {
       expect(result.download_url.length).toBeGreaterThan(0);
     });
   });
+
+  describe("create() and document.generate() — skill-driven extraction", () => {
+    const testFilePath = "tests/integration/assets/google_invoice.pdf";
+
+    it("should create a skill and use it for batch document generation via file_id", async () => {
+      const skill = await client.skills.create({
+        prompt: INVOICE_SKILL_PROMPT,
+        jsonSchema: INVOICE_SCHEMA,
+        name: `invoice-doc-generate-${Date.now()}`,
+        description:
+          "Invoice skill for document generation integration test.",
+      });
+
+      expect(skill).toBeTruthy();
+      expect(skill.id).toBeDefined();
+      expect(skill.name).toBeDefined();
+
+      const uploadedFile = await client.files.upload({
+        filePath: testFilePath,
+        purpose: "vision",
+        checkDuplicate: true,
+      });
+
+      expect(uploadedFile).toHaveProperty("id");
+
+      const result: PredictionResponse = await client.document.generate({
+        fileId: uploadedFile.id,
+        model: "vlm-1",
+        domain: "document.invoice",
+        batch: true,
+        config: {
+          skills: [
+            {
+              skill_name: skill.name,
+              version: skill.version,
+            },
+          ],
+        } as any,
+        metadata: {
+          environment: "dev",
+          allowTraining: false,
+        },
+      });
+
+      expect(result).toHaveProperty("id");
+      expect(typeof result.id).toBe("string");
+      expect(result).toHaveProperty("created_at");
+      expect(typeof result.created_at).toBe("string");
+      expect(result).toHaveProperty("status");
+      expect(result.status).toBe("pending");
+
+      const completed: PredictionResponse = await client.predictions.wait(
+        result.id,
+      );
+
+      expect(completed.status).toBe("completed");
+      expect(completed).toHaveProperty("response");
+      expect(completed.response).toBeTruthy();
+      expect(completed).toHaveProperty("completed_at");
+    });
+  });
 });