mirror/gradio
2
0
Fork 0
mirror of https://github.com/gradio-app/gradio.git synced 2025-04-12 12:40:29 +08:00
Code Issues Packages Projects Releases Wiki Activity

Ensure all status are reported internally when calling predict (#8485)

Browse Source 
* fix param name

* format

* fix

* add changeset

---------

Co-authored-by: gradio-pr-bot <gradio-pr-bot@users.noreply.github.com>
This commit is contained in:
pngwn 2024-06-06 15:57:00 +01:00 committed by GitHub
parent c79639448c
commit f8ebacecce
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
4 changed files with 12 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
.changeset/quiet-colts-burn.md Normal file
View File

@ -0,0 +1,6 @@
---
"@gradio/client": minor
"gradio": minor
---
feat:Ensure all status are reported internally when calling `predict`

3
client/js/src/client.ts
View File

@ -102,7 +102,8 @@ export class Client {
endpoint: string | number,
data: unknown[] | Record<string, unknown>,
event_data?: unknown,
trigger_id?: number | null
trigger_id?: number | null,
all_events?: boolean
) => SubmitIterable<GradioEvent>;
predict: (
endpoint: string | number,

2
client/js/src/utils/predict.ts
View File

@ -30,7 +30,7 @@ export async function predict(
}
return new Promise(async (resolve, reject) => {
const app = this.submit(endpoint, data);
const app = this.submit(endpoint, data, null, null, true);
let result: unknown;
for await (const message of app) {

5
client/js/src/utils/submit.ts
View File

@ -28,7 +28,8 @@ export function submit(
endpoint: string | number,
data: unknown[] | Record<string, unknown>,
event_data?: unknown,
trigger_id?: number | null
trigger_id?: number | null,
all_events?: boolean
): SubmitIterable<GradioEvent> {
try {
const { hf_token } = this.options;
@ -87,7 +88,7 @@ export function submit(
// event subscription methods
function fire_event(event: GradioEvent): void {
if (events_to_publish[event.type]) {
if (all_events || events_to_publish[event.type]) {
push_event(event);
}
}