feat(wren-ui): pass invalid sql from AI service

wren-ui/src/apollo/server/adaptors/wrenAIAdaptor.ts

@@ -507,7 +507,7 @@ export class WrenAIAdaptor implements IWrenAIAdaptor {
   }
 
   private transformAskResult(body: any): AskResult {
-    const { type, intent_reasoning } = body;
+    const { type, intent_reasoning, invalid_sql } = body;
     const { status, error } = this.transformStatusAndError(body);
     const candidates = (body?.response || []).map((candidate: any) => ({
       type: candidate?.type?.toUpperCase() as AskCandidateType,
@@ -521,6 +521,7 @@ export class WrenAIAdaptor implements IWrenAIAdaptor {
       error,
       response: candidates,
       intentReasoning: intent_reasoning,
+      invalidSql: invalid_sql,
     };
   }
 

wren-ui/src/apollo/server/models/adaptor.ts

@@ -97,6 +97,7 @@ export interface AskResponse<R, S> {
   status: S;
   response: R | null;
   error: WrenAIError | null;
+  invalid_sql?: string;
 }
 
 export interface AskDetailInput {
@@ -122,6 +123,7 @@ export type AskResult = AskResponse<
   AskResultStatus
 > & {
   intentReasoning?: string;
+  invalidSql?: string;
 };
 
 export enum RecommendationQuestionStatus {

wren-ui/src/apollo/server/resolvers/askingResolver.ts

@@ -45,6 +45,7 @@ export interface AskingTask {
   }>;
   error: WrenAIError | null;
   intentReasoning?: string;
+  invalidSql?: string;
 }
 
 // DetailedThread is a type that represents a detailed thread, which is a thread with responses.
@@ -205,7 +206,6 @@ export class AskingResolver {
         false,
       );
     }
-
     // construct candidates from response
     const candidates = await Promise.all(
       (askResult.response || []).map(async (response) => {
@@ -219,14 +219,15 @@ export class AskingResolver {
         };
       }),
     );
-
-    return {
+    const res = {
       type: askResult.type,
       status: askResult.status,
       error: askResult.error,
       candidates,
       intentReasoning: askResult.intentReasoning,
+      invalidSql: askResult.invalid_sql,
     };
+    return res;
   }
 
   public async createThread(

wren-ui/src/apollo/server/schema.ts

@@ -585,6 +585,7 @@ export const typeDefs = gql`
     error: Error
     candidates: [ResultCandidate!]!
     intentReasoning: String
+    invalidSql: String
   }
 
   input InstantRecommendedQuestionsInput {