def compute_embedding()

in src/prepdocs/prepdocs.py [0:0]

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

def compute_embedding(text, embedding_deployment, embedding_model):
    refresh_openai_token()
    embedding_args = {"deployment_id": embedding_deployment} if args.openaihost == "azure" else {}
    return openai.Embedding.create(**embedding_args, model=embedding_model, input=text)["data"][0]["embedding"]