Foundation Models Instruments Template

import SwiftUI
import FoundationModels

// MARK: - Tools

struct GenerateCraftIdeaTool: Tool {
    static let name = "generateCraftIdea"
    static let description = "Generates a craft project idea based on the crafter's interests."

    struct Arguments: Codable {
        let theme: String
    }

    func call(arguments: Arguments) async throws -> ToolOutput {
        // In production this would call into your data layer.
        return ToolOutput("Here is a craft idea based on theme: \(arguments.theme)")
    }
}

struct SwitchToTutorialModeTool: Tool {
    static let name = "switchToTutorialMode"
    static let description = "Switches the session to tutorial generation mode for the chosen craft."

    struct Arguments: Codable {
        let selectedCraft: String
    }

    func call(arguments: Arguments) async throws -> ToolOutput {
        return ToolOutput("Switching to tutorial mode for: \(arguments.selectedCraft)")
    }
}

// MARK: - Dynamic Instructions

struct BrainstormDynamicInstructions: DynamicInstructions {
    var instructions: Instructions {
        Instructions(
            """
            You are a creative craft brainstorming assistant.
            Suggest ideas and use generateCraftIdea to propose projects.
            When the user picks one, call switchToTutorialMode with the selected craft.
            """,
            // FIXED: both tools are now registered — previously only GenerateCraftIdeaTool was listed
            tools: [GenerateCraftIdeaTool(), SwitchToTutorialModeTool()]
        )
    }
}

// MARK: - ViewModel

@MainActor
final class BrainstormViewModel: ObservableObject {
    @Published var messages: [String] = []
    @Published var isLoading = false

    private let session = LanguageModelSession(
        dynamicInstructions: BrainstormDynamicInstructions()
    )

    func send(_ userMessage: String) async {
        isLoading = true
        defer { isLoading = false }
        do {
            let response = try await session.respond(to: Prompt(userMessage))
            messages.append("You: \(userMessage)")
            messages.append("Assistant: \(response.content)")
        } catch {
            messages.append("Error: \(error.localizedDescription)")
        }
    }
}

// MARK: - View

struct BrainstormView: View {
    @StateObject private var vm = BrainstormViewModel()
    @State private var input = ""

    var body: some View {
        NavigationStack {
            List(vm.messages, id: \.self) { msg in
                Text(msg).font(.footnote)
            }
            HStack {
                TextField("Ask for craft ideas…", text: $input)
                    .textFieldStyle(.roundedBorder)
                Button("Send") {
                    let text = input
                    input = ""
                    Task { await vm.send(text) }
                }
                .disabled(input.isEmpty || vm.isLoading)
            }
            .padding()
            .navigationTitle("Craft Brainstorm")
        }
    }
}