From a7456cd22866703bbdeacf82f99756cec4723636 Mon Sep 17 00:00:00 2001 From: fly6516 Date: Fri, 16 May 2025 00:56:35 +0800 Subject: [PATCH] feat: init ai-improve.ts --- bun.lock | 11 ++++++- package.json | 5 ++- src/actions/ai-improve.ts | 64 +++++++++++++++++++++++++++++++++++++++ 3 files changed, 78 insertions(+), 2 deletions(-) create mode 100644 src/actions/ai-improve.ts diff --git a/bun.lock b/bun.lock index 2071c9d..65edb31 100644 --- a/bun.lock +++ b/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=="], diff --git a/package.json b/package.json index 1ccafd4..7334b93 100644 --- a/package.json +++ b/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", diff --git a/src/actions/ai-improve.ts b/src/actions/ai-improve.ts new file mode 100644 index 0000000..7e65716 --- /dev/null +++ b/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 => { + 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; +} \ No newline at end of file