def compute_embedding_in_batch()

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


def compute_embedding_in_batch(texts):
    refresh_openai_token()
    embedding_args = {"deployment_id": args.openaideployment} if args.openaihost == "azure" else {}
    emb_response = openai.Embedding.create(**embedding_args, model=args.openaimodelname, input=texts)
    return [data.embedding for data in emb_response.data]