@gradio/client 2.0.4 → 2.2.0

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,
@@ -237,12 +238,15 @@ export function submit(
 								time: new Date()
 							});
 						} else {
+							const is_connection_error =
+								output?.error === BROKEN_CONNECTION_MSG;
 							fire_event({
 								type: "status",
 								stage: "error",
 								endpoint: _endpoint,
 								fn_index,
 								message: output.error,
+								broken: is_connection_error,
 								queue: false,
 								time: new Date()
 							});
@@ -409,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)
@@ -464,11 +466,15 @@ export function submit(
 						});
 						close();
 					} else if (status !== 200) {
+						const is_connection_error =
+							response?.error === BROKEN_CONNECTION_MSG;
 						fire_event({
 							type: "status",
 							stage: "error",
-							broken: false,
-							message: response.detail,
+							broken: is_connection_error,
+							message: is_connection_error
+								? BROKEN_CONNECTION_MSG
+								: response.detail || response.error,
 							queue: true,
 							endpoint: _endpoint,
 							fn_index,

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"
-			]
-		}
-	};
-});