diff --git a/packages/foundation-models/src/azure-openai/azure-openai-embedding-response.ts b/packages/foundation-models/src/azure-openai/azure-openai-embedding-response.ts index a3b167b9..dc192deb 100644 --- a/packages/foundation-models/src/azure-openai/azure-openai-embedding-response.ts +++ b/packages/foundation-models/src/azure-openai/azure-openai-embedding-response.ts @@ -34,7 +34,7 @@ export class AzureOpenAiEmbeddingResponse { * Parses the Azure OpenAI response and returns all embeddings. * @returns The embedding vectors. */ - getEmbeddings(): number[][] | undefined { + getEmbeddings(): number[][] { return this.data.data.map(({ embedding }) => embedding); }