#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time    : 2025/3/16 09:46
# @Author  : old-tom
# @File    : llm_agent
# @Project : llmFunctionCallDemo
# @Desc    : llm代理，负责初始化模型
# Press the green button in the gutter to run the script.
from llmagent.llm_agent import ChatAgent

dsr = ChatAgent()

if __name__ == '__main__':
    ##########  测试 多轮对话 #########
    dsr.multi_with_stream('你是什么模型', 1)
    dsr.multi_with_stream('你能做什么', 1)
    dsr.multi_with_stream('我的上一个问题是什么?请直接返回问题,不要有多余输出及思考过程', 1)
    dsr.multi_with_stream('我的第一个问题是什么?请直接返回问题,不要有多余输出及思考过程', 1)

    ##########  测试 多轮对话-相机选择 #########
    dsr.multi_with_tool_call_stream('播放南卡口相机', 1)
    dsr.multi_with_tool_call_stream('1', 1)


    # print(dsr.invoke_with_tool_call('今天昆明天怎么样'))
    ##########  测试 function call #########
    # print(dsr.invoke_with_tool_call('播放南卡口相机'))
    ## [{'name': 'play_video', 'args': {'camera_name': '北卡口1号道相机'}, 'id': 'call_apnb8fiqdkaz313katcs3tjf', 'type': 'tool_call'}]
    # dsr.multi_with_tool_call_stream('将大屏切换为-1分屏', 1)
    ## [{'name': 'split_screen', 'args': {'split_n': 2}, 'id': 'call_2o7c94f591xag8p6lcyice9q', 'type': 'tool_call'}]
    # print(dsr.invoke_with_tool('播放北卡口入境1号道录像,从今天到2025-03-16 02:09:31'))
    ## 由于大模型没有联网，所以无法判断‘今天’
    ## [{'name': 'play_video_record', 'args': {'camera_name': '北卡口入境1号道', 'start_time': '2023-10-25 00:00:00', 'end_time': '2025-03-16 02:09:31'}, 'id': 'call_6gllr9wofeaoomgiz13bgnk6', 'type': 'tool_call'}]
    # print(dsr.invoke_with_tool('播放北卡口入境1号道录像,从25年3月1号到2号'))
    ## [{'name': 'play_video_record', 'args': {'camera_name': '北卡口入境1号道', 'start_time': '2025-03-01 00:00:00', 'end_time': '2025-03-02 23:59:59'}, 'id': 'call_qj78jpep6k5wjwo90dkrz4qt', 'type': 'tool_call'}]
    # print(dsr.invoke_with_tool('播放北卡口入境1号从25年3月1号到2号的录像'))
    ## [{'name': 'play_video_record', 'args': {'camera_name': '北卡口入境1号相机', 'start_time': '2025-03-01 00:00:00', 'end_time': '2025-03-02 23:59:59'}, 'id': 'call_ixpyke880tdhopv27c6moci0', 'type': 'tool_call'}]
    # print(dsr.invoke_with_tool('跳转到首页'))
    ## [{'name': 'switch_page', 'args': {'page_name': '首页'}, 'id': 'call_3ev11bjkyrqyzxaman3402az', 'type': 'tool_call'}]
    # print(dsr.invoke_with_tool('打开日志'))
    ## [{'name': 'switch_page', 'args': {'page_name': '系统日志'}, 'id': 'call_acwy2yk7xz3bgpt28suujca2', 'type': 'tool_call'}]
    # print(dsr.invoke_with_tool('放大地图到2级'))
    ## [{'name': 'zoom_in', 'args': {'level_n': 2}, 'id': 'call_ip1kang0lbr63y8gya8nrpwc', 'type': 'tool_call'}]
    # print(dsr.invoke_with_tool_call('查看成都天府k00航班2004年1月1日入境预报航班人员明细'))
    ## [{'name': 'view_flight_details', 'args': {'airport_name': '成都天府机场', 'flight_code': 'K00', 'flight_date': '2004-01-01', 'ie_type': '入境'}, 'id': 'call_igummeorjq4r2pqjyr9tq6xq', 'type': 'tool_call'}]