From 21759d04c57fccdc7d4822ffb90d1ed0ef188a32 Mon Sep 17 00:00:00 2001
From: davitcoin <88998061+Davitcoin@users.noreply.github.com>
Date: Thu, 10 Oct 2024 11:50:36 -0600
Subject: [PATCH] Added new Gemini-exp-0827 and Gpt4omini

New models added
Image input enabled, be careful with api usage, you can enable it or disable  it at lib/models/lim/{Aimodel}
---
 lib/models/llm/google-llm-list.ts | 21 +++++++++++++++++----
 lib/models/llm/openai-llm-list.ts | 22 ++++++++++++++++++++--
 types/llms.ts                     |  3 +++
 3 files changed, 40 insertions(+), 6 deletions(-)

diff --git a/lib/models/llm/google-llm-list.ts b/lib/models/llm/google-llm-list.ts
index d039eda4af..cd49597b6c 100644
--- a/lib/models/llm/google-llm-list.ts
+++ b/lib/models/llm/google-llm-list.ts
@@ -2,7 +2,7 @@ import { LLM } from "@/types"
 
 const GOOGLE_PLATORM_LINK = "https://ai.google.dev/"
 
-// Google Models (UPDATED 12/22/23) -----------------------------
+// Google Models (UPDATED 9/09/24) -----------------------------
 
 // Gemini 1.5 Flash
 const GEMINI_1_5_FLASH: LLM = {
@@ -31,9 +31,16 @@ const GEMINI_PRO: LLM = {
   provider: "google",
   hostedId: "gemini-pro",
   platformLink: GOOGLE_PLATORM_LINK,
-  imageInput: false
+  imageInput: true
+}
+const GEMINI_1_5_PRO_EXP0827: LLM = {
+  modelId: "gemini-1.5-pro-exp-0827",
+  modelName: "Gemini 1.5 Pro Experimental 0827",
+  provider: "google",
+  hostedId: "gemini-1.5-pro-exp-0827",
+  platformLink: GOOGLE_PLATORM_LINK,
+  imageInput: true
 }
-
 // Gemini Pro Vision (UPDATED 12/22/23)
 const GEMINI_PRO_VISION: LLM = {
   modelId: "gemini-pro-vision",
@@ -44,4 +51,10 @@ const GEMINI_PRO_VISION: LLM = {
   imageInput: true
 }
 
-export const GOOGLE_LLM_LIST: LLM[] = [GEMINI_PRO, GEMINI_PRO_VISION, GEMINI_1_5_PRO, GEMINI_1_5_FLASH]
+export const GOOGLE_LLM_LIST: LLM[] = [
+  GEMINI_PRO,
+  GEMINI_1_5_PRO_EXP0827,
+  GEMINI_PRO_VISION,
+  GEMINI_1_5_PRO,
+  GEMINI_1_5_FLASH
+]
diff --git a/lib/models/llm/openai-llm-list.ts b/lib/models/llm/openai-llm-list.ts
index 48823ab981..b83224128e 100644
--- a/lib/models/llm/openai-llm-list.ts
+++ b/lib/models/llm/openai-llm-list.ts
@@ -2,7 +2,23 @@ import { LLM } from "@/types"
 
 const OPENAI_PLATORM_LINK = "https://platform.openai.com/docs/overview"
 
-// OpenAI Models (UPDATED 1/25/24) -----------------------------
+// OpenAI Models (UPDATED 10/09/24) -----------------------------
+
+const GPT4oMini: LLM = {
+  modelId: "gpt-4o-mini",
+  modelName: "GPT-4o Mini",
+  provider: "openai",
+  hostedId: "gpt-4o-mini",
+  platformLink: OPENAI_PLATORM_LINK,
+  imageInput: true,
+  pricing: {
+    currency: "USD",
+    unit: "1M tokens",
+    inputCost: 0.15,
+    outputCost: 0.6
+  }
+}
+
 const GPT4o: LLM = {
   modelId: "gpt-4o",
   modelName: "GPT-4o",
@@ -72,7 +88,7 @@ const GPT3_5Turbo: LLM = {
   provider: "openai",
   hostedId: "gpt-3.5-turbo",
   platformLink: OPENAI_PLATORM_LINK,
-  imageInput: false,
+  imageInput: true,
   pricing: {
     currency: "USD",
     unit: "1M tokens",
@@ -86,5 +102,7 @@ export const OPENAI_LLM_LIST: LLM[] = [
   GPT4Turbo,
   GPT4Vision,
   GPT4,
+  GPT4oMini,
   GPT3_5Turbo
+
 ]
diff --git a/types/llms.ts b/types/llms.ts
index 6f89acf612..9e5070b808 100644
--- a/types/llms.ts
+++ b/types/llms.ts
@@ -10,6 +10,7 @@ export type LLMID =
 
 // OpenAI Models (UPDATED 5/13/24)
 export type OpenAILLMID =
+  |"gpt-4o-mini" // GPT-4o Mini
   | "gpt-4o" // GPT-4o
   | "gpt-4-turbo-preview" // GPT-4 Turbo
   | "gpt-4-vision-preview" // GPT-4 Vision
@@ -22,6 +23,8 @@ export type GoogleLLMID =
   | "gemini-pro-vision" // Gemini Pro Vision
   | "gemini-1.5-pro-latest" // Gemini 1.5 Pro
   | "gemini-1.5-flash" // Gemini 1.5 Flash
+  | "gemini-1.5-pro-exp-0827" // Gemini 1.5 Pro Exp 0827
+
 
 // Anthropic Models
 export type AnthropicLLMID =