from data_analysis_agent import DataAnalysisAgent
from config.llm_config import LLMConfig


def main():
    llm_config = LLMConfig()
    # 如果希望强制运行到最大轮数，设置 force_max_rounds=True
    agent = DataAnalysisAgent(llm_config, force_max_rounds=False)
    files = ["./UB IOV Support_TR.csv"]
    report = agent.analyze(
        user_input="基于所有有关远程控制的问题，以及涉及车控APP的运维工单的数据，输出若干个重要的统计指标，并绘制相关图表。总结一份，车控APP，及远程控制工单健康度报告，最后生成汇报给我。",
        files=files,
    )
    print(report)
 

if __name__ == "__main__":
    main()