feat: Support grouped generation responses and improve generation ID extraction for polling.
This commit is contained in:
@@ -269,8 +269,15 @@ const handleGenerate = async () => {
|
|||||||
|
|
||||||
const response = await aiService.runGeneration(payload)
|
const response = await aiService.runGeneration(payload)
|
||||||
|
|
||||||
// Response can be a single generation or an array of grouped generations
|
// Response can be a single generation, an array, or a Group response with 'generations'
|
||||||
const generations = Array.isArray(response) ? response : [response]
|
let generations = []
|
||||||
|
if (response && response.generations) {
|
||||||
|
generations = response.generations
|
||||||
|
} else if (Array.isArray(response)) {
|
||||||
|
generations = response
|
||||||
|
} else {
|
||||||
|
generations = [response]
|
||||||
|
}
|
||||||
|
|
||||||
for (const gen of generations) {
|
for (const gen of generations) {
|
||||||
if (gen && gen.id) {
|
if (gen && gen.id) {
|
||||||
|
|||||||
@@ -211,8 +211,15 @@ const handleGenerate = async () => {
|
|||||||
|
|
||||||
const response = await aiService.runGeneration(payload)
|
const response = await aiService.runGeneration(payload)
|
||||||
|
|
||||||
if (response && response.id) {
|
let genId = null
|
||||||
pollStatus(response.id)
|
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 {
|
} else {
|
||||||
// Fallback immediate response
|
// Fallback immediate response
|
||||||
generatedResult.value = response
|
generatedResult.value = response
|
||||||
|
|||||||
Reference in New Issue
Block a user