Add GPT4-O

superagent-ai · May 14, 2024 · 734e662 · 734e662
1 parent 439741b
commit 734e662
Show file tree

Hide file tree

Showing 6 changed files with 10 additions and 1 deletion.
diff --git a/libs/superagent/app/tools/advanced_scraper.py b/libs/superagent/app/tools/advanced_scraper.py
@@ -21,7 +21,6 @@ def _run(self, url: str) -> str:
  "waitBeforeScraping": 1,
  "fastLane": True,
  "removeCSSselectors": "default",
- "triggersDownload": True,
  "timeout": 45,
  }
 

diff --git a/libs/superagent/app/utils/llm.py b/libs/superagent/app/utils/llm.py
@@ -15,6 +15,7 @@
  "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",
+ "GPT_4_O": "gpt-4o",
 }
 
 LLM_REVERSE_MAPPING = {v: k for k, v in LLM_MAPPING.items()}
@@ -36,6 +37,7 @@
  "GPT_4_TURBO_2024_04_09",
  "GPT_4_0125_PREVIEW",
  "GPT_4_32K",
+ "GPT_4_O",
  ]
 }
 

diff --git a/libs/superagent/prisma/migrations/20240514064040_gpt_4_o/migration.sql b/libs/superagent/prisma/migrations/20240514064040_gpt_4_o/migration.sql
@@ -0,0 +1,2 @@
+-- AlterEnum
+ALTER TYPE "LLMModel" ADD VALUE 'GPT_4_0';
diff --git a/libs/superagent/prisma/schema.prisma b/libs/superagent/prisma/schema.prisma
@@ -45,6 +45,7 @@ enum LLMModel {
  GPT_4_TURBO
  GPT_4_TURBO_PREVIEW
  GPT_4_TURBO_2024_04_09
+ GPT_4_0
  MISTRAL_7B_INSTRUCT_V01
  MIXTRAL_8X7B_INSTRUCT_V01
 }

diff --git a/libs/ui/config/site.ts b/libs/ui/config/site.ts
@@ -130,6 +130,10 @@ export const siteConfig = {
  value: "GPT_4_TURBO_PREVIEW",
  title: "gpt-4-turbo-preview",
  },
+ {
+ value: "GPT_4_O",
+ title: "gpt-4o",
+ },
  ],
  },
  {

diff --git a/libs/ui/models/models.ts b/libs/ui/models/models.ts
@@ -17,6 +17,7 @@ export enum LLMModel {
  GPT_4_0613,
  GPT_4_32K_0613,
  GPT_4_1106_PREVIEW,
+ GPT_4_O,
  MISTRAL_7B_INSTRUCT_V01,
 }