ai

udecode · Oct 25, 2024 · 5b97b2d · 5b97b2d
1 parent 9468f73
commit 5b97b2d
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 5 deletions.
diff --git a/templates/plate-playground-template/src/app/api/ai/copilot/route.ts b/templates/plate-playground-template/src/app/api/ai/copilot/route.ts
@@ -5,7 +5,12 @@ import { generateText } from 'ai';
 import type { NextRequest } from 'next/server';
 
 export async function POST(req: NextRequest) {
-  const { prompt, system, apiKey: key } = await req.json();
+  const {
+    prompt,
+    system,
+    apiKey: key,
+    model = 'gpt-4o-mini',
+  } = await req.json();
 
   const apiKey = key || process.env.OPENAI_API_KEY;
 
@@ -16,15 +21,13 @@ export async function POST(req: NextRequest) {
     );
   }
 
-  console.log(apiKey);
-
   const openai = createOpenAI({ apiKey });
 
   try {
     const result = await generateText({
       abortSignal: req.signal,
       maxTokens: 50,
-      model: openai('gpt-4o-mini'),
+      model: openai(model),
       prompt: prompt,
       system,
       temperature: 0.7,

diff --git a/templates/plate-playground-template/src/components/openai/settings-dialog.tsx b/templates/plate-playground-template/src/components/openai/settings-dialog.tsx
@@ -61,6 +61,7 @@ export function SettingsDialog() {
       body: {
         ...completeOptions.body,
         apiKey: tempKey,
+        model: model.value,
       },
     });
   };
@@ -184,7 +185,7 @@ export function SettingsDialog() {
           </Popover>
 
           <Button type="submit" className="w-full">
-            Set API Key
+            Save
           </Button>
         </form>
         <p className="mt-4 text-sm text-muted-foreground">