fix kimi

app/api/ask-ai/route.ts

@@ -223,7 +223,8 @@ export async function PUT(request: NextRequest) {
   const userToken = request.cookies.get(MY_TOKEN_KEY())?.value;
 
   const body = await request.json();
-  const { prompt, html, previousPrompt, provider, selectedElementHtml } = body;
+  const { prompt, html, previousPrompt, provider, selectedElementHtml, model } =
+    body;
 
   if (!prompt || !html) {
     return NextResponse.json(
@@ -232,7 +233,15 @@ export async function PUT(request: NextRequest) {
     );
   }
 
-  const selectedModel = MODELS[0];
+  const selectedModel = MODELS.find(
+    (m) => m.value === model || m.label === model
+  );
+  if (!selectedModel) {
+    return NextResponse.json(
+      { ok: false, error: "Invalid model selected" },
+      { status: 400 }
+    );
+  }
 
   let token = userToken;
   let billTo: string | null = null;

components/editor/ask-ai/index.tsx

@@ -67,6 +67,10 @@ export function AskAI({
   const [controller, setController] = useState<AbortController | null>(null);
   const [isFollowUp, setIsFollowUp] = useState(true);
 
+  const selectedModel = useMemo(() => {
+    return MODELS.find((m: { value: string }) => m.value === model);
+  }, [model]);
+
   const callAi = async (redesignMarkdown?: string) => {
     if (isAiWorking) return;
     if (!redesignMarkdown && !prompt.trim()) return;
@@ -178,7 +182,9 @@ export function AskAI({
               setPrompt("");
               setisAiWorking(false);
               setHasAsked(true);
-              setModel(MODELS[0].value);
+              if (selectedModel?.isThinker) {
+                setModel(MODELS[0].value);
+              }
               if (audio.current) audio.current.play();
 
               // Now we have the complete HTML including </html>, so set it to be sure
@@ -444,7 +450,7 @@ export function AskAI({
                 id="diff-patch-checkbox"
                 checked={isFollowUp}
                 onCheckedChange={(e) => {
-                  if (e === true && !isSameHtml) {
+                  if (e === true && !isSameHtml && selectedModel?.isThinker) {
                     setModel(MODELS[0].value);
                   }
                   setIsFollowUp(e === true);