mirror of
https://github.com/invoke-ai/InvokeAI.git
synced 2026-01-14 14:07:55 -05:00
add credits to queue item status changed (#7993)
* display credit column in queue list if shouldShowCredits is true * add credits when queue item status changes * chore(ui): typegen --------- Co-authored-by: Mary Hipp <maryhipp@Marys-MacBook-Air.local> Co-authored-by: psychedelicious <4822129+psychedelicious@users.noreply.github.com>
This commit is contained in:
@@ -241,6 +241,7 @@ class QueueItemStatusChangedEvent(QueueItemEventBase):
|
||||
batch_status: BatchStatus = Field(description="The status of the batch")
|
||||
queue_status: SessionQueueStatus = Field(description="The status of the queue")
|
||||
session_id: str = Field(description="The ID of the session (aka graph execution state)")
|
||||
credits: Optional[float] = Field(default=None, description="The total credits used for this queue item")
|
||||
|
||||
@classmethod
|
||||
def build(
|
||||
@@ -263,6 +264,7 @@ class QueueItemStatusChangedEvent(QueueItemEventBase):
|
||||
completed_at=str(queue_item.completed_at) if queue_item.completed_at else None,
|
||||
batch_status=batch_status,
|
||||
queue_status=queue_status,
|
||||
credits=queue_item.credits,
|
||||
)
|
||||
|
||||
|
||||
|
||||
@@ -257,7 +257,7 @@ class SessionQueueItemWithoutGraph(BaseModel):
|
||||
api_output_fields: Optional[list[FieldIdentifier]] = Field(
|
||||
default=None, description="The nodes that were used as output from the API"
|
||||
)
|
||||
credits: Optional[int] = Field(default=None, description="The total credits used for this queue item")
|
||||
credits: Optional[float] = Field(default=None, description="The total credits used for this queue item")
|
||||
|
||||
@classmethod
|
||||
def queue_item_dto_from_dict(cls, queue_item_dict: dict) -> "SessionQueueItemDTO":
|
||||
|
||||
@@ -17209,6 +17209,12 @@ export type components = {
|
||||
* @description The ID of the session (aka graph execution state)
|
||||
*/
|
||||
session_id: string;
|
||||
/**
|
||||
* Credits
|
||||
* @description The total credits used for this queue item
|
||||
* @default null
|
||||
*/
|
||||
credits: number | null;
|
||||
};
|
||||
/**
|
||||
* QueueItemsRetriedEvent
|
||||
|
||||
@@ -337,6 +337,7 @@ export const setEventListeners = ({ socket, store, setIsConnected }: SetEventLis
|
||||
error_message,
|
||||
error_traceback,
|
||||
destination,
|
||||
credits,
|
||||
} = data;
|
||||
|
||||
log.debug({ data }, `Queue item ${item_id} status updated: ${status}`);
|
||||
@@ -354,6 +355,7 @@ export const setEventListeners = ({ socket, store, setIsConnected }: SetEventLis
|
||||
error_type,
|
||||
error_message,
|
||||
error_traceback,
|
||||
credits,
|
||||
},
|
||||
});
|
||||
})
|
||||
|
||||
Reference in New Issue
Block a user