dramaling-vocab-learning/backend/DramaLing.Api/Contracts/Services/AI/Generation/IGenerationStateManager.cs

using DramaLing.Api.Data;
using DramaLing.Api.Models.Entities;

namespace DramaLing.Api.Services.AI.Generation;

public interface IGenerationStateManager
{
    Task UpdateRequestStatusAsync(Guid requestId, string overallStatus, string geminiStatus, string replicateStatus);
    Task UpdateGeminiResultAsync(Guid requestId, string optimizedPrompt);
    Task CompleteRequestAsync(Guid requestId, Guid imageId, long totalProcessingTimeMs);
    Task MarkRequestAsFailedAsync(Guid requestId, string stage, string? errorMessage);
}