feat: add openai o1 models

This commit is contained in:
Carl-Robert Linnupuu 2024-09-13 15:34:27 +03:00
parent 0e53cb6fbb
commit a7efa6fe44

View file

@ -14,7 +14,8 @@ object CodeGPTAvailableModels {
fun getToolWindowModels(pricingPlan: PricingPlan?): List<CodeGPTModel> {
return when (pricingPlan) {
null, ANONYMOUS -> listOf(
CodeGPTModel("GPT-4o", "gpt-4o", Icons.OpenAI, INDIVIDUAL),
CodeGPTModel("o1-preview", "o1-preview", Icons.OpenAI, INDIVIDUAL),
CodeGPTModel("o1-mini", "o1-mini", Icons.OpenAI, FREE),
CodeGPTModel("Claude 3.5 Sonnet", "claude-3.5-sonnet", Icons.Anthropic, INDIVIDUAL),
CodeGPTModel("Llama 3.1 (405B)", "llama-3.1-405b", Icons.Meta, INDIVIDUAL),
CodeGPTModel("DeepSeek Coder V2", "deepseek-coder-v2", Icons.DeepSeek, INDIVIDUAL),
@ -23,15 +24,16 @@ object CodeGPTAvailableModels {
)
FREE -> listOf(
CodeGPTModel("GPT-4o", "gpt-4o", Icons.OpenAI, INDIVIDUAL),
CodeGPTModel("o1-preview", "o1-preview", Icons.OpenAI, INDIVIDUAL),
CodeGPTModel("o1-mini", "o1-mini", Icons.OpenAI, FREE),
CodeGPTModel("Claude 3.5 Sonnet", "claude-3.5-sonnet", Icons.Anthropic, INDIVIDUAL),
CodeGPTModel("GPT-4o mini", "gpt-4o-mini", Icons.OpenAI, ANONYMOUS),
CodeGPTModel("Llama 3 (70B)", "llama-3-70b", Icons.Meta, FREE),
CodeGPTModel("Mixtral (8x22B)", "mixtral-8x22b", Icons.CodeGPTModel, FREE),
CodeGPTModel("Code Llama (70B)", "codellama:chat", Icons.Meta, FREE),
)
INDIVIDUAL -> listOf(
CodeGPTModel("o1-preview", "o1-preview", Icons.OpenAI, INDIVIDUAL),
CodeGPTModel("GPT-4o", "gpt-4o", Icons.OpenAI, INDIVIDUAL),
CodeGPTModel("Claude 3 Opus", "claude-3-opus", Icons.Anthropic, INDIVIDUAL),
CodeGPTModel("Claude 3.5 Sonnet", "claude-3.5-sonnet", Icons.Anthropic, INDIVIDUAL),
@ -44,7 +46,8 @@ object CodeGPTAvailableModels {
@JvmStatic
val ALL_CHAT_MODELS: List<CodeGPTModel> = listOf(
CodeGPTModel("GPT-4o", "gpt-4o", Icons.OpenAI, INDIVIDUAL),
CodeGPTModel("o1-preview", "o1-preview", Icons.OpenAI, INDIVIDUAL),
CodeGPTModel("o1-mini", "o1-mini", Icons.OpenAI, FREE),
CodeGPTModel("GPT-4o mini", "gpt-4o-mini", Icons.OpenAI, ANONYMOUS),
CodeGPTModel("Claude 3 Opus", "claude-3-opus", Icons.Anthropic, INDIVIDUAL),
CodeGPTModel("Claude 3.5 Sonnet", "claude-3.5-sonnet", Icons.Anthropic, INDIVIDUAL),