fix fields

pingcap · Aug 21, 2024 · a83220e · a83220e
1 parent d700a8e
commit a83220e
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 7 deletions.
diff --git a/frontend/app/src/experimental/chat-verify-service/api.ts b/frontend/app/src/experimental/chat-verify-service/api.ts
@@ -16,18 +16,19 @@ 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(),
   }).or(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
@@ -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>
   );
-}
+}