From a83220e23f1a0819e2eec04ca90d3fe911a05025 Mon Sep 17 00:00:00 2001
From: Jagger <634750802@qq.com>
Date: Wed, 21 Aug 2024 18:23:23 +0800
Subject: [PATCH] fix fields

---
 .../app/src/experimental/chat-verify-service/api.ts    |  5 +++--
 .../chat-verify-service/message-verify.tsx             | 10 +++++-----
 2 files changed, 8 insertions(+), 7 deletions(-)

diff --git a/frontend/app/src/experimental/chat-verify-service/api.ts b/frontend/app/src/experimental/chat-verify-service/api.ts
index 280d991f..37093dd3 100644
--- a/frontend/app/src/experimental/chat-verify-service/api.ts
+++ b/frontend/app/src/experimental/chat-verify-service/api.ts
@@ -16,10 +16,11 @@ const verifyResponse = z.object({
 });
 
 const getVerifyResponse = z.object({
-  state: z.enum([VerifyState.CREATED, VerifyState.EXTRACTING, VerifyState.VALIDATING, VerifyState.SUCCESS, VerifyState.FAILED]),
+  status: z.enum([VerifyState.CREATED, VerifyState.EXTRACTING, VerifyState.VALIDATING, VerifyState.SUCCESS, VerifyState.FAILED]),
   message: z.string().optional(),
   runs: z.object({
     sql: z.string(),
+    explanation: z.string(),
   }).and(z.object({
     success: z.literal(true),
     results: z.any().array().array(),
@@ -27,7 +28,7 @@ const getVerifyResponse = z.object({
     success: z.literal(false),
     sql_error_code: z.number(),
     sql_error_message: z.string(),
-    warnings: z.string(),
+    warnings: z.string().array(),
   }))).array().optional(),
 });
 
diff --git a/frontend/app/src/experimental/chat-verify-service/message-verify.tsx b/frontend/app/src/experimental/chat-verify-service/message-verify.tsx
index 9af6226c..7341e223 100644
--- a/frontend/app/src/experimental/chat-verify-service/message-verify.tsx
+++ b/frontend/app/src/experimental/chat-verify-service/message-verify.tsx
@@ -25,7 +25,7 @@ export function MessageVerify ({ user, assistant }: { user: ChatMessageControlle
 
   const shouldPoll = !!verifyId && !!assistant && isSuperuser;
   const { data: result, mutate } = useSWR(shouldPoll && `experimental.chat-message.${assistant.id}.verify`, () => getVerify(verifyId!), { revalidateOnMount: false, revalidateOnFocus: false, errorRetryCount: 0 });
-  const finished = result ? isFinalVerifyState(result.state) : false;
+  const finished = result ? isFinalVerifyState(result.status) : false;
 
   useEffect(() => {
     if (shouldPoll && !finished) {
@@ -53,18 +53,18 @@ export function MessageVerify ({ user, assistant }: { user: ChatMessageControlle
 
   return (
     <Alert
-      variant={result ? result.state === VerifyState.SUCCESS ? 'success' : result.state === VerifyState.FAILED ? 'destructive' : undefined : undefined}
+      variant={result ? result.status === VerifyState.SUCCESS ? 'success' : result.status === VerifyState.FAILED ? 'destructive' : undefined : undefined}
       className={cn('transition-opacity', isVerifying && 'opacity-50')}
     >
       {isVerifying
         ? <Loader2Icon className="animate-spin repeat-infinite" />
-        : result?.state === VerifyState.SUCCESS
+        : result?.status === VerifyState.SUCCESS
           ? <CheckCircle2Icon />
-          : result?.state === VerifyState.FAILED
+          : result?.status === VerifyState.FAILED
             ? <InformationCircleIcon />
             : undefined}
       <AlertTitle>Verify chat response</AlertTitle>
       <AlertDescription>{result?.message}</AlertDescription>
     </Alert>
   );
-}
\ No newline at end of file
+}