ChatGPT 如何實現自然語言搜索企業數據？

作者：金木 2024-08-07 10:39:47

人工智能

你可以根據具體的 REST API 文檔調整 API URL 和請求參數。確保 API 返回的數據格式與你的 format_response 函數兼容。

使用 ChatGPT 實現 function calling 進行自然語言搜索企業數據，并通過 REST API 查詢數據，可以遵循以下步驟：

1、定義 function calling 接口：

定義一個函數接口，用于處理自然語言查詢并通過 REST API 查詢數據。

2、解析自然語言查詢：

使用 ChatGPT 的內置能力解析用戶的自然語言查詢，將其轉換為結構化的查詢參數。

3、構建 REST API 請求：

使用解析出的參數構建 API 請求。

4、發送請求并獲取響應：

調用 REST API 并獲取響應數據。

5、返回查詢結果：

將查詢結果格式化并返回給用戶。

以下是一個具體的示例代碼：

import requests


import json


# 定義函數接口


def search_companies(query):


    # 解析查詢


    entities = parse_query(query)


    # 構建 API 請求


    api_response = call_api(entities)


    # 格式化響應


    response = format_response(api_response)


    return response






# 解析查詢函數


def parse_query(query):


    # 這里可以使用任意的 NLP 庫進行查詢解析


    # 為簡單起見，這里手動解析


    entities = {}


    if "收入" in query:


        entities["revenue"] = "1000000"  # 示例值


    if "科技公司" in query:


        entities["industry"] = "technology"


    if "2023年" in query:


        entities["year"] = "2023"


    return entities






# 構建 API 請求并調用 API


def call_api(entities):


    api_url = "https://api.example.com/companies"


    params = {


        "year": entities.get("year", "2023"),


        "revenue_gt": entities.get("revenue", "1000000"),


        "industry": entities.get("industry", "technology"),


    }


    response = requests.get(api_url, params=params)


    return response.json()






# 格式化響應函數


def format_response(api_response):


    response = "找到以下符合條件的公司:\n"


    for company in api_response:


        response += f"公司名: {company['name']}, 收入: {company['revenue']}, 行業: {company['industry']}\n"


    return response






# ChatGPT function calling 示例


def chatgpt_function_calling(query):


    response = search_companies(query)


    return response






# 用戶輸入的查詢


user_query = "查找2023年收入超過100萬美元的科技公司"


# 調用 function calling 接口


print(chatgpt_function_calling(user_query))

在上面的代碼中：

parse_query 函數解析自然語言查詢。
call_api 函數構建并發送 REST API 請求。
search_companies 函數執行整個查詢流程，包括解析查詢、構建 API 請求、調用 API 和格式化響應。
chatgpt_function_calling 函數模擬 ChatGPT 的 function calling

接口，接收用戶的自然語言查詢并返回結果。