Merge pull request #16 from pipecat-ai/cst/gemini

Add Google Gemini LLM options
pipecat-ai · Nov 18, 2024 · 96b5947 · 96b5947
2 parents 4d2d8f3 + 8cca23d
commit 96b5947
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 2 deletions.
diff --git a/clients/web/app/workspaces/[workspaceId]/APIKeysSection.tsx b/clients/web/app/workspaces/[workspaceId]/APIKeysSection.tsx
@@ -37,7 +37,7 @@ export default function APIKeysSection({ formState, setFormState }: Props) {
   const config = {
     daily: {
       key: dailyKey,
-      required: true,
+      required: false,
     },
     [llmProvider]: {
       key: llmKey,

diff --git a/clients/web/app/workspaces/[workspaceId]/ConfigurationSection.tsx b/clients/web/app/workspaces/[workspaceId]/ConfigurationSection.tsx
@@ -102,6 +102,9 @@ export default function ConfigurationSection({
             <ToggleGroupItem id="openai" value="openai" aria-label="OpenAI">
               OpenAI
             </ToggleGroupItem>
+            <ToggleGroupItem id="google" value="google" aria-label="Google">
+              Google
+            </ToggleGroupItem>
           </ToggleGroup>
         </ConfigurationItem>
       </fieldset>

diff --git a/clients/web/lib/llm.ts b/clients/web/lib/llm.ts
@@ -1,4 +1,4 @@
-export type LLMProvider = "anthropic" | "together" | "groq" | "openai";
+export type LLMProvider = "anthropic" | "together" | "groq" | "openai" | "google";
 
 export type LLMMessageRole = "system" | "user" | "assistant";
 
@@ -68,6 +68,17 @@ const llmModels: LLMModel[] = [
     model: "gpt-4o-mini",
     label: "gpt-4o mini",
   },
+  // Google
+  {
+    service: "google",
+    model: "gemini-1.5-flash",
+    label: "Gemini 1.5 Flash"
+  },
+  {
+    service: "google",
+    model: "gemini-1.5-pro",
+    label: "Gemini 1.5 Pro"
+  },
 ];
 
 export const defaultModel = llmModels[0];