From b238253935990d2f88b106c2c68463b7bf622c43 Mon Sep 17 00:00:00 2001
From: Carl Brenner <carl@ceacle.com>
Date: Thu, 31 Oct 2024 12:01:31 +0100
Subject: [PATCH] Hide google and groq models

---
 packages/lib/constants.ts | 96 +++++++++++++++++++--------------------
 packages/lib/types.ts     |  2 +-
 2 files changed, 49 insertions(+), 49 deletions(-)

diff --git a/packages/lib/constants.ts b/packages/lib/constants.ts
index adf79bf..22e157c 100644
--- a/packages/lib/constants.ts
+++ b/packages/lib/constants.ts
@@ -34,54 +34,54 @@ export const MODELS: Models = {
       // },
     },
   },
-  /**
-   * https://ai.google.dev/pricing#1_5flash
-   */
-  google: {
-    name: 'Google',
-    list: {
-      'gemini-1.5-pro-002': {
-        cost: { input: 1.25, output: 5.00 },
-        /**
-         * Prompts longer than 128k
-         * cost: { input: 2.50, output: 10.0 },
-         */
-        label: 'Gemini 1.5 Pro',
-      },
-      'gemini-1.5-flash-002': {
-        cost: { input: 0.075, output: 0.30 },
-        /**
-         * Prompts longer than 128k
-         * cost: { input: 0.15, output: 0.60 },
-         */
-        label: 'Gemini 1.5 Flash',
-      },
-    },
-  },
-  /**
-   * https://console.groq.com/docs/models
-   */
-  groq: {
-    name: 'Groq',
-    list: {
-      'llama-3.1-70b-versatile': {
-        cost: { input: 0.59, output: 0.79 },
-        label: 'Llama 3.1 70B',
-      },
-      'llama-3.1-8b-instant': {
-        cost: { input: 0.05, output: 0.08 },
-        label: 'Llama 3.1 8B',
-      },
-      // 'llama3-groq-70b-8192-tool-use-preview': {
-      //   cost: { input: 0.89, output: 0.89 },
-      //   label: 'Llama 3 Groq 70B Tool Use Preview 8k',
-      // },
-      // 'llama3-groq-8b-8192-tool-use-preview': {
-      //   cost: { input: 0.19, output: 0.19 },
-      //   label: 'Llama 3.1 8B',
-      // },
-    },
-  },
+  // /**
+  //  * https://ai.google.dev/pricing#1_5flash
+  //  */
+  // google: {
+  //   name: 'Google',
+  //   list: {
+  //     'gemini-1.5-pro-002': {
+  //       cost: { input: 1.25, output: 5.00 },
+  //       /**
+  //        * Prompts longer than 128k
+  //        * cost: { input: 2.50, output: 10.0 },
+  //        */
+  //       label: 'Gemini 1.5 Pro',
+  //     },
+  //     'gemini-1.5-flash-002': {
+  //       cost: { input: 0.075, output: 0.30 },
+  //       /**
+  //        * Prompts longer than 128k
+  //        * cost: { input: 0.15, output: 0.60 },
+  //        */
+  //       label: 'Gemini 1.5 Flash',
+  //     },
+  //   },
+  // },
+  // /**
+  //  * https://console.groq.com/docs/models
+  //  */
+  // groq: {
+  //   name: 'Groq',
+  //   list: {
+  //     'llama-3.1-70b-versatile': {
+  //       cost: { input: 0.59, output: 0.79 },
+  //       label: 'Llama 3.1 70B',
+  //     },
+  //     'llama-3.1-8b-instant': {
+  //       cost: { input: 0.05, output: 0.08 },
+  //       label: 'Llama 3.1 8B',
+  //     },
+  //     // 'llama3-groq-70b-8192-tool-use-preview': {
+  //     //   cost: { input: 0.89, output: 0.89 },
+  //     //   label: 'Llama 3 Groq 70B Tool Use Preview 8k',
+  //     // },
+  //     // 'llama3-groq-8b-8192-tool-use-preview': {
+  //     //   cost: { input: 0.19, output: 0.19 },
+  //     //   label: 'Llama 3.1 8B',
+  //     // },
+  //   },
+  // },
   /**
    * https://openai.com/pricing
    */
diff --git a/packages/lib/types.ts b/packages/lib/types.ts
index 2b7e038..5576e60 100644
--- a/packages/lib/types.ts
+++ b/packages/lib/types.ts
@@ -66,7 +66,7 @@ export type MessageRole = 'system' | 'user'
 export type MessageStatus = 'pending' | 'done' | 'error'
 export type MessageType = 'error' | 'prompt' | 'response'
 
-export type ModelVendor = 'anthropic' | 'google' | 'groq' | 'openai'
+export type ModelVendor = 'anthropic' /* | 'google' | 'groq' */ | 'openai'
 
 export type Models = {
   [key in ModelVendor]: {