@gradio/client 2.1.0 → 2.2.1

package/src/test/upload_files.test.ts

@@ -3,18 +3,21 @@ import { describe, it, expect, afterEach, beforeAll, afterAll } from "vitest";
 import { Client } from "..";
 import { initialise_server } from "./server";
 
-const server = initialise_server();
+let server: Awaited<ReturnType<typeof initialise_server>>;
 
-beforeAll(() => server.listen());
+beforeAll(async () => {
+	server = await initialise_server();
+	await server.start({ quiet: true });
+});
 afterEach(() => server.resetHandlers());
-afterAll(() => server.close());
+afterAll(() => server.stop());
 
 describe("upload_files", () => {
 	it("should upload files successfully", async () => {
 		const root_url = "https://hmb-hello-world.hf.space";
 
 		const client = await Client.connect("hmb/hello_world", {
-			hf_token: "hf_token"
+			token: "hf_token"
 		});
 
 		const files = [new Blob([], { type: "image/jpeg" })];

package/src/test/view_api.test.ts

@@ -8,11 +8,14 @@ const app_reference = "hmb/hello_world";
 const secret_app_reference = "hmb/secret_world";
 const secret_direct_app_reference = "https://hmb-secret-world.hf.space";
 
-const server = initialise_server();
+let server: Awaited<ReturnType<typeof initialise_server>>;
 
-beforeAll(() => server.listen());
+beforeAll(async () => {
+	server = await initialise_server();
+	await server.start({ quiet: true });
+});
 afterEach(() => server.resetHandlers());
-afterAll(() => server.close());
+afterAll(() => server.stop());
 
 describe("view_api", () => {
 	test("viewing the api of a running, public app", async () => {

package/src/types.ts

@@ -402,6 +402,9 @@ export interface Status {
 	changed_state_ids?: number[];
 	time_limit?: number;
 	session_not_found?: boolean;
+	used_cache?: "full" | "partial" | null;
+	cache_duration?: number;
+	avg_time?: number;
 }
 
 export interface StatusMessage extends Status {

package/src/utils/submit.ts

@@ -12,6 +12,7 @@ import type {
 } from "../types";
 
 import { skip_queue, post_message, handle_payload } from "../helpers/data";
+import { get_zerogpu_origin } from "../helpers/zerogpu";
 import {
 	handle_message,
 	map_data_to_params,
@@ -412,15 +413,13 @@ export function submit(
 					hostname = window?.location?.hostname;
 				}
 
-				let hfhubdev = "dev.spaces.huggingface.tech";
-				const origin = hostname.includes(".dev.")
-					? `https://moon-${hostname.split(".")[1]}.${hfhubdev}`
-					: `https://huggingface.co`;
+				const origin = get_zerogpu_origin(hostname);
 
 				const is_zerogpu_iframe =
 					typeof window !== "undefined" &&
 					typeof document !== "undefined" &&
 					window.parent != window &&
+					!!origin &&
 					window.supports_zerogpu_headers;
 				const zerogpu_auth_promise = is_zerogpu_iframe
 					? post_message<Map<string, string>>("zerogpu-headers", origin)
@@ -667,6 +666,9 @@ export function submit(
 			if (values.length > 0) {
 				return Promise.resolve(values.shift() as (typeof values)[0]);
 			}
+			if (done) {
+				return Promise.resolve({ value: undefined, done: true });
+			}
 			return new Promise((resolve) => resolvers.push(resolve));
 		}
 

package/vite.config.ts

@@ -0,0 +1,104 @@
+import { defineConfig, createLogger } from "vite";
+import { svelte } from "@sveltejs/vite-plugin-svelte";
+import { playwright } from "@vitest/browser-playwright";
+
+const TEST_MODE = process.env.TEST_MODE || "browser";
+
+const logger = createLogger();
+const originalWarning = logger.warn;
+
+logger.warn = (log, options) => {
+	if (log.includes("was created with unknown prop")) return false;
+	if (log.includes("https://svelte.dev")) return false;
+	if (log.includes("[vite-plugin-svelte]")) return false;
+	if (log && log.includes("[MSW]")) return false;
+	if (log && log.includes("node:")) return false;
+	originalWarning(log, options);
+};
+
+const originalError = logger.error;
+
+logger.error = (log, options) => {
+	if (log.includes("was created with unknown prop")) return false;
+	if (log.includes("https://svelte.dev")) return false;
+	if (log.includes("[vite-plugin-svelte]")) return false;
+	if (log && log.includes("[MSW]")) return false;
+	if (log && log.includes("(node:")) return false;
+	originalError(log, options);
+};
+
+const original_info = logger.info;
+logger.info = (log, options) => {
+	if (log.includes("was created with unknown prop")) return false;
+	if (log.includes("https://svelte.dev")) return false;
+	if (log.includes("[vite-plugin-svelte]")) return false;
+	if (log && log.includes("[MSW]")) return false;
+	if (log && log.includes("node:")) return false;
+	original_info(log, options);
+};
+export default defineConfig(({ mode }) => {
+	const production = mode === "production";
+	const isBrowserBuild = process.env.BROWSER_BUILD === "true";
+
+	if (mode === "preview") {
+		return {
+			entry: "index.html"
+		};
+	}
+
+	return {
+		customLogger: logger,
+		build: {
+			emptyOutDir: false,
+			lib: {
+				entry: "src/index.ts",
+				formats: ["es"],
+				fileName: isBrowserBuild ? `browser` : `index`
+			},
+			rollupOptions: {
+				input: "src/index.ts",
+				output: {
+					dir: "dist"
+				}
+			}
+		},
+		plugins: [svelte()],
+		define: {
+			BROWSER_BUILD: JSON.stringify(isBrowserBuild)
+		},
+		mode: process.env.MODE || "development",
+		test: {
+			include: ["./src/test/*.test.*"],
+			onConsoleLog(log) {
+				if (log.includes("[MSW]")) return false;
+				if (log.includes("node:")) return false;
+				if (log.includes("data: '\"404\"'")) return false;
+				if (log.includes("Too many arguments")) return false;
+			},
+			...(TEST_MODE === "node"
+				? { environment: "node" }
+				: {
+						browser: {
+							enabled: true,
+							provider: playwright(),
+							instances: [
+								{
+									browser: "chromium"
+								}
+							]
+						}
+					})
+		},
+		ssr: {
+			target: "node",
+			format: "esm",
+			noExternal: [
+				"ws",
+				"semiver",
+				"bufferutil",
+				"@gradio/upload",
+				"fetch-event-stream"
+			]
+		}
+	};
+});

package/vite.config.js

@@ -1,52 +0,0 @@
-import { defineConfig } from "vite";
-import { svelte } from "@sveltejs/vite-plugin-svelte";
-
-const TEST_MODE = process.env.TEST_MODE || "happy-dom";
-
-export default defineConfig(({ mode }) => {
-	const production = mode === "production";
-	const isBrowserBuild = process.env.BROWSER_BUILD === "true";
-
-	if (mode === "preview") {
-		return {
-			entry: "index.html"
-		};
-	}
-
-	return {
-		build: {
-			emptyOutDir: false,
-			lib: {
-				entry: "src/index.ts",
-				formats: ["es"],
-				fileName: isBrowserBuild ? `browser` : `index`
-			},
-			rollupOptions: {
-				input: "src/index.ts",
-				output: {
-					dir: "dist"
-				}
-			}
-		},
-		plugins: [svelte()],
-		define: {
-			BROWSER_BUILD: JSON.stringify(isBrowserBuild)
-		},
-		mode: process.env.MODE || "development",
-		test: {
-			include: ["./src/test/*.test.*"],
-			environment: TEST_MODE
-		},
-		ssr: {
-			target: "node",
-			format: "esm",
-			noExternal: [
-				"ws",
-				"semiver",
-				"bufferutil",
-				"@gradio/upload",
-				"fetch-event-stream"
-			]
-		}
-	};
-});