Python中使用OpenAI API并结合function_tools来获取天气情况

admin

要在Python中使用OpenAI API并结合function_tools来获取天气情况，你可以按照以下步骤进行操作。假设你已经有了OpenAI的API密钥，并且已经安装了openai库。

1. 安装所需的库

首先，确保你已经安装了openai库。如果没有安装，可以使用以下命令进行安装：

pip install openai
2. 编写Python代码

接下来，你可以编写一个Python脚本来调用OpenAI API，并使用function_tools来获取天气情况。

1. import openai
   
2. import requests
   
3. ​
   
4. # 设置OpenAI API密钥
   
5. openai.api_key = 'your-openai-api-key'
   
6. ​
   
7. # 定义一个函数来获取天气情况
   
8. def get_weather(city: str) -> str:
   
9.     """
   
10.     获取指定城市的天气情况
    
11.     :param city: 城市名称
    
12.     :return: 天气情况描述
    
13.     """
    
14.     # 这里使用一个简单的天气API作为示例
    
15.     api_key = 'your-weather-api-key'  # 你需要一个天气API的密钥
    
16.     url = f'http://api.weatherapi.com/v1/current.json?key={api_key}&q={city}'
    
17.     response = requests.get(url)
    
18.     data = response.json()
    
19.    
    
20.     if 'error' in data:
    
21.         return f"无法获取{city}的天气信息: {data['error']['message']}"
    
22.    
    
23.     weather = data['current']['condition']['text']
    
24.     temperature = data['current']['temp_c']
    
25.     return f"{city}的天气情况: {weather}, 温度: {temperature}°C"
    
26. ​
    
27. # 定义function_tools
    
28. function_tools = [
    
29.     {
    
30.         "name": "get_weather",
    
31.         "description": "获取指定城市的天气情况",
    
32.         "parameters": {
    
33.             "type": "object",
    
34.             "properties": {
    
35.                 "city": {
    
36.                     "type": "string",
    
37.                     "description": "城市名称"
    
38.                 }
    
39.             },
    
40.             "required": ["city"]
    
41.         }
    
42.     }
    
43. ]
    
44. ​
    
45. # 调用OpenAI API
    
46. def chat_with_gpt(prompt):
    
47.     response = openai.ChatCompletion.create(
    
48.         model="gpt-3.5-turbo",
    
49.         messages=[{"role": "user", "content": prompt}],
    
50.         functions=function_tools,
    
51.         function_call="auto"
    
52.     )
    
53.    
    
54.     return response
    
55. ​
    
56. # 处理OpenAI的响应
    
57. def handle_response(response):
    
58.     message = response['choices'][0]['message']
    
59.    
    
60.     if message.get("function_call"):
    
61.         function_name = message["function_call"]["name"]
    
62.         if function_name == "get_weather":
    
63.             city = eval(message["function_call"]["arguments"])["city"]
    
64.             return get_weather(city)
    
65.    
    
66.     return message["content"]
    
67. ​
    
68. # 示例对话
    
69. prompt = "请问北京的天气怎么样？"
    
70. response = chat_with_gpt(prompt)
    
71. result = handle_response(response)
    
72. print(result)

复制代码

3. 运行代码

将上述代码保存为一个Python文件（例如weather_bot.py），然后运行它。你应该会看到类似以下的输出：

北京的天气情况: 晴, 温度: 25°C
4. 注意事项

• 你需要替换your-openai-api-key和your-weather-api-key为你自己的API密钥。
• 这个示例使用了weatherapi.com作为天气数据源，你可以根据需要替换为其他天气API。
• function_tools是OpenAI API中的一个功能，允许你定义自定义函数并在对话中调用它们。
  
  

通过这种方式，你可以结合OpenAI的强大语言模型和自定义函数来获取实时的天气信息。