@retab/node 0.0.45 → 0.0.48

package/dist/api/consensus/client.js

@@ -5,7 +5,7 @@ export default class APIConsensus extends CompositionClient {
         super(client);
     }
     async reconcile(params) {
-        return await this._fetchJson(ZReconciliationResponse, {
+        return this._fetchJson(ZReconciliationResponse, {
             url: "/v1/consensus/reconcile",
             method: "POST",
             body: ZReconciliationRequest.parse(params),

package/dist/api/deployments/client.js

@@ -10,7 +10,7 @@ export default class APIDeployments extends CompositionClient {
             throw new Error("Either 'document' or 'documents' must be provided.");
         }
         let url = `/v1/deployments/extract/${project_id}/${iteration_id}`;
-        return await this._fetchJson(ZSchema, {
+        return this._fetchJson(ZSchema, {
             url,
             method: "POST",
             body: {

package/dist/api/documents/client.d.ts

@@ -1,8 +1,9 @@
 import { CompositionClient } from "@/client";
-import { DocumentExtractRequest, RetabParsedChatCompletion, ParseRequest, ParseResult, DocumentCreateMessageRequest, DocumentMessage, DocumentCreateInputRequest } from "@/types";
+import { DocumentExtractRequest, RetabParsedChatCompletion, ParseRequest, ParseResult, DocumentCreateMessageRequest, DocumentMessage, DocumentCreateInputRequest, RetabParsedChatCompletionChunk } from "@/types";
 export default class APIDocuments extends CompositionClient {
     constructor(client: CompositionClient);
     extract(params: DocumentExtractRequest): Promise<RetabParsedChatCompletion>;
+    extractStream(params: DocumentExtractRequest): Promise<AsyncGenerator<RetabParsedChatCompletionChunk>>;
     parse(params: ParseRequest): Promise<ParseResult>;
     createMessages(params: DocumentCreateMessageRequest): Promise<DocumentMessage>;
     createInputs(params: DocumentCreateInputRequest): Promise<DocumentMessage>;

package/dist/api/documents/client.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"client.d.ts","sourceRoot":"","sources":["../../../src/api/documents/client.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,MAAM,UAAU,CAAC;AAC7C,OAAO,EAA2B,sBAAsB,EAAE,yBAAyB,EAA8B,YAAY,EAAE,WAAW,EAA+B,4BAA4B,EAAE,eAAe,EAAmD,0BAA0B,EAA+B,MAAM,SAAS,CAAC;AAGlV,MAAM,CAAC,OAAO,OAAO,YAAa,SAAQ,iBAAiB;gBAC3C,MAAM,EAAE,iBAAiB;IAG/B,OAAO,CAAC,MAAM,EAAE,sBAAsB,GAAG,OAAO,CAAC,yBAAyB,CAAC;IAO3E,KAAK,CAAC,MAAM,EAAE,YAAY,GAAG,OAAO,CAAC,WAAW,CAAC;IAOjD,cAAc,CAAC,MAAM,EAAE,4BAA4B,GAAG,OAAO,CAAC,eAAe,CAAC;IAO9E,YAAY,CAAC,MAAM,EAAE,0BAA0B,GAAG,OAAO,CAAC,eAAe,CAAC;CAOnF"}
+{"version":3,"file":"client.d.ts","sourceRoot":"","sources":["../../../src/api/documents/client.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,MAAM,UAAU,CAAC;AAC7C,OAAO,EAA2B,sBAAsB,EAAE,yBAAyB,EAA8B,YAAY,EAAE,WAAW,EAA+B,4BAA4B,EAAE,eAAe,EAAmD,0BAA0B,EAA+B,8BAA8B,EAAmC,MAAM,SAAS,CAAC;AAGnZ,MAAM,CAAC,OAAO,OAAO,YAAa,SAAQ,iBAAiB;gBAC3C,MAAM,EAAE,iBAAiB;IAG/B,OAAO,CAAC,MAAM,EAAE,sBAAsB,GAAG,OAAO,CAAC,yBAAyB,CAAC;IAQ3E,aAAa,CAAC,MAAM,EAAE,sBAAsB,GAAG,OAAO,CAAC,cAAc,CAAC,8BAA8B,CAAC,CAAC;IAQtG,KAAK,CAAC,MAAM,EAAE,YAAY,GAAG,OAAO,CAAC,WAAW,CAAC;IAOjD,cAAc,CAAC,MAAM,EAAE,4BAA4B,GAAG,OAAO,CAAC,eAAe,CAAC;IAO9E,YAAY,CAAC,MAAM,EAAE,0BAA0B,GAAG,OAAO,CAAC,eAAe,CAAC;CAOnF"}

package/dist/api/documents/client.js

@@ -1,32 +1,41 @@
 import { CompositionClient } from "@/client";
-import { ZDocumentExtractRequest, ZRetabParsedChatCompletion, ZParseResult, ZParseRequest, ZDocumentMessage, ZDocumentCreateMessageRequest, ZDocumentCreateInputRequest } from "@/types";
+import { ZDocumentExtractRequest, ZRetabParsedChatCompletion, ZParseResult, ZParseRequest, ZDocumentMessage, ZDocumentCreateMessageRequest, ZDocumentCreateInputRequest, ZRetabParsedChatCompletionChunk } from "@/types";
 export default class APIDocuments extends CompositionClient {
     constructor(client) {
         super(client);
     }
     async extract(params) {
-        return await this._fetchJson(ZRetabParsedChatCompletion, {
+        let request = await ZDocumentExtractRequest.parseAsync(params);
+        return this._fetchJson(ZRetabParsedChatCompletion, {
             url: "/v1/documents/extract",
             method: "POST",
-            body: await ZDocumentExtractRequest.parseAsync(params),
+            body: request,
+        });
+    }
+    async extractStream(params) {
+        let request = await ZDocumentExtractRequest.parseAsync(params);
+        return this._fetchStream(ZRetabParsedChatCompletionChunk, {
+            url: "/v1/documents/extract",
+            method: "POST",
+            body: { ...request, stream: true },
         });
     }
     async parse(params) {
-        return await this._fetchJson(ZParseResult, {
+        return this._fetchJson(ZParseResult, {
             url: "/v1/documents/parse",
             method: "POST",
             body: await ZParseRequest.parseAsync(params),
         });
     }
     async createMessages(params) {
-        return await this._fetchJson(ZDocumentMessage, {
+        return this._fetchJson(ZDocumentMessage, {
             url: "/v1/documents/create_messages",
             method: "POST",
             body: await ZDocumentCreateMessageRequest.parseAsync(params),
         });
     }
     async createInputs(params) {
-        return await this._fetchJson(ZDocumentMessage, {
+        return this._fetchJson(ZDocumentMessage, {
             url: "/v1/documents/create_inputs",
             method: "POST",
             body: await ZDocumentCreateInputRequest.parseAsync(params),

package/dist/api/models/client.js

@@ -6,7 +6,7 @@ export default class APIModels extends CompositionClient {
         super(client);
     }
     async list(params) {
-        return await this._fetchJson(z.object({ data: z.array(ZModel) }), {
+        return this._fetchJson(z.object({ data: z.array(ZModel) }), {
             url: "/v1/models",
             method: "GET",
             params: params,

package/dist/api/schemas/client.js

@@ -5,7 +5,7 @@ export default class APISchemas extends CompositionClient {
         super(client);
     }
     async generate(params) {
-        return await this._fetchJson(ZSchema, {
+        return this._fetchJson(ZSchema, {
             url: "/v1/schemas/generate",
             method: "POST",
             body: await ZGenerateSchemaRequest.parseAsync(params),

package/dist/client.d.ts

@@ -11,6 +11,7 @@ type FetchParams = {
 export declare class AbstractClient {
     protected _fetch(_: FetchParams): Promise<Response>;
     protected _fetchJson<ZodSchema extends z.ZodType<any, any, any>>(bodyType: ZodSchema, params: FetchParams): Promise<z.output<ZodSchema>>;
+    protected _fetchStream<ZodSchema extends z.ZodType<any, any, any>>(schema: ZodSchema, params: FetchParams): Promise<AsyncGenerator<z.output<ZodSchema>>>;
 }
 export declare class CompositionClient extends AbstractClient {
     protected _client: AbstractClient;
@@ -22,7 +23,6 @@ export declare class APIError extends Error {
     info: string;
     constructor(status: number, info: string);
 }
-export declare function streamResponse<ZodSchema extends z.ZodType<any, any, any>>(response: Response, schema: ZodSchema): AsyncGenerator<z.output<ZodSchema>>;
 export declare const DateOrISO: z.ZodUnion<[z.ZodDate, z.ZodEffects<z.ZodEffects<z.ZodString, string, string>, Date, string>]>;
 type AuthTypes = {
     "bearer": string;

package/dist/client.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"client.d.ts","sourceRoot":"","sources":["../src/client.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,CAAC,MAAM,KAAK,CAAC;AAEzB,KAAK,WAAW,GAAG;IACjB,GAAG,EAAE,MAAM,CAAC;IACZ,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAC7B,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAC9B,QAAQ,CAAC,EAAE,kBAAkB,GAAG,qBAAqB,CAAC;IACtD,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAC3B,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC;CACjB,CAAC;AACF,qBAAa,cAAc;IACzB,SAAS,CAAC,MAAM,CAAC,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,QAAQ,CAAC;cAGnC,UAAU,CAAC,SAAS,SAAS,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,EAAE,QAAQ,EAAE,SAAS,EAAE,MAAM,EAAE,WAAW,GAAG,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;CAQ/I;AAED,qBAAa,iBAAkB,SAAQ,cAAc;IACnD,SAAS,CAAC,OAAO,EAAE,cAAc,CAAC;gBACtB,MAAM,EAAE,cAAc;IAIlC,SAAS,CAAC,MAAM,CAAC,MAAM,EAAE,WAAW,GAAG,OAAO,CAAC,QAAQ,CAAC;CAGzD;AAED,qBAAa,QAAS,SAAQ,KAAK;IACjC,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,EAAE,MAAM,CAAC;gBACD,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM;CAKzC;AAED,wBAAuB,cAAc,CAAC,SAAS,SAAS,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,EAAE,QAAQ,EAAE,QAAQ,EAAE,MAAM,EAAE,SAAS,GAAG,cAAc,CAAC,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAsC5J;AAED,eAAO,MAAM,SAAS,gGAKpB,CAAC;AAEH,KAAK,SAAS,GAAG;IAAE,QAAQ,EAAE,MAAM,CAAA;CAAE,GAAG;IAAE,WAAW,EAAE,MAAM,CAAA;CAAE,GAAG;IAAE,QAAQ,EAAE,MAAM,CAAA;CAAE,GAAG,EAAE,CAAC;AAC5F,MAAM,MAAM,aAAa,GAAG;IAC1B,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB,GAAG,SAAS,CAAC;AAEd,qBAAa,aAAc,SAAQ,cAAc;IAC/C,OAAO,EAAE,aAAa,CAAC;gBACX,OAAO,CAAC,EAAE,aAAa;IAK7B,MAAM,CAAC,MAAM,EAAE;QACnB,GAAG,EAAE,MAAM,CAAC;QACZ,MAAM,EAAE,MAAM,CAAC;QACf,MAAM,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;QAC7B,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;QAC9B,QAAQ,CAAC,EAAE,kBAAkB,GAAG,qBAAqB,CAAC;QACtD,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;KAC5B,GAAG,OAAO,CAAC,QAAQ,CAAC;CAmDtB"}
+{"version":3,"file":"client.d.ts","sourceRoot":"","sources":["../src/client.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,CAAC,MAAM,KAAK,CAAC;AAEzB,KAAK,WAAW,GAAG;IACjB,GAAG,EAAE,MAAM,CAAC;IACZ,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAC7B,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAC9B,QAAQ,CAAC,EAAE,kBAAkB,GAAG,qBAAqB,CAAC;IACtD,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAC3B,IAAI,CAAC,EAAE,MAAM,EAAE,CAAC;CACjB,CAAC;AAyCF,qBAAa,cAAc;IACzB,SAAS,CAAC,MAAM,CAAC,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,QAAQ,CAAC;cAGnC,UAAU,CAAC,SAAS,SAAS,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,EAAE,QAAQ,EAAE,SAAS,EAAE,MAAM,EAAE,WAAW,GAAG,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;cAQ9H,YAAY,CAAC,SAAS,SAAS,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,WAAW,GAAG,OAAO,CAAC,cAAc,CAAC,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC;CAQ/J;AAED,qBAAa,iBAAkB,SAAQ,cAAc;IACnD,SAAS,CAAC,OAAO,EAAE,cAAc,CAAC;gBACtB,MAAM,EAAE,cAAc;IAIlC,SAAS,CAAC,MAAM,CAAC,MAAM,EAAE,WAAW,GAAG,OAAO,CAAC,QAAQ,CAAC;CAGzD;AAED,qBAAa,QAAS,SAAQ,KAAK;IACjC,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,EAAE,MAAM,CAAC;gBACD,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM;CAKzC;AAED,eAAO,MAAM,SAAS,gGAKpB,CAAC;AAEH,KAAK,SAAS,GAAG;IAAE,QAAQ,EAAE,MAAM,CAAA;CAAE,GAAG;IAAE,WAAW,EAAE,MAAM,CAAA;CAAE,GAAG;IAAE,QAAQ,EAAE,MAAM,CAAA;CAAE,GAAG,EAAE,CAAC;AAC5F,MAAM,MAAM,aAAa,GAAG;IAC1B,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB,GAAG,SAAS,CAAC;AAEd,qBAAa,aAAc,SAAQ,cAAc;IAC/C,OAAO,EAAE,aAAa,CAAC;gBACX,OAAO,CAAC,EAAE,aAAa;IAK7B,MAAM,CAAC,MAAM,EAAE;QACnB,GAAG,EAAE,MAAM,CAAC;QACZ,MAAM,EAAE,MAAM,CAAC;QACf,MAAM,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;QAC7B,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;QAC9B,QAAQ,CAAC,EAAE,kBAAkB,GAAG,qBAAqB,CAAC;QACtD,IAAI,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;KAC5B,GAAG,OAAO,CAAC,QAAQ,CAAC;CAmDtB"}

package/dist/client.js

@@ -1,35 +1,5 @@
 import * as z from "zod";
-export class AbstractClient {
-    _fetch(_) {
-        throw new Error("Method not implemented");
-    }
-    async _fetchJson(bodyType, params) {
-        let response = await this._fetch(params);
-        if (!response.ok) {
-            throw new APIError(response.status, await response.text());
-        }
-        if (response.headers.get("Content-Type") !== "application/json")
-            throw new APIError(response.status, "Response is not JSON");
-        return bodyType.parse(await response.json());
-    }
-}
-export class CompositionClient extends AbstractClient {
-    constructor(client) {
-        super();
-        this._client = client;
-    }
-    _fetch(params) {
-        return this._client["_fetch"](params);
-    }
-}
-export class APIError extends Error {
-    constructor(status, info) {
-        super(`API Error ${status}: ${info}`);
-        this.status = status;
-        this.info = info;
-    }
-}
-export async function* streamResponse(response, schema) {
+async function* streamResponse(schema, response) {
     let body = "";
     let depth = 0;
     let inString = false;
@@ -61,7 +31,7 @@ export async function* streamResponse(response, schema) {
                 else if (char === "}") {
                     depth--;
                     if (depth === 0) {
-                        yield schema.parse(JSON.parse(body.slice(0, prevBodyLength + i + 1)));
+                        yield schema.parseAsync(JSON.parse(body.slice(0, prevBodyLength + i + 1)));
                         body = body.slice(prevBodyLength + i + 1);
                         prevBodyLength = -i - 1;
                     }
@@ -72,6 +42,45 @@ export async function* streamResponse(response, schema) {
             break;
     }
 }
+export class AbstractClient {
+    _fetch(_) {
+        throw new Error("Method not implemented");
+    }
+    async _fetchJson(bodyType, params) {
+        let response = await this._fetch(params);
+        if (!response.ok) {
+            throw new APIError(response.status, await response.text());
+        }
+        if (response.headers.get("Content-Type") !== "application/json")
+            throw new APIError(response.status, "Response is not JSON");
+        return bodyType.parseAsync(await response.json());
+    }
+    async _fetchStream(schema, params) {
+        let response = await this._fetch(params);
+        if (!response.ok) {
+            throw new APIError(response.status, await response.text());
+        }
+        if (response.headers.get("Content-Type") !== "application/stream+json")
+            throw new APIError(response.status, "Response is not stream JSON");
+        return streamResponse(schema, response);
+    }
+}
+export class CompositionClient extends AbstractClient {
+    constructor(client) {
+        super();
+        this._client = client;
+    }
+    _fetch(params) {
+        return this._client["_fetch"](params);
+    }
+}
+export class APIError extends Error {
+    constructor(status, info) {
+        super(`API Error ${status}: ${info}`);
+        this.status = status;
+        this.info = info;
+    }
+}
 export const DateOrISO = z.union([
     z.date(),
     z.string().refine(val => !isNaN(Date.parse(val)), {