diff --git a/src/actions/ai-improve.ts b/src/actions/ai-improve.ts
index ab36e07..84c1bfb 100644
--- a/src/actions/ai-improve.ts
+++ b/src/actions/ai-improve.ts
@@ -7,7 +7,8 @@ import {
 } from "@/types/ai-improve";
 import { openai } from "@/lib/ai";
 import { CoreMessage, generateText } from "ai";
-import { prisma } from "@/lib/prisma"; // Prisma客户端
+import { PrismaClient } from '@/generated/client';
+const prisma = new PrismaClient();
 
 /**
  * 调用AI优化代码
diff --git a/src/app/(app)/problems/[id]/page.tsx b/src/app/(app)/problems/[id]/page.tsx
index 901c6fc..81343f9 100644
--- a/src/app/(app)/problems/[id]/page.tsx
+++ b/src/app/(app)/problems/[id]/page.tsx
@@ -11,6 +11,7 @@ import {
 import { Locale } from "@/config/i18n";
 import { useEffect, useState } from "react";
 import { useTranslations } from "next-intl";
+import { usePathname } from "next/navigation";
 import Dockview from "@/components/dockview";
 import { useDockviewStore } from "@/stores/dockview";
 import { AIProblemEditor } from "@/components/ai-optimized-editor";
diff --git a/src/app/(app)/test/page.tsx b/src/app/(app)/test/page.tsx
new file mode 100644
index 0000000..b8087a4
--- /dev/null
+++ b/src/app/(app)/test/page.tsx
@@ -0,0 +1,10 @@
+import { AIProblemEditor } from '@/components/ai-optimized-editor';
+
+export default function TestAiEditorPage() {
+    return (
+        <div className="container mx-auto p-4">
+            <h1 className="text-2xl font-bold mb-4">AI 编辑器测试页</h1>
+            <AIProblemEditor />
+        </div>
+    );
+}
\ No newline at end of file