feat: init ai-improve.ts

bun.lock

@@ -31,16 +31,18 @@
         "@radix-ui/react-tooltip": "^1.1.8",
         "@tanstack/react-table": "^8.21.2",
         "@types/vscode": "^1.97.0",
-        "ai": "^4.2.0",
+        "ai": "^4.3.15",
         "bcrypt": "^5.1.1",
         "class-variance-authority": "^0.7.1",
         "clsx": "^2.1.1",
+        "complexity": "^0.0.6",
         "date-fns": "^4.1.0",
         "devicons-react": "^1.4.0",
         "dockerode": "^4.0.4",
         "dockview": "^4.2.1",
         "framer-motion": "^12.7.3",
         "github-markdown-css": "^5.8.1",
+        "lib": "^5.1.0",
         "lucide-react": "^0.482.0",
         "monaco-editor": "<=0.36.1",
         "monaco-languageclient": "<=5.0.1",
@@ -64,6 +66,7 @@
         "tailwind-merge": "^3.0.1",
         "tailwindcss-animate": "^1.0.7",
         "tar-stream": "^3.1.7",
+        "types": "^0.1.1",
         "uuid": "^11.1.0",
         "vscode-languageclient": "^9.0.1",
         "vscode-ws-jsonrpc": "^3.4.0",
@@ -732,6 +735,8 @@
 
     "commander": ["commander@4.1.1", "https://registry.npmmirror.com/commander/-/commander-4.1.1.tgz", {}, "sha512-NOKm8xhkzAjzFx8B2v5OAHT+u5pRQc2UCa2Vq9jYL/31o2wi9mxBA7LIFs3sV5VSC49z6pEhfbMULvShKj26WA=="],
 
+    "complexity": ["complexity@0.0.6", "", {}, "sha512-5NzE1NMBLxj1OraF+A2revTYzuVR+/nqISyYP3CT5HwVxaWhjvaRc5BDc1OsJc66WeS1sre3PxTyDQfxMPh5Vg=="],
+
     "concat-map": ["concat-map@0.0.1", "https://registry.npmmirror.com/concat-map/-/concat-map-0.0.1.tgz", {}, "sha512-/Srv4dswyQNBfohGpz9o6Yb3Gz3SrUDqBH5rTuhGR7ahtlbYKnVxw2bCFMRljaA7EXHaXZ8wsHdodFvbkhKmqg=="],
 
     "console-control-strings": ["console-control-strings@1.1.0", "https://registry.npmmirror.com/console-control-strings/-/console-control-strings-1.1.0.tgz", {}, "sha512-ty/fTekppD2fIwRvnZAVdeOiGd1c7YXEixbgJTNzqcxJWKQnjJ/V1bNEEE6hygpM3WjwHFUVK6HTjWSzV4a8sQ=="],
@@ -1180,6 +1185,8 @@
 
     "levn": ["levn@0.4.1", "https://registry.npmmirror.com/levn/-/levn-0.4.1.tgz", { "dependencies": { "prelude-ls": "^1.2.1", "type-check": "~0.4.0" } }, "sha512-+bT2uH4E5LGE7h/n3evcS/sQlJXCpIp6ym8OWJ5eV6+67Dsql/LaaT7qJBAt2rzfoa/5QBGBhxDix1dMt2kQKQ=="],
 
+    "lib": ["lib@5.1.0", "", {}, "sha512-RqaGSpLSf6vtSls86+kUqMuQkNiXLtUFbKrh/mKOuZmi3ENJOXHsScw8/grKOLzDgYTgOvpFPGgmNsEh5f2yuA=="],
+
     "lilconfig": ["lilconfig@3.1.3", "https://registry.npmmirror.com/lilconfig/-/lilconfig-3.1.3.tgz", {}, "sha512-/vlFKAoH5Cgt3Ie+JLhRbwOsCQePABiU3tJ1egGvyQ+33R/vcwM2Zl2QR/LzjsBeItPt3oSVXapn+m4nQDvpzw=="],
 
     "lines-and-columns": ["lines-and-columns@1.2.4", "https://registry.npmmirror.com/lines-and-columns/-/lines-and-columns-1.2.4.tgz", {}, "sha512-7ylylesZQ/PV29jhEDl3Ufjo6ZX7gCqJr5F7PKrqc93v7fzSymt1BpwEU8nAUXs8qzzvqhbjhK5QZg6Mt/HkBg=="],
@@ -1776,6 +1783,8 @@
 
     "typed-array-length": ["typed-array-length@1.0.7", "https://registry.npmmirror.com/typed-array-length/-/typed-array-length-1.0.7.tgz", { "dependencies": { "call-bind": "^1.0.7", "for-each": "^0.3.3", "gopd": "^1.0.1", "is-typed-array": "^1.1.13", "possible-typed-array-names": "^1.0.0", "reflect.getprototypeof": "^1.0.6" } }, "sha512-3KS2b+kL7fsuk/eJZ7EQdnEmQoaho/r6KUef7hxvltNA5DR8NAUM+8wJMbJyZ4G9/7i3v5zPBIMN5aybAh2/Jg=="],
 
