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)
|
||||
|
||||
// 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) {
|
||||
|
||||
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user