Function getSecondaryModel

lib/queryLLM.ts:356–367

(mainModel: string)

};

export function getSecondaryModel(mainModel: string): string {
  if (mainModel in baseSecondaryModelMapping) {
    // @ts-ignore
    return baseSecondaryModelMapping[mainModel];
  } else if (isOSModel(mainModel)) {
    // Use OS model as secondary if primary model is OS
    return mainModel;
  } else {
    // Default for fine-tuned models
    return "gpt-3.5-turbo-0125";
  }
}

export async function queryEmbedding(
  textToEmbed: string | string[],