極簡 Python：10分鐘會(huì)用 OpenAI / Kimi API

如果你是 OpenAI 的用戶，請(qǐng)將下面的代碼復(fù)制進(jìn) Colab：

!pip install openaifrom openai import OpenAI
client = OpenAI(    api_key="sk-yTmYibGsJ637VUkki2GZUzZ9vh7EWFEJ",    )completion = client.chat.completions.create(  model="gpt-3.5-turbo",  messages=[    {"role": "user", "content": "你好"}  ])
print(completion)

如果你是其他 AI 的用戶，以 Kimi 為例，復(fù)制這個(gè)：

!pip install openaifrom openai import OpenAI
client = OpenAI(    api_key="sk-yTmYibGsJ637VUkki2GZUzZ9vh7EWFEJ",    base_url="https://api.moonshot.cn/v1"    )completion = client.chat.completions.create(  model="moonshot-v1-8k",  messages=[    {"role": "user", "content": "你好"}  ])
print(completion)

注意：你要把 api_key 換成你的

細(xì)心的你會(huì)發(fā)現(xiàn)，無論是什么 AI API，大家的接口調(diào)用方法，都幾乎一致。唯變化的，是 base_url 和 model 的值。

選擇代碼單元格，點(diǎn)擊左側(cè)的“運(yùn)行”按鈕或快捷鍵 Ctrl + Enter，開始運(yùn)行代碼。 

剛開始運(yùn)行時(shí)，Colab 會(huì)安裝 OpenAI 庫，可能需要幾十秒，并輸出類似這樣的內(nèi)容：

安裝完成后，你會(huì)看到回復(fù)被打印出來，類似這樣:

仔細(xì)看這里的中文部分：‘你好！今天我能幫你做什么？’

接口是通的！恭喜，你已經(jīng)成功地與 AI 完成了第一次對(duì)話!

仔細(xì)看回復(fù)，是不是很奇怪？里面有著 id、choices 這樣的詞。 

其實(shí)，這就像一個(gè)包裹，AI 的回復(fù)內(nèi)容被裝在了好幾層「盒子」里面。我們要一層一層地打開，才能看到最里面的回復(fù)內(nèi)容。 completion 就是最外面的那個(gè)大「盒子」，我們需要先打開它。 接下來是 choices，"choices"的意思是「選擇」，這里面裝著幾個(gè)備選的回復(fù)，但通常我們只需要第一個(gè)。

在 Python 里，我們用 [0] 來表示第一個(gè)。再打開 choices[0]，我們會(huì)看到一個(gè)叫 message 的盒子。這里面裝的就是 ChatGPT 的回復(fù)了，但還差一點(diǎn)點(diǎn)。 最后，message 里面還有一層叫 content 的盒子，打開它，我們就能看到 ChatGPT 的回復(fù)內(nèi)容了。 

所以，想要看到 ChatGPT 的回復(fù)，我們需要一步一步地「拆盒子」: completion → choices → [0] → message → content 在代碼里，就是這樣寫: completion.choices[0].message.content。

現(xiàn)在，讓我們把 print(completion) 改成 print(completion.choices[0].message.content) ：

!pip install openaifrom openai import OpenAI
client = OpenAI(    api_key="sk-yTmYibGsJ637VUkki2GZUzZ9vh7EWFEJ",    )completion = client.chat.completions.create(  model="gpt-3.5-turbo",  messages=[    {"role": "user", "content": "你好"}  ])
print(completion.choices[0].message.content)

這時(shí)候再運(yùn)行代碼，你就能看到 ChatGPT 干凈利落的回復(fù)了: 

雖然這些"盒子"的名字看起來有點(diǎn)復(fù)雜，但你只需要記住用 completion.choices[0].message.content 就能打開最后一個(gè)"盒子"，看到回復(fù)內(nèi)容。 以后如果你在學(xué)習(xí) Python 時(shí)遇到了類似的情況，不要害怕，就想象成在拆一個(gè)套娃，一層一層地打開，最后你就能得到你想要的結(jié)果了。

如果每次提問，都要重復(fù)改代碼，就太累了

所以，我們可以試試函數(shù)

def poem(s):  print(f"{s}，從此君王不早朝")

poem("春宵苦短日高起")poem("但使龍城飛將在")poem("我為將軍解戰(zhàn)袍")poem("雞哥練習(xí)兩年半")

春宵苦短日高起，從此君王不早朝但使龍城飛將在，從此君王不早朝我為將軍解戰(zhàn)袍，從此君王不早朝雞哥練習(xí)兩年半，從此君王不早朝

如果你希望學(xué)習(xí)更多模型的調(diào)用，比如聲音轉(zhuǎn)文字、文字轉(zhuǎn)圖片，或者更加細(xì)致的定義處理風(fēng)格（嚴(yán)謹(jǐn)還是更有創(chuàng)意的），可以參考 OpenAI 的官方文檔。

文檔地址：

https://platform.openai.com/docs/api-reference

接下來本大聰明還會(huì)「以實(shí)用為核心」，再更新一篇新手極其友好的「Python Cheat Sheet」，幫助大家溫故知新。