dependencies: [
    .package(url: "https://github.com/mattt/AnyLanguageModel.git", branch: "main")
]

import AnyLanguageModel

import Foundation
import CoreML
import MLX

let models = [(any LanguageModel)] = [
    SystemLanguageModel(), // Apple Foundation Models
    CoreMLLanguageModel(url: "path/to/some.mlpackage"),
    MLXLanguageModel(modelId: "mlx-community/Qwen3-0.6B-4bit"),
    OllamaLanguageModel(model: "qwen3") // `ollama pull qwen3:0.6b`
    AnthropicLanguageModel(
        apiKey: ProcessInfo.processInfo.environment["ANTHROPIC_API_KEY"]!,
        model: "claude-sonnet-4-5-20250929"
    ),
    OpenAILanguageModel(
        apiKey: ProcessInfo.processInfo.environment["OPENAI_API_KEY"]!,
        model: "gpt-4o-mini"
    ),
]

struct WeatherTool: Tool {
    let name = "getWeather"
    let description = "Retrieve the latest weather information for a city"

    @Generable
    struct Arguments {
        @Guide(description: "The city to fetch the weather for")
        var city: String
    }

    func call(arguments: Arguments) async throws -> String {
        "The weather in \(arguments.city) is sunny and 72°F / 23°C"
    }
}

for model in models {
    let session = LanguageModelSession(model: model, tools: [WeatherTool()])
    let response = try await session.respond(to: "What's the weather in Cupertino?")
    print(response.text) // "It's sunny and 72°F in Cupertino"
}