def main()

in scripts/generate_workloads/workloads.py [0:0]

• 11 lines of code
• 2 McCabe index (conditional complexity)

def main():
    workloads = read_file(FILE_PATH)

    client = AzureOpenAI(
        api_key = os.getenv("AZURE_OPENAI_API_KEY"),
        api_version = "2024-02-01",
        azure_endpoint = os.getenv("AZURE_OPENAI_ENDPOINT")
    )

    finished_workloads = []

    for workload in workloads:
        finished_workloads.append(get_responses(workload, client, DEPLOYMENT_MODEL))
    
    write_file(SUCCESSFUL_WORKLOADS, finished_workloads)