import PrismIntelligence

// Use the default system model
let config = PrismAppleIntelligenceConfiguration(
    model: .system(useCase: .general),
    instructions: "You are a helpful cooking assistant."
)

// Built-in system model for general use
let general = PrismAppleIntelligenceModelReference.system(useCase: .general)

// System model optimized for content tagging
let tagger = PrismAppleIntelligenceModelReference.system(useCase: .contentTagging)

// A named adapter registered with the system
let customAdapter = PrismAppleIntelligenceModelReference.adapterName("my-fine-tuned-model")

// An adapter loaded from a local file
let fileAdapter = PrismAppleIntelligenceModelReference.adapterFile(
    URL(fileURLWithPath: "/path/to/adapter.mlmodelc")
)

import PrismIntelligence

let provider = PrismAppleIntelligenceProvider(
    configuration: .init(
        model: .system(useCase: .general),
        instructions: "You are a concise recipe assistant."
    )
)

let request = PrismLanguageIntelligenceRequest(
    prompt: "Give me a quick pasta recipe"
)

let response = try await provider.generate(request: request)
print(response.text)

let request = PrismLanguageIntelligenceRequest(
    prompt: "Explain Swift concurrency in simple terms"
)

for try await chunk in provider.stream(request: request) {
    print(chunk.text, terminator: "")
}

let provider = PrismAppleIntelligenceProvider(
    configuration: .init()
)

let status = await provider.status()

if status.isAvailable {
    print("Apple Intelligence ready")
    print("Streaming: \(status.supportsStreaming)")
    print("Adapters: \(status.supportsModelAdapters)")
} else {
    print("Not available: \(status.reason ?? "unknown")")
}

import PrismIntelligence

let remote = PrismRemoteIntelligenceProvider(
    endpoint: URL(string: "https://api.example.com/v1/chat")!,
    apiKey: "sk-...",
    model: "gpt-4"
)

let request = PrismLanguageIntelligenceRequest(
    prompt: "Summarize this article"
)

let response = try await remote.generate(request: request)

import PrismIntelligence

func generateText(prompt: String) async throws -> String {
    let appleProvider = PrismAppleIntelligenceProvider(
        configuration: .init()
    )

    let status = await appleProvider.status()
    let request = PrismLanguageIntelligenceRequest(prompt: prompt)

    if status.isAvailable {
        let response = try await appleProvider.generate(request: request)
        return response.text
    }

    // Fall back to remote
    let remote = PrismRemoteIntelligenceProvider(
        endpoint: URL(string: "https://api.example.com/v1/chat")!,
        apiKey: "sk-..."
    )
    let response = try await remote.generate(request: request)
    return response.text
}

import PrismIntelligence

let provider = PrismAppleIntelligenceProvider(
    configuration: .init(
        model: .system(useCase: .contentTagging),
        instructions: "Tag content with relevant categories. Return tags as a comma-separated list."
    )
)

let articles = [
    "Apple announces new MacBook Pro with M5 chip",
    "Champions League final ends in dramatic penalty shootout",
    "New study links Mediterranean diet to longevity"
]

for article in articles {
    let request = PrismLanguageIntelligenceRequest(prompt: article)
    let response = try await provider.generate(request: request)
    print("\(article)")
    print("  Tags: \(response.text)\n")
}