@gradio/client 2.7.0 → 2.7.1

package/dist/utils/submit.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"submit.d.ts","sourceRoot":"","sources":["../../src/utils/submit.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAGX,WAAW,EAMX,cAAc,EACd,MAAM,UAAU,CAAC;AAyBlB,OAAO,EAAE,MAAM,EAAE,MAAM,WAAW,CAAC;AAQnC,wBAAgB,MAAM,CACrB,IAAI,EAAE,MAAM,EACZ,QAAQ,EAAE,MAAM,GAAG,MAAM,EACzB,IAAI,GAAE,OAAO,EAAE,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAM,EAC9C,UAAU,CAAC,EAAE,OAAO,EACpB,UAAU,CAAC,EAAE,MAAM,GAAG,IAAI,EAC1B,UAAU,CAAC,EAAE,OAAO,EACpB,kBAAkB,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GACzC,cAAc,CAAC,WAAW,CAAC,CA0vB7B"}
+{"version":3,"file":"submit.d.ts","sourceRoot":"","sources":["../../src/utils/submit.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAGX,WAAW,EAMX,cAAc,EACd,MAAM,UAAU,CAAC;AAyBlB,OAAO,EAAE,MAAM,EAAE,MAAM,WAAW,CAAC;AAQnC,wBAAgB,MAAM,CACrB,IAAI,EAAE,MAAM,EACZ,QAAQ,EAAE,MAAM,GAAG,MAAM,EACzB,IAAI,GAAE,OAAO,EAAE,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAM,EAC9C,UAAU,CAAC,EAAE,OAAO,EACpB,UAAU,CAAC,EAAE,MAAM,GAAG,IAAI,EAC1B,UAAU,CAAC,EAAE,OAAO,EACpB,kBAAkB,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GACzC,cAAc,CAAC,WAAW,CAAC,CAiwB7B"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@gradio/client",
-  "version": "2.7.0",
+  "version": "2.7.1",
   "description": "Gradio API client",
   "type": "module",
   "main": "dist/index.cjs",

package/src/test/submit.test.ts

@@ -1,4 +1,12 @@
-import { describe, beforeAll, afterEach, afterAll, test, expect } from "vitest";
+import {
+	describe,
+	beforeAll,
+	afterEach,
+	afterAll,
+	test,
+	expect,
+	vi
+} from "vitest";
 import { HttpResponse, http } from "msw";
 
 import { Client } from "../client";
@@ -119,6 +127,37 @@ describe("submit iterator", () => {
 		expect(result).toEqual({ value: undefined, done: true });
 	});
 
+	test("send_chunk() and close_stream() called before queue/join responds reach the event in order", async () => {
+		const app = await Client.connect("hmb/hello_world");
+		app.stream_status.open = true;
+
+		let respond_to_join: () => void = () => {};
+		const join_released = new Promise<void>((r) => (respond_to_join = r));
+		const stream_paths: string[] = [];
+		server.use(
+			http.post(`${direct_space_url}/queue/join`, async () => {
+				await join_released;
+				return HttpResponse.json({ event_id: "slow-event" });
+			}),
+			http.post(/\/stream\//, ({ request }) => {
+				stream_paths.push(new URL(request.url).pathname);
+				return HttpResponse.json({ msg: "success" });
+			})
+		);
+
+		const iterator = app.submit("/predict", ["hi"]);
+		iterator.send_chunk({ data: ["chunk"] });
+		iterator.close_stream();
+		respond_to_join();
+
+		await vi.waitFor(() =>
+			expect(stream_paths).toEqual([
+				"/stream/slow-event",
+				"/stream/slow-event/close"
+			])
+		);
+	});
+
 	test("for-await loop terminates when data and complete arrive in the same SSE callback", async () => {
 		const app = await Client.connect("hmb/hello_world", {
 			events: ["data", "status"]

package/src/utils/submit.ts

@@ -773,6 +773,20 @@ export function submit(
 			return new Promise((resolve) => resolvers.push(resolve));
 		}
 
+		// The event id is only known once queue/join has responded.
+		const post_to_event = (suffix: string, body: unknown): void => {
+			job.then(
+				() => {
+					if (!event_id_final) return;
+					this.post_data(
+						`${config.root}${api_prefix}/stream/${event_id_final}${suffix}`,
+						body
+					);
+				},
+				() => {}
+			);
+		};
+
 		const iterator: SubmitIterable<GradioEvent> = {
 			[Symbol.asyncIterator]: () => iterator,
 			next,
@@ -786,18 +800,11 @@ export function submit(
 			},
 			cancel,
 			send_chunk: (payload: Record<string, unknown>) => {
-				this.post_data(`${config.root}${api_prefix}/stream/${event_id_final}`, {
-					...payload,
-					session_hash: this.session_hash
-				});
+				post_to_event("", { ...payload, session_hash: this.session_hash });
 			},
 			close_stream: () => {
-				this.post_data(
-					`${config.root}${api_prefix}/stream/${event_id_final}/close`,
-					{}
-				);
-
 				close();
+				post_to_event("/close", {});
 			},
 			event_id: () => event_id_final,
 			wait_for_id: async () => {