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:
Mary Hipp Rogers
2025-05-06 08:54:44 -04:00
committed by GitHub
parent 862e2a3e49
commit ba0bac2aa5
4 changed files with 11 additions and 1 deletions

View File

@@ -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,
)

View File

@@ -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":

View File

@@ -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

View File

@@ -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,
},
});
})