Azure: move models function

This commit is contained in:
Enrico Ros
2025-01-29 11:52:19 -08:00
parent 94e9f2678d
commit 8b8088b74a
2 changed files with 9 additions and 10 deletions
@@ -2,16 +2,6 @@ import { LLM_IF_OAI_Chat, LLM_IF_OAI_Fn, LLM_IF_OAI_Json, LLM_IF_OAI_Vision } fr
import type { ModelDescriptionSchema } from '../../llm.server.types';
import { _knownOpenAIChatModels } from './openai.models';
import { wireOpenPipeModelOutputSchema } from '../openpipe.wiretypes';
export function azureModelToModelDescription(azureDeploymentRef: string, openAIModelIdBase: string, modelCreated: number, modelUpdated?: number): ModelDescriptionSchema {
// if the deployment name mataches an OpenAI model prefix, use that
const known = _knownOpenAIChatModels.find(base => azureDeploymentRef == base.idPrefix);
return fromManualMapping(_knownOpenAIChatModels, known ? azureDeploymentRef : openAIModelIdBase, modelCreated, modelUpdated, undefined, true);
}
// [LM Studio]
export function lmStudioModelToModelDescription(modelId: string): ModelDescriptionSchema {
@@ -622,3 +622,12 @@ export function openAISortModels(a: ModelDescriptionSchema, b: ModelDescriptionS
// }
// return bId.localeCompare(aId);
}
// [Azure]
export function azureModelToModelDescription(azureDeploymentRef: string, openAIModelIdBase: string, modelCreated: number, modelUpdated?: number): ModelDescriptionSchema {
// if the deployment name mataches an OpenAI model prefix, use that
const known = _knownOpenAIChatModels.find(base => azureDeploymentRef == base.idPrefix);
return fromManualMapping(_knownOpenAIChatModels, known ? azureDeploymentRef : openAIModelIdBase, modelCreated, modelUpdated, undefined, true);
}