+    "types": ["types@0.1.1", "", {}, "sha512-JuntZtJj4MKLE9x/XBs7IjsznYhzETwr34pw3XJTKvgYtAMdeMG+o8x8U85E5Lm6eCPa1DdOdGVsHMwq4ZnZAg=="],
+
     "typescript": ["typescript@5.8.3", "https://registry.npmmirror.com/typescript/-/typescript-5.8.3.tgz", { "bin": { "tsc": "bin/tsc", "tsserver": "bin/tsserver" } }, "sha512-p1diW6TqL9L07nNxvRMM7hMMw4c5XOo/1ibL4aAIGmSAt9slTE1Xgw5KWuof2uTOvCg9BY7ZRi+GaF+7sfgPeQ=="],
 
     "unbox-primitive": ["unbox-primitive@1.1.0", "https://registry.npmmirror.com/unbox-primitive/-/unbox-primitive-1.1.0.tgz", { "dependencies": { "call-bound": "^1.0.3", "has-bigints": "^1.0.2", "has-symbols": "^1.1.0", "which-boxed-primitive": "^1.1.1" } }, "sha512-nWJ91DjeOkej/TA8pXQ3myruKpKEYgqvpw9lz4OPHj/NWFNluYrjbz9j01CJ8yKQd2g4jFoOkINCTW2I5LEEyw=="],

package.json

@@ -40,16 +40,18 @@
     "@radix-ui/react-tooltip": "^1.1.8",
     "@tanstack/react-table": "^8.21.2",
     "@types/vscode": "^1.97.0",
-    "ai": "^4.2.0",
+    "ai": "^4.3.15",
     "bcrypt": "^5.1.1",
     "class-variance-authority": "^0.7.1",
     "clsx": "^2.1.1",
+    "complexity": "^0.0.6",
     "date-fns": "^4.1.0",
     "devicons-react": "^1.4.0",
     "dockerode": "^4.0.4",
     "dockview": "^4.2.1",
     "framer-motion": "^12.7.3",
     "github-markdown-css": "^5.8.1",
+    "lib": "^5.1.0",
     "lucide-react": "^0.482.0",
     "monaco-editor": "<=0.36.1",
     "monaco-languageclient": "<=5.0.1",
@@ -73,6 +75,7 @@
     "tailwind-merge": "^3.0.1",
     "tailwindcss-animate": "^1.0.7",
     "tar-stream": "^3.1.7",
+    "types": "^0.1.1",
     "uuid": "^11.1.0",
     "vscode-languageclient": "^9.0.1",
     "vscode-ws-jsonrpc": "^3.4.0",

src/actions/ai-improve.ts

@@ -0,0 +1,64 @@
+"use server";
+
+import {
+    AnalyzeComplexityResponse,
+    AnalyzeComplexityResponseSchema,
+    Complexity,
+} from "@/types/complexity";
+import { openai } from "@/lib/ai";
+import { CoreMessage, generateText } from "ai";
+
+export const analyzeComplexity = async (
+    content: string
+): Promise<AnalyzeComplexityResponse> => {
+    const model = openai("gpt-4o-mini");
+
+    const prompt = `
+  Analyze the time and space complexity of the following programming code snippet.
+  Determine the Big O notation from this list: ${Complexity.options.join(", ")}.
+  Provide your response as a JSON object with one key:
+  1. "time": A string representing the time complexity (e.g., "O(N)", "O(N^2)").
+  2. "space": A string representing the space complexity (e.g., "O(1)", "O(N)").
+
+  Code to analyze:
+  \`\`\`
+  ${content}
+  \`\`\`
+
+  Respond ONLY with the JSON object. Do not include any other text or markdown formatting like \`\`\`json before or after the object.
+  `;
+
+    const messages: CoreMessage[] = [{role: "user", content: prompt}];
+
+    let text;
+    try {
+        const response = await generateText({
+            model: model,
+            messages: messages,
+        });
+        text = response.text;
+    } catch (error) {
+        console.error("Error generating text with OpenAI:", error);
+        throw new Error("Failed to generate response from OpenAI");
+    }
+
+    let llmResponseJson;
+    try {
+        const cleanedText = text.trim();
+        llmResponseJson = JSON.parse(cleanedText);
+    } catch (error) {
+        console.error("Failed to parse LLM response as JSON:", error);
+        console.error("LLM raw output:", text);
+        throw new Error("Invalid JSON response from LLM");
+    }
+
+    const validationResult =
+        AnalyzeComplexityResponseSchema.safeParse(llmResponseJson);
+
+    if (!validationResult.success) {
+        console.error("Zod validation failed:", validationResult.error.format());
+        throw new Error("Response validation failed");
+    }
+
+    return validationResult;
+}