@roadiehq/rag-ai-backend-embeddings-openai
Version:
The OpenAI backend module for the @roadiehq/rag-ai plugin.
53 lines (47 loc) • 1.52 kB
TypeScript
/*
* Copyright 2024 Larder Software Limited
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
export interface Config {
/**
* OpenAI Embeddings configuration
*
*/
ai: {
embeddings: {
openai: {
/**
* The API key for accessing OpenAI services. Defaults to process.env.OPENAI_API_KEY
*/
openAiApiKey?: string;
/**
* Specify URL of self-hosted OpenAI compliant endpoint. Defaults to OpenAI's public API https://api.openai.com
*/
openAiBaseUrl?: string;
/**
* Name of the OpenAI model to use to create Embeddings. Defaults to text-embedding-3-small
*/
modelName?: string;
/**
* The size of the batch to use when creating embeddings.
*/
batchSize?: number;
/**
* The number of dimensions to generate. Defaults to use the default value from the chosen model
*/
embeddingsDimensions?: number;
};
};
};
}