diff --git a/guardrails/utils/openai_utils/v1.py b/guardrails/utils/openai_utils/v1.py index 1fa01792a..8249d791f 100644 --- a/guardrails/utils/openai_utils/v1.py +++ b/guardrails/utils/openai_utils/v1.py @@ -272,7 +272,9 @@ async def create_embedding( model=model, input=input, ) - return [r.embedding for r in embeddings.data] + # Use a list comprehension with a local variable lookup to improve speed + data = embeddings.data + return [r.embedding for r in data] async def create_completion( self, engine: str, prompt: str, *args, **kwargs