OpenAI Function Calling

import openai
import requests
import json

OPENAI_API_KEY = "YOUR-OPENAI-API-KEY"
ZIXIANG_API_KEY = "YOUR-ZIXIANG-API-KEY"

# 定义函数描述
functions = [
    {
        "name": "zixiang_search",
        "description": "使用梓享 AI 搜索进行联网搜索，输入应为搜索查询字符串，输出将返回搜索结果的详细信息，包括网页标题、网页URL、网页摘要、网站名称、网站Icon、网页发布时间等。",
        "parameters": {
            "type": "object",
            "properties": {
                "query": {
                    "type": "string",
                    "description": "搜索关键词或问题"
                },
                "engine": {
                    "type": "string",
                    "enum": ["china", "global"],
                    "description": "搜索引擎，china表示中文搜索，global表示全球搜索"
                },
                "count": {
                    "type": "integer",
                    "description": "返回结果数量",
                    "default": 10
                }
            },
            "required": ["query"]
        }
    }
]

def call_zixiang_search(query: str, engine: str = "china", count: int = 10):
    """调用梓享搜索 API"""
    url = "https://search.aiserver.cloud/v1/api"
    headers = {
        "Authorization": f"Bearer {ZIXIANG_API_KEY}",
        "Content-Type": "application/json"
    }
    data = {
        "engine": engine,
        "query": query,
        "max_results": count
    }

    response = requests.post(url, headers=headers, json=data)
    result = response.json()

    if result.get("code") == 0:
        webpages = result["data"]["results"]
        formatted_results = []
        for idx, page in enumerate(webpages, start=1):
            formatted_results.append(
                f"引用: {idx}\n"
                f"标题: {page['title']}\n"
                f"URL: {page['url']}\n"
                f"摘要: {page['content']}\n"
                f"网站名称: {page.get('siteName', '')}\n"
                f"网站图标: {page.get('siteIcon', '')}\n"
                f"发布时间: {page.get('dateLastCrawled', '')}"
            )
        return "\n\n".join(formatted_results)
    else:
        return f"搜索失败: {result.get('msg')}"

# 第一次调用 - 让 GPT 决定是否需要搜索
messages = [
    {"role": "user", "content": "请告诉我小米SU7电动车的最新销量数据"}
]

response = openai.ChatCompletion.create(
    model="gpt-4",
    messages=messages,
    functions=functions,
    function_call="auto"
)

# 如果 GPT 决定调用函数
if response.choices[0].message.get("function_call"):
    function_call = response.choices[0].message["function_call"]
    function_name = function_call["name"]
    function_args = json.loads(function_call["arguments"])

    # 调用搜索 API
    if function_name == "zixiang_search":
        search_result = call_zixiang_search(**function_args)

        # 将搜索结果返回给 GPT
        messages.append(response.choices[0].message)
        messages.append({
            "role": "function",
            "name": function_name,
            "content": search_result
        })

        # 第二次调用 - 让 GPT 基于搜索结果生成答案
        final_response = openai.ChatCompletion.create(
            model="gpt-4",
            messages=messages
        )

        print(final_response.choices[0].message["content"])