@sap-ai-sdk/foundation-models
Version:
SAP Cloud SDK for AI is the official Software Development Kit (SDK) for **SAP AI Core**, **SAP Generative AI Hub**, and **Orchestration Service**.
41 lines • 1.24 kB
JavaScript
import { createLogger } from '@sap-cloud-sdk/util';
const logger = createLogger({
package: 'foundation-models',
messageContext: 'azure-openai-embedding-response'
});
/**
* Azure OpenAI embedding response.
*/
export class AzureOpenAiEmbeddingResponse {
rawResponse;
/**
* The embedding response.
*/
data;
constructor(rawResponse) {
this.rawResponse = rawResponse;
this.data = rawResponse.data;
}
/**
* Parses the Azure OpenAI response and returns the embedding.
* @param dataIndex - The index of the data to parse.
* @returns The embedding vector.
*/
getEmbedding(dataIndex = 0) {
this.logInvalidDataIndex(dataIndex);
return this.data.data[dataIndex]?.embedding;
}
/**
* Parses the Azure OpenAI response and returns all embeddings.
* @returns The embedding vectors.
*/
getEmbeddings() {
return this.data.data.map(({ embedding }) => embedding);
}
logInvalidDataIndex(dataIndex) {
if (dataIndex < 0 || dataIndex >= this.data.data.length) {
logger.error(`Data index ${dataIndex} is out of bounds.`);
}
}
}
//# sourceMappingURL=azure-openai-embedding-response.js.map