diff --git a/libs/superagent/app/utils/llm.py b/libs/superagent/app/utils/llm.py index e4deb74a8..e277f22dc 100644 --- a/libs/superagent/app/utils/llm.py +++ b/libs/superagent/app/utils/llm.py @@ -4,11 +4,17 @@ "GPT_3_5_TURBO_16K_0613": "gpt-3.5-turbo-16k-0613", "GPT_3_5_TURBO_0613": "gpt-3.5-turbo-0613", "GPT_3_5_TURBO_1106": "gpt-3.5-turbo-1106", + "GPT_3_5_TURBO": "gpt-3.5-turbo", "GPT_4_0613": "gpt-4-0613", "GPT_4_32K_0613": "gpt-4-32k-0613", + "GPT_4_32K": "gpt-4-32k", "GPT_4_1106_PREVIEW": "gpt-4-1106-preview", "GPT_4_TURBO_PREVIEW": "gpt-4-turbo-preview", "GPT_3_5_TURBO_0125": "gpt-3.5-turbo-0125", + "GPT_4": "gpt-4", + "GPT_4_TURBO": "gpt-4-turbo", + "GPT_4_TURBO_2024_04_09": "gpt-4-turbo-2024-04-09", + "GPT_4_0125_PREVIEW": "gpt-4-0125-preview", } LLM_REVERSE_MAPPING = {v: k for k, v in LLM_MAPPING.items()} @@ -16,6 +22,7 @@ LLM_PROVIDER_MAPPING = { "OPENAI": [ + "GPT_3_5_TURBO", "GPT_3_5_TURBO_16K_0613", "GPT_3_5_TURBO_0613", "GPT_3_5_TURBO_1106", @@ -24,6 +31,11 @@ "GPT_4_32K_0613", "GPT_4_1106_PREVIEW", "GPT_4_TURBO_PREVIEW", + "GPT_4", + "GPT_4_TURBO", + "GPT_4_TURBO_2024_04_09", + "GPT_4_0125_PREVIEW", + "GPT_4_32K", ] } diff --git a/libs/superagent/prisma/migrations/20240503060243_update_gpt_4_models/migration.sql b/libs/superagent/prisma/migrations/20240503060243_update_gpt_4_models/migration.sql new file mode 100644 index 000000000..c6d5951fa --- /dev/null +++ b/libs/superagent/prisma/migrations/20240503060243_update_gpt_4_models/migration.sql @@ -0,0 +1,14 @@ +-- AlterEnum +-- This migration adds more than one value to an enum. +-- With PostgreSQL versions 11 and earlier, this is not possible +-- in a single migration. This can be worked around by creating +-- multiple migrations, each migration adding only one value to +-- the enum. + + +ALTER TYPE "LLMModel" ADD VALUE 'GPT_3_5_TURBO'; +ALTER TYPE "LLMModel" ADD VALUE 'GPT_4'; +ALTER TYPE "LLMModel" ADD VALUE 'GPT_4_32K'; +ALTER TYPE "LLMModel" ADD VALUE 'GPT_4_0125_PREVIEW'; +ALTER TYPE "LLMModel" ADD VALUE 'GPT_4_TURBO'; +ALTER TYPE "LLMModel" ADD VALUE 'GPT_4_TURBO_2024_04_09'; diff --git a/libs/superagent/prisma/schema.prisma b/libs/superagent/prisma/schema.prisma index d10bb6620..65d5d151a 100644 --- a/libs/superagent/prisma/schema.prisma +++ b/libs/superagent/prisma/schema.prisma @@ -30,14 +30,21 @@ enum LLMProvider { } enum LLMModel { + GPT_3_5_TURBO GPT_3_5_TURBO_16K_0613 GPT_3_5_TURBO_0613 GPT_3_5_TURBO_1106 GPT_3_5_TURBO_0125 + + GPT_4 GPT_4_0613 + GPT_4_32K GPT_4_32K_0613 GPT_4_1106_PREVIEW + GPT_4_0125_PREVIEW + GPT_4_TURBO GPT_4_TURBO_PREVIEW + GPT_4_TURBO_2024_04_09 MISTRAL_7B_INSTRUCT_V01 MIXTRAL_8X7B_INSTRUCT_V01 }