Swift OpenAI API - 生成心理医生回复

以下是一段使用Swift编写的代码示例，用于向OpenAI请求生成文本，并设置角色为心理医生：

import Foundation

func generateText(prompt: String) {
    let apiKey = 'YOUR_API_KEY'
    let urlString = 'https://api.openai.com/v1/engines/davinci-codex/completions'
    
    let parameters: [String: Any] = [
        'prompt': prompt,
        'max_tokens': 50,
        'temperature': 0.8,
        'n': 3
    ]
    
    guard let url = URL(string: urlString) else {
        print('Invalid URL')
        return
    }
    
    var request = URLRequest(url: url)
    request.httpMethod = 'POST'
    request.setValue('Bearer (apiKey)', forHTTPHeaderField: 'Authorization')
    request.setValue('application/json', forHTTPHeaderField: 'Content-Type')
    
    guard let httpBody = try? JSONSerialization.data(withJSONObject: parameters, options: []) else {
        print('Invalid request body')
        return
    }
    
    request.httpBody = httpBody
    
    let task = URLSession.shared.dataTask(with: request) { (data, response, error) in
        if let error = error {
            print('Request error: (error)')
            return
        }
        
        guard let data = data else {
            print('Empty response data')
            return
        }
        
        do {
            if let json = try JSONSerialization.jsonObject(with: data, options: []) as? [String: Any] {
                if let completions = json['choices'] as? [[String: Any]] {
                    for completion in completions {
                        if let text = completion['text'] as? String {
                            print(text)
                        }
                    }
                }
            }
        } catch {
            print('Error parsing response: (error)')
        }
    }
    
    task.resume()
}

// 使用示例
let prompt = '用户发送一段文字'
generateText(prompt: prompt)

请注意，上述代码中的 YOUR_API_KEY 需要被替换为您自己的 OpenAI API 密钥。此外，还要确保您已经安装了 Swift 的 URLSession 库以及 Foundation 库。

在请求中，您可以将 OpenAI对象 设置为 心理医生，以便生成与心理医生相关的回复。