Function download_file

src/inference.py:1115–1132

(model, ids)

        model.delete_file(i)

def download_file(model, ids):
    model = RemoteModel(model)
    contents = []
    for i in ids:
        contents += model.download_file(i)
    data = {}
    for d in contents:
        data[d["custom_id"]] = d["response"]["body"]["choices"][0]["message"]
    
    errors = []
    for i in ids:
        errors += model.collect_errors(i)
    
    for e in errors:
        data[e["custom_id"]] = None

    with open("prompt_chat_results/gpt-4/time/rd1_batch_outputs.json", "w", encoding = "utf-8") as f:
        f.write(json.dumps(data, sort_keys=True, indent=4, separators=(',', ': ')))


def batch_run(model, ids):