基于GPT搭建私有知识库聊天机器人（五）函数调用

> Entering new  chain...

Invoking: `query_bus_by_date` with `{'drv_date': '2023-07-14', 'start_name': '成都', 'target_name': '稻城'}`


[{'boardingAddress': '', 'boardingCode': '', 'boardingName': '', 'busNo': '川A88888', 'busTypeName': '大型高一', 'childSAmount': '500', 'displayExtraFlag': '', 'drvTime': '2023-07-14 20:00', 'endName': '亚丁', 'extraFlag': '0', 'fullPrice': '0.02', 'halfPrice': '130.00', 'isForbid': '0', 'isPassStation': '1', 'isRealNameLine': '1', 'mile': '100', 'motName': 'XXXX', 'passId': '6', 'refundTimeLimit': '', 'schId': '5JhYm', 'schTypeId': '1', 'scheduleType': '1', 'seatAmount': '4998', 'seatTypeName': '', 'signId': 'v21ofWkSTmttd8mVuwxZKL5p', 'startCityName': '成都', 'startCityNo': '510100', 'startDrvTime': '', 'startStationName': '城北客运站', 'startStationNo': 'cbcz', 'stationGuidePrice': '', 'stopAddress': '', 'stopCode': '', 'stopName': '稻城', 'stopTime': '', 'targetCityName': '', 'targetCityNo': '', 'targetStationName': '', 'targetStationNo': '', 'useMinutes': ''}, {'boardingAddress': '', 'boardingCode': '', 'boardingName': '', 'busNo': '川A88888', 'busTypeName': '大型高一', 'childSAmount': '4', 'displayExtraFlag': '', 'drvTime': '2023-07-14 07:30', 'endName': '亚丁', 'extraFlag': '0', 'fullPrice': '0.02', 'halfPrice': '130.00', 'isForbid': '0', 'isPassStation': '1', 'isRealNameLine': '1', 'mile': '100', 'motName': 'XXXXX', 'passId': '6', 'refundTimeLimit': '', 'schId': 'fPDLG', 'schTypeId': '0', 'scheduleType': '1', 'seatAmount': '42', 'seatTypeName': '', 'signId': 'mVpTnMHS7i9ZEQxl9JddzkI4', 'startCityName': '成都', 'startCityNo': '510100', 'startDrvTime': '', 'startStationName': '城北客运站', 'startStationNo': 'cbcz', 'stationGuidePrice': '', 'stopAddress': '', 'stopCode': '', 'stopName': '稻城', 'stopTime': '', 'targetCityName': '', 'targetCityNo': '', 'targetStationName': '', 'targetStationNo': '', 'useMinutes': ''}]

根据查询结果，我找到了两个班次可供选择：

1. 班次号：川A88888，车型：大型高一，出发时间：2023-07-14 20:00，起点：成都城北客运站，终点：稻城亚丁。全票价格为0.02元，半票价格为130.00元。该班次由XXXX运营，共有4998个座位。

2. 班次号：川A88888，车型：大型高一，出发时间：2023-07-14 07:30，起点：成都城北客运站，终点：稻城亚丁。全票价格为0.02元，半票价格为130.00元。该班次由XXXX运营，共有42个座位。

请问您对以上两个班次有什么要求或者偏好吗？

> Finished chain.

from langchain.tools import BaseTool
from langchain.chat_models import ChatOpenAI
from langchain.schema import HumanMessage
from langchain.tools import format_tool_to_openai_function
from langchain.agents import AgentType, initialize_agent
from pydantic import BaseModel, Field
from typing import Optional, Type
from datetime import date
import requests
import json
import os

llm = ChatOpenAI(temperature=0, model="gpt-3.5-turbo-0613") #注意这里要用gpt-3.5-turbo-0613模型才能支持函数调用

# 定义入参解析规则
class ScheduleCheckInput(BaseModel):
    drv_date: str = Field(..., description="日期，请格式化为yyyy-mm-dd，日期当天从%s开始计算" % date.today())
    start_name: str = Field(..., description="起点")
    target_name: str = Field(..., description="终点")


class BusTool(BaseTool):
    name = "query_bus_by_date"
    description = "根据日期查询起止点的班次信息"

    def _run(self, drv_date, start_name, target_name):
        url = "http://test.test.com/"
        data = {"body": {
                    "startName": "%s",
                    "targetNo": "%s",
                    "drvTime": "%s"}
                }
        json_data = json.dumps(data) % (start_name, target_name, drv_date)
        response = requests.post(url, data=json_data.encode("utf-8"))
        return response.json().get("body").get("data")

    def _arun(self):
        raise NotImplementedError("This tool does not support async1")

    args_schema: Optional[Type[BaseModel]] = ScheduleCheckInput


def search_schedule(query: str) -> str:
    bus_tools = [BusTool()]
    open_ai_agent = initialize_agent(bus_tools,
                                     llm,
                                     agent=AgentType.OPENAI_FUNCTIONS,
                                     verbose=True)
    return open_ai_agent.run(query)


if __name__ == '__main__':
    search_schedule("请帮我查询13号成都到绵阳的班次信息")