import sys
sys.path.append('../..')
sys.path.append('.')
sys.path.append('./')
from Tools.RagQATool import rag_qa
from Tools.ExtractInfo import extract_info
from Tools.FetchFromOtherSys import fetch_data_from_external_systems
from Tools.ReportGenerationTool import generator

# 检索
def dm01():
    print(rag_qa("为什么有些机器人会'迷路'？"))


# 生成报告
def dm02():
    print(extract_info("根据我的扫地机器人的使用情况，生成一份使用报告,要2026年6月份的"))


# 第三方系统中获取数据
def dm03():
    # query中有时间信息
    a = extract_info("根据我的扫地机器人的使用情况，生成一份使用报告,要2026年6月份的")
    b = fetch_data_from_external_systems(info_dict=a)
    print('有时间的数据生成：', b)

    # query中没有时间信息
    a1 = extract_info("根据我的扫地机器人的使用情况，生成一份使用报告")
    b1 = fetch_data_from_external_systems(info_dict=a1)
    print('没时间的数据生成：', b1)


# 报告生成
def dm04():
    b = generator("根据我的扫地机器人的使用情况，生成一份使用报告,要2026年6月份的")
    print(b)


if __name__ == '__main__':
    # dm01()
    # dm02()
    dm03()
    # dm04()