调用外部 API

使用 OpenAI API 构建 AI 系统

Francesca Donadoni

Curriculum Manager, DataCamp

用于 API 的 Python requests 库

显示对芝加哥艺术学院 API 请求的示意图

用于 API 的 Python requests 库

导入库
```
import requests
```

提供 URL

url = "https://api.artic.edu/api/v1/artworks/search"

提供查询参数
```
querystring = {"q":keyword}
```

获取响应

response = requests.request("GET", url, params=querystring)

¹ https://api.artic.edu/docs/#quick-start

封装函数

import requests

def get_artwork(keyword):
    url = "https://api.artic.edu/api/v1/artworks/search"
    querystring = {"q":keyword}
    response = requests.request("GET", url, params=querystring)
    return response.text

添加上下文

response = client.chat.completions.create(
    model="gpt-4o-mini",

    messages=[
{"role": "system",
 "content": "You are an AI assistant, a specialist in history 
    of art. You should interpret the user prompt, and based on it extract one 
    keyword for recommending artwork related to their preference."},

{"role": "user", 
 "content": "I don't have much time to visit the museum and would 
    like some recommendations. I like the seaside and quiet places."}
],

将函数添加到 tools

    function_definition=[{"type": "function",
            "function" : {
                "name": "get_artwork",
                "description": "This function calls the Art Institute of Chicago API 
                                to find artwork that matches a keyword",
                "parameters": {

                    "type": "object",
                    "properties": {
                        "artwork keyword": {
                            "type": "string",
                            "description": "The keyword to be passed to the 
                                            get_artwork function."}}},
                "result": {"type": "string"}
            }} ] )

整合到一起

import json

if response.choices[0].finish_reason=='tool_calls':
    function_call = response.choices[0].message.tool_calls[0].function

    if function_call.name == "get_artwork":
        artwork_keyword = json.loads(function_call.arguments)["artwork keyword"]
        artwork = get_artwork(artwork_keyword)

        if artwork: 
            print(f"Here are some recommendations: 
                    {[i['title'] for i in json.loads(artwork)['data']]}")
        else:
            print("Apologies, I couldn't make any recommendations based on the request.")

    else:
        print("Apologies, I couldn't find any artwork.")

else:
    print("I am sorry, but I could not understand your request.")

最终响应

Here are some recommendations: ['Seaside, Port of Honfleur', 
'Little Landscape at the Seaside (Kleine Landschaft am Meer)', 
'Museum of Contemporary Art, Niterói, Rio de Janeiro, Brazil, Four Sketches',
'A seaside outing', 'Stacks of Wheat (End of Day, Autumn)', 'Stack of Wheat', 
'Stacks of Wheat (Sunset, Snow Effect)', 'Stacks of Wheat (End of Summer)', 
'Stack of Wheat (Thaw, Sunset)', 'Waitress at a Seaside Teahouse']

一幅印象派海边有船的画

¹ Seaside, Port of Honfleur | 芝加哥艺术学院

Let's practice!

使用 OpenAI API 构建 AI 系统