Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(ui): socket event types

Browse Source
This commit is contained in:
psychedelicious 2023-04-30 13:58:19 +10:00
parent 5b558af2b3
commit deae5fbaec
1 changed files with 11 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
invokeai/frontend/web/src/services/events/types.ts
View File

@ -1,4 +1,4 @@
import { Graph, GraphExecutionState } from '../api'; import { Graph, GraphExecutionState, InvokeAIMetadata } from '../api';
/** /**
* A progress image, we get one for each step in the generation * A progress image, we get one for each step in the generation
@ -17,6 +17,12 @@ export type AnyInvocation = NonNullable<Graph['nodes']>[string];
export type AnyResult = GraphExecutionState['results'][string]; export type AnyResult = GraphExecutionState['results'][string];
export type BaseNode = {
id: string;
type: string;
[key: string]: NonNullable<InvokeAIMetadata['node']>[string];
};
/** /**
* A `generator_progress` socket.io event. * A `generator_progress` socket.io event.
* *
@ -24,7 +30,7 @@ export type AnyResult = GraphExecutionState['results'][string];
*/ */
export type GeneratorProgressEvent = { export type GeneratorProgressEvent = {
graph_execution_state_id: string; graph_execution_state_id: string;
node: AnyInvocation; node: BaseNode;
source_node_id: string; source_node_id: string;
progress_image?: ProgressImage; progress_image?: ProgressImage;
step: number; step: number;
@ -40,7 +46,7 @@ export type GeneratorProgressEvent = {
*/ */
export type InvocationCompleteEvent = { export type InvocationCompleteEvent = {
graph_execution_state_id: string; graph_execution_state_id: string;
node: AnyInvocation; node: BaseNode;
source_node_id: string; source_node_id: string;
result: AnyResult; result: AnyResult;
}; };
@ -52,7 +58,7 @@ export type InvocationCompleteEvent = {
*/ */
export type InvocationErrorEvent = { export type InvocationErrorEvent = {
graph_execution_state_id: string; graph_execution_state_id: string;
node: AnyInvocation; node: BaseNode;
source_node_id: string; source_node_id: string;
error: string; error: string;
}; };
@ -64,7 +70,7 @@ export type InvocationErrorEvent = {
*/ */
export type InvocationStartedEvent = { export type InvocationStartedEvent = {
graph_execution_state_id: string; graph_execution_state_id: string;
node: AnyInvocation; node: BaseNode;
source_node_id: string; source_node_id: string;
}; };