From c8586c671cd3a42812d42bae17c0e6168e9e1b96 Mon Sep 17 00:00:00 2001
From: fly6516 <fly6516@outlook.com>
Date: Fri, 16 May 2025 10:14:28 +0800
Subject: [PATCH] =?UTF-8?q?refactor(actions):=20=E4=BC=98=E5=8C=96=20AI?=
 =?UTF-8?q?=E6=94=B9=E8=BF=9B=E4=BB=A3=E7=A0=81=E4=B8=AD=E7=9A=84=E6=95=B0?=
 =?UTF-8?q?=E6=8D=AE=E5=BA=93=E6=9F=A5=E8=AF=A2?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- 移除了不必要的 PrismaClient 实例化
- 更新了 problemId 字段的使用，使其与数据库结构保持一致
---
 src/actions/ai-improve.ts | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/src/actions/ai-improve.ts b/src/actions/ai-improve.ts
index 84c1bfb..eeb45c9 100644
--- a/src/actions/ai-improve.ts
+++ b/src/actions/ai-improve.ts
@@ -7,8 +7,7 @@ import {
 } from "@/types/ai-improve";
 import { openai } from "@/lib/ai";
 import { CoreMessage, generateText } from "ai";
-import { PrismaClient } from '@/generated/client';
-const prisma = new PrismaClient();
+import prisma from '@/generated/client';
 
 /**
  * 调用AI优化代码
@@ -25,7 +24,7 @@ export const optimizeCode = async (
     if (input.problemId) {
         try {
             const problem = await prisma.problem.findUnique({
-                where: { problemId: input.problemId },
+                where: { id: input.problemId },
             });
             if (problem) {
                 problemDetails = `