import Foundation

enum Prompt {
    static let defaultTask =
        "Given a web search query, retrieve relevant passages that answer the query"

    static func wrap(_ text: String, _ kind: InputKind) -> String {
        switch kind {
        case .document:
            return text
        case .query:
            return "Instruct: \(defaultTask)\nQuery: \(text)"
        }
    }
}