feat: Support grouped generation responses and improve generation ID extraction for polling.

This commit is contained in:
xds
2026-02-16 00:25:41 +03:00
parent 75e33cca9a
commit 2aa9156a20
2 changed files with 20 additions and 6 deletions

View File

@@ -269,8 +269,15 @@ const handleGenerate = async () => {
const response = await aiService.runGeneration(payload)
// Response can be a single generation or an array of grouped generations
const generations = Array.isArray(response) ? response : [response]
// Response can be a single generation, an array, or a Group response with 'generations'
let generations = []
if (response && response.generations) {
generations = response.generations
} else if (Array.isArray(response)) {
generations = response
} else {
generations = [response]
}
for (const gen of generations) {
if (gen && gen.id) {

View File

@@ -211,8 +211,15 @@ const handleGenerate = async () => {
const response = await aiService.runGeneration(payload)
if (response && response.id) {
pollStatus(response.id)
let genId = null
if (response && response.generations && response.generations.length > 0) {
genId = response.generations[0].id
} else if (response && response.id) {
genId = response.id
}
if (genId) {
pollStatus(genId)
} else {
// Fallback immediate response
generatedResult.value = response