Source code for the paper "ORA: Job Runtime Prediction for High-Performance Computing Platforms Using the Online Retrieval-Augmented Language Model"

1. Install Python 3.9. Follow the tutorial for setting up a local RAG environment using Langchain: [https://www.langchain.com.cn/docs/tutorials/local_rag/].
2. If using OpenAI's API, please fill in your OpenAI API Key in main.py.
3. Train and evaluate by running python main.py. The result records will be saved in the res directory.