def try_request()

in src/scripts/gen_embeddings.py [0:0]

• 12 lines of code
• 4 McCabe index (conditional complexity)

def try_request(model_name: str, samples: list, retries: int = 0):
    if retries >= 5:
        return
    try:
        response = openai.Embedding.create(input=samples, model=model_name)['data']
        return response
    except openai.error.InvalidRequestError:
        return
    except Exception as e:
        print(f'Retrying after catching an exception, try {retries + 1}\n{e}')
        time.sleep(5)
        return try_request(model_name, samples, retries + 1)