@microsoft/teams-ai/lib/internals/testing/TestEmbeddings.d.ts

SDK focused on building AI based applications for Microsoft Teams.

/**
 * @module teams-ai
 */
/**
 * Copyright (c) Microsoft Corporation. All rights reserved.
 * Licensed under the MIT License.
 */
import { EmbeddingsModel, EmbeddingsResponse, EmbeddingsResponseStatus } from '../../types/EmbeddingsModel';
/**
 * A test model that can be used to test the prompt completion system.
 */
export declare class TestEmbeddings implements EmbeddingsModel {
    /**
     * Default output to return for test embeddings.
     */
    static TEST_OUTPUT: number[][];
    /**
     *
     * @param {EmbeddingsResponseStatus} status Optional. Status of the embeddings response. Defaults to `success`.
     * @param {number[][] | undefined} output Optional. Embeddings to generate. Defaults to `[[1,2,3,4,5]]`.
     * @param {string} message Optional. Message to return with response.
     */
    constructor(status?: EmbeddingsResponseStatus, output?: number[][] | undefined, message?: string);
    /**
     * Status of the embeddings response.
     */
    status: EmbeddingsResponseStatus;
    /**
     * Generated embeddings.
     */
    output: number[][] | undefined;
    /**
     * Message to return with response.
     */
    message?: string;
    /**
     * Returns a generated set of test embeddings
     * @param {string} model Name of the model to use (or deployment for Azure).
     * @param {string | string[]} inputs Input to generate embeddings for.
     * @returns {Promise<EmbeddingsResponse>} The generated embeddings.
     */
    createEmbeddings(model: string, inputs: string | string[]): Promise<EmbeddingsResponse>;
}
//# sourceMappingURL=TestEmbeddings.d.ts.map