Skip to content

Commit f8ebace

Browse files
pngwngradio-pr-bot
andauthored
Ensure all status are reported internally when calling predict (#8485)
* fix param name * format * fix * add changeset --------- Co-authored-by: gradio-pr-bot <[email protected]>
1 parent c796394 commit f8ebace

File tree

4 files changed

+12
-4
lines changed

4 files changed

+12
-4
lines changed

.changeset/quiet-colts-burn.md

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
---
2+
"@gradio/client": minor
3+
"gradio": minor
4+
---
5+
6+
feat:Ensure all status are reported internally when calling `predict`

client/js/src/client.ts

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -102,7 +102,8 @@ export class Client {
102102
endpoint: string | number,
103103
data: unknown[] | Record<string, unknown>,
104104
event_data?: unknown,
105-
trigger_id?: number | null
105+
trigger_id?: number | null,
106+
all_events?: boolean
106107
) => SubmitIterable<GradioEvent>;
107108
predict: (
108109
endpoint: string | number,

client/js/src/utils/predict.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ export async function predict(
3030
}
3131

3232
return new Promise(async (resolve, reject) => {
33-
const app = this.submit(endpoint, data);
33+
const app = this.submit(endpoint, data, null, null, true);
3434
let result: unknown;
3535

3636
for await (const message of app) {

client/js/src/utils/submit.ts

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,8 @@ export function submit(
2828
endpoint: string | number,
2929
data: unknown[] | Record<string, unknown>,
3030
event_data?: unknown,
31-
trigger_id?: number | null
31+
trigger_id?: number | null,
32+
all_events?: boolean
3233
): SubmitIterable<GradioEvent> {
3334
try {
3435
const { hf_token } = this.options;
@@ -87,7 +88,7 @@ export function submit(
8788

8889
// event subscription methods
8990
function fire_event(event: GradioEvent): void {
90-
if (events_to_publish[event.type]) {
91+
if (all_events || events_to_publish[event.type]) {
9192
push_event(event);
9293
}
9394
}

0 commit comments

Comments
 (0)