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) {
|
||||
@@ -892,7 +899,7 @@ const confirmAddToAlbum = async () => {
|
||||
</div>
|
||||
<i class="pi pi-spin pi-spinner text-violet-500 text-xl mb-2 relative z-10"></i>
|
||||
<span class="text-[10px] text-violet-300/70 relative z-10 capitalize">{{ item.status
|
||||
}}...</span>
|
||||
}}...</span>
|
||||
<span v-if="item.progress"
|
||||
class="text-[9px] text-violet-400/60 font-mono mt-1 relative z-10">{{
|
||||
item.progress }}%</span>
|
||||
@@ -956,7 +963,7 @@ const confirmAddToAlbum = async () => {
|
||||
@click.stop="reuseAsset(item)" />
|
||||
</div>
|
||||
<p class="text-[10px] text-white/70 line-clamp-1 leading-tight">{{ item.prompt
|
||||
}}</p>
|
||||
}}</p>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
|
||||
@@ -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