screenshot-to-code/frontend/src/lib/models.ts
2024-07-11 16:51:20 -04:00

36 lines
969 B
TypeScript

// Keep in sync with backend (llm.py)
// Order here matches dropdown order
export enum CodeGenerationModel {
CLAUDE_3_5_SONNET_2024_06_20 = "claude-3-5-sonnet-20240620",
GPT_4O_2024_05_13 = "gpt-4o-2024-05-13",
GPT_4_TURBO_2024_04_09 = "gpt-4-turbo-2024-04-09",
GPT_4_VISION = "gpt_4_vision",
CLAUDE_3_SONNET = "claude_3_sonnet",
}
// Will generate a static error if a model in the enum above is not in the descriptions
export const CODE_GENERATION_MODEL_DESCRIPTIONS: {
[key in CodeGenerationModel]: {
name: string;
inBeta: boolean;
};
} = {
"gpt-4o-2024-05-13": { name: "GPT-4o", inBeta: false },
"claude-3-5-sonnet-20240620": {
name: "Claude 3.5 Sonnet",
inBeta: false,
},
"gpt-4-turbo-2024-04-09": {
name: "GPT-4 Turbo (deprecated)",
inBeta: false,
},
gpt_4_vision: {
name: "GPT-4 Vision (deprecated)",
inBeta: false,
},
claude_3_sonnet: {
name: "Claude 3 (deprecated)",
inBeta: false,
},
};