import time
from module.qa_chain import QAChain

if __name__ == "__main__":     

    # 使用自定义配置文件
    custom_config = "config/qa_config.json"
    custom_chain = QAChain(config_path=custom_config)
    try:
        while True:
            question = input("请输入问题: ")
            if question == "exit":
                break
            start = time.time()
            result = custom_chain.invoke({"question": question})
            print(f"回答: {result['answer']}")
            print(f"耗时: {time.time() - start} 秒")
    except Exception as e:
        print(f"自定义配置处理失败: {e}")