Press n or j to go to the next uncovered block, b, p or k for the previous block.
| 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 | 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 3x 3x 3x 3x 2x 2x 2x 2x 2x 1x 1x | import type {
AIProviderFunctionConfig,
ResponseTransformFunction,
} from '@shared/types/ai-providers/config';
import type { CreateEmbeddingsResponseBody } from '@shared/types/api/routes/embeddings-api';
import { AIProvider } from '@shared/types/constants';
import { openAIErrorResponseTransform } from '../openai/utils';
// TODOS: this configuration does not enforce the maximum token limit for the input parameter. If you want to enforce this, you might need to add a custom validation function or a max property to the ParameterConfig interface, and then use it in the input configuration. However, this might be complex because the token count is not a simple length check, but depends on the specific tokenization method used by the model.
export const azureOpenAIEmbedConfig: AIProviderFunctionConfig = {
model: {
param: 'model',
},
input: {
param: 'input',
required: true,
},
user: {
param: 'user',
},
encoding_format: {
param: 'encoding_format',
required: false,
},
dimensions: {
param: 'dimensions',
},
};
export const azureOpenAIEmbedResponseTransform: ResponseTransformFunction = (
aiProviderResponseBody,
aiProviderResponseStatus,
) => {
if (aiProviderResponseStatus !== 200 && 'error' in aiProviderResponseBody) {
return openAIErrorResponseTransform(
aiProviderResponseBody,
AIProvider.AZURE_OPENAI,
);
}
return aiProviderResponseBody as unknown as CreateEmbeddingsResponseBody;
};
|