feat(api): return list of enqueued item ids when enqueuing

invokeai/app/services/session_queue/session_queue_common.py

@@ -332,6 +332,7 @@ class EnqueueBatchResult(BaseModel):
     requested: int = Field(description="The total number of queue items requested to be enqueued")
     batch: Batch = Field(description="The batch that was enqueued")
     priority: int = Field(description="The priority of the enqueued batch")
+    item_ids: list[int] = Field(description="The IDs of the queue items that were enqueued")
 
 
 class RetryItemsResult(BaseModel):

invokeai/app/services/session_queue/session_queue_sqlite.py

@@ -133,6 +133,18 @@ class SqliteSessionQueue(SessionQueueBase):
                     """,
                     values_to_insert,
                 )
+            with self._conn:
+                cursor = self._conn.cursor()
+                cursor.execute(
+                    """--sql
+                    SELECT item_id
+                    FROM session_queue
+                    WHERE batch_id = ?
+                    ORDER BY item_id DESC;
+                    """,
+                    (batch.batch_id,)  # batch_id is the 4th element in the tuple
+                )
+                item_ids = [row[0] for row in cursor.fetchall()]
         except Exception:
             raise
         enqueue_result = EnqueueBatchResult(
@@ -141,6 +153,7 @@ class SqliteSessionQueue(SessionQueueBase):
             enqueued=enqueued_count,
             batch=batch,
             priority=priority,
+            item_ids=item_ids,
         )
         self.__invoker.services.events.emit_batch_enqueued(enqueue_result)
         return enqueue_result