import Foundation

/// Convenience builders mirroring the sibling clients, one per job type.
public extension JobSpec {
    static func domainAvailabilityBulk(domains: [String]) -> JobSpec {
        .domainAvailabilityBulk(DomainAvailabilityBulkJob(domains: domains))
    }

    static func embedding(input: String, model: String? = nil) -> JobSpec {
        .embedding(EmbeddingJob(input: input, model: model))
    }

    static func llmChat(messages: [ChatMessage], model: String? = nil, maxTokens: Int? = nil, temperature: Double? = nil) -> JobSpec {
        .llmChat(LlmChatJob(messages: messages, model: model, maxTokens: maxTokens, temperature: temperature))
    }

    static func llmRawCompletion(prompt: String, model: String? = nil, maxTokens: Int? = nil, temperature: Double? = nil) -> JobSpec {
        .llmRawCompletion(LlmRawCompletionJob(prompt: prompt, model: model, maxTokens: maxTokens, temperature: temperature))
    }

    static func nlpAnalyze(input: String, language: NlpAnalyzeJobLanguage? = nil) -> JobSpec {
        .nlpAnalyze(NlpAnalyzeJob(input: input, language: language))
    }

    static func nlpDetectLanguage(input: String) -> JobSpec {
        .nlpDetectLanguage(NlpDetectLanguageJob(input: input))
    }
}

public extension ChatMessage {
    static func system(_ content: String) -> ChatMessage { ChatMessage(role: .system, content: content) }
    static func user(_ content: String) -> ChatMessage { ChatMessage(role: .user, content: content) }
    static func assistant(_ content: String) -> ChatMessage { ChatMessage(role: .assistant, content: content) }
}