最近 ChatGPT 突然爆火。抱著好奇的心態(tài)我也去官網(wǎng)注冊(cè)賬號(hào)體驗(yàn)了一下，因?yàn)榫W(wǎng)站人數(shù)太多，一時(shí)半會(huì)竟然注冊(cè)不了，不過(guò)最終還是成功注冊(cè)了。 

在我和 ChatGPT 聊了十幾分鐘之后，這個(gè)東西令我大為震驚，雖然它有些地方顯得笨笨的，帶點(diǎn)機(jī)器人的味道，但是絲毫不影響我對(duì)它的喜愛(ài)。感嘆之余我思考了一下，能不能將它接入到我們現(xiàn)有的應(yīng)用中。于是我仔細(xì)瀏覽了一下 OpenAI 官網(wǎng)，果然已經(jīng)提供了比較多的接口供我們使用。

準(zhǔn)備工作

ChatGPT API 接入需要如下條件。

1. 已成功注冊(cè) OpenAI 的賬號(hào)。
2. 創(chuàng)建 API KEY，這個(gè) API KEY 是用于 HTTP 請(qǐng)求身份驗(yàn)證的，可以創(chuàng)建多個(gè)。如下圖點(diǎn)擊“Create new secret key“，注意這個(gè)創(chuàng)建之后需要馬上復(fù)制好保存，關(guān)閉彈框之后就看不到了。

3. 官方 API 文檔鏈接 api-reference。
4. 注意 API 調(diào)用是收費(fèi)的，但是 OpenAI 已經(jīng)為我們免費(fèi)提供了18美元的用量，足夠大家放心使用。

補(bǔ)全接口示例

該接口功能較多，支持最常用的問(wèn)答功能。

1. 請(qǐng)求方式，Post
2. url，api.openai.com/v1/completi…
3. 請(qǐng)求體 (json)。

{   "model": "text-davinci-003",   "prompt": "Say this is a test",   "max_tokens": 7,   "temperature": 0,   "top_p": 1,   "n": 1,   "stream": false } 復(fù)制代碼

4. 接口文檔，platform.openai.com/docs/api-re…

請(qǐng)求參數(shù)解析

請(qǐng)求示例

本文使用OkHttp作為網(wǎng)絡(luò)請(qǐng)求框架，Moshi作為序列/反序列化框架。編寫(xiě)單元測(cè)試對(duì)該接口進(jìn)行請(qǐng)求，代碼如下。

1. OkHttp 的初始化。

private final OkHttpClient okHttpClient = new OkHttpClient.Builder()         .connectTimeout(10, TimeUnit.SECONDS)         .writeTimeout(10, TimeUnit.SECONDS)         .readTimeout(30, TimeUnit.SECONDS)         .callTimeout(60, TimeUnit.SECONDS)         .build(); 復(fù)制代碼

2. 請(qǐng)求實(shí)體類(lèi)。

private static class CompletionRequest {     private String model = "text-davinci-003";     private String prompt;     private Integer max_tokens = 256;     private float temperature = 0.5f;     private Integer top_p = 1;     private Integer n = 1;     private Boolean stream = false;     private Boolean logprobs;     private String stop; } 復(fù)制代碼

3. 請(qǐng)求方法。

使用OkHttp發(fā)送請(qǐng)求。需注意所有的 Post 請(qǐng)求需要添加頭字段Authorization，值為Bearer YOUR_API_KEY來(lái)驗(yàn)證身份。

public void completion(String prompt) throws IOException {     CompletionRequest completionRequest = new CompletionRequest();     completionRequest.setPrompt(prompt);      String reqJson = moshi.adapter(CompletionRequest.class).toJson(completionRequest);     System.out.println("reqJson: " + reqJson);     Request request = new Request.Builder()             .url("https://api.openai.com/v1/completions")             // 將 API_KEY 替換成你自己的 API_KEY             .header("Authorization", "Bearer " + API_KEY)             .post(RequestBody.create(MEDIA_TYPE_JSON, reqJson))             .build();      try (Response response = okHttpClient.newCall(request).execute()) {         if (!response.isSuccessful()) throw new IOException("Unexpected code " + response);         System.out.println(response.body().string());     } } 復(fù)制代碼

4. 在completion()測(cè)試方法中調(diào)用上面的completion(String prompt)方法

@Test public void completion() {     try {         chatGPT.completion("你能背誦多少位圓周率");     } catch (IOException e) {         e.printStackTrace();     } } 復(fù)制代碼

5. 查看請(qǐng)求結(jié)果

{   "id": "cmpl-6j9PhjAom9GyxBNCVacvckcVNKVZg",   "object": "text_completion",   "created": 1676218965,   "model": "text-davinci-003",   "choices": [     {       "text": "？\n\n我可以背誦50位圓周率：3.1415926535897932384626433832795028841971693993751058209749445923078164062862089986280348253421170679。",       "index": 0,       "logprobs": null,       "finish_reason": "stop"     }   ],   "usage": {     "prompt_tokens": 21,     "completion_tokens": 73,     "total_tokens": 94   } } 復(fù)制代碼

總結(jié)

通過(guò)上面一個(gè)例子，相信大家已經(jīng)學(xué)會(huì)了如何將 ChatGPT API 接入到自己的應(yīng)用了。OpenAI 還提供了其他有用的接口，這里就不一個(gè)一個(gè)說(shuō)明了，大家可以前往官網(wǎng)自行查看文檔。

本文鏈接：http://51huadong.com/cloundnews/11008714.html