从零搭建基于 AI GitHub 分析平台 (17) 保存答案

添加 saveAnswer 函数

修改 src/server/api/routers/project.ts:

saveAnswer: authenticatedProcedure.input( z.object({
    projectId: z.string(),
    question: z.string(),
    answer: z.string(),
    fileReferences: z.any(),
  })).mutation( async ( { ctx, input }) => {
    return await ctx.db.question.create({
      data: {
        answer: input.answer,
        fileReferences: input.fileReferences,
        projectId: input.projectId,
        question: input.question,
        userId: ctx.user.userId!,
      }
    })
  })

修改 AskQuestionCard 页面

修改 `src/app/(protected)/dashboard/AskQuestionCard.tsx’

import { api } from '@/trpc/react';
import { toast } from 'sonner';
...
const [ fileReferences, setFileReferences ] = useState<{ fileName :string, sourceCode: string, summary:string }[]>([])
const saveAnswer = api.project.saveAnswer.useMutation();
...
const onSubmit = async ( e: React.FormEvent<HTMLFormElement>) => {
    setAnswer("");
    setFileReferences([]);
    e.preventDefault();
    if(!project?.id) return;
    setLoading(true);
    
    const {output , fileReferences} = await askQuestion(question,project.id);
    setFileReferences(fileReferences);
    setAnswer(output);

  }
...
    <div className='flex items-center gap-2'>
      <DialogTitle>
        <Image src='/logo.png' alt='aigithub' width={40} height={40} />
      </DialogTitle>
      <Button disabled={saveAnswer.isPending} variant={'outline'} onClick={()=>{
        saveAnswer.mutate({
          projectId: project!.id,
          question,
          answer,
          fileReferences
        }, {
          onSuccess: ()=>{
            toast.success('答案保存成功')
          },
          onError: () => {
            toast.error('答案保存错误');
          }
        })
      }}>
        保存答案
      </Button>
    </div>  

---

作者：Bearalise
出处：从零搭建基于 AI GitHub 分析平台 (17) 保存答案
版权：本文版权归作者所有
转载：欢迎转载，但未经作者同意，必须保留此段声明，必须在文章中给出原文链接。