feat: provide origin question and returning invalid sql when adjusting

wren-ui/src/apollo/client/graphql/__types__.ts

@@ -32,6 +32,7 @@ export type AdjustThreadResponseInput = {
 export type AdjustmentTask = {
   __typename?: 'AdjustmentTask';
   error?: Maybe<Error>;
+  invalidSql?: Maybe<Scalars['String']>;
   queryId?: Maybe<Scalars['String']>;
   sql?: Maybe<Scalars['String']>;
   status?: Maybe<AskingTaskStatus>;

wren-ui/src/apollo/client/graphql/home.ts

@@ -125,6 +125,7 @@ const COMMON_RESPONSE = gql`
       }
       sql
       traceId
+      invalidSql
     }
   }
 
@@ -400,6 +401,7 @@ export const ADJUSTMENT_TASK = gql`
       }
       sql
       traceId
+      invalidSql
     }
   }
 `;

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

@@ -662,13 +662,13 @@ export class WrenAIAdaptor implements IWrenAIAdaptor {
   ): Promise<AsyncQueryResponse> {
     try {
       const body = {
+        question: input.question,
         tables: input.tables,
         sql_generation_reasoning: input.sqlGenerationReasoning,
         sql: input.sql,
         project_id: input.projectId.toString(),
         configurations: input.configurations,
       };
-
       const res = await axios.post(
         `${this.wrenAIBaseEndpoint}/v1/ask-feedbacks`,
         body,
@@ -725,6 +725,7 @@ export class WrenAIAdaptor implements IWrenAIAdaptor {
           type: result.type?.toUpperCase() as AskCandidateType,
         })) || [],
       traceId: body.trace_id,
+      invalidSql: body.invalid_sql,
     };
   }
 

wren-ui/src/apollo/server/backgrounds/adjustmentBackgroundTracker.ts

@@ -208,6 +208,7 @@ export class AdjustmentBackgroundTaskTracker
       tables: adjustment.payload?.retrievedTables,
       sqlGenerationReasoning: adjustment.payload?.sqlGenerationReasoning,
       sql: originalThreadResponse.sql,
+      question: originalThreadResponse.question,
     });
     const queryId = response.queryId;
 

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

@@ -294,6 +294,7 @@ export interface InstructionResult {
 
 // ask feedback
 export interface AskFeedbackInput {
+  question: string;
   tables: string[];
   sqlGenerationReasoning: string;
   sql: string;
@@ -318,4 +319,5 @@ export interface AskFeedbackResult {
     sql: string;
   }>;
   traceId?: string;
+  invalidSql?: string;
 }

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

@@ -46,6 +46,7 @@ export interface AdjustmentTask {
   error: WrenAIError | null;
   sql: string;
   traceId: string;
+  invalidSql?: string;
 }
 
 export interface AskingTask {
@@ -554,6 +555,7 @@ export class AskingResolver {
       error: adjustmentTask?.error,
       sql: adjustmentTask?.response?.[0]?.sql,
       traceId: adjustmentTask?.traceId,
+      invalidSql: adjustmentTask?.invalidSql,
     };
   }
 
@@ -742,6 +744,7 @@ export class AskingResolver {
         error: adjustmentTask?.error,
         sql: adjustmentTask?.response?.[0]?.sql,
         traceId: adjustmentTask?.traceId,
+        invalidSql: adjustmentTask?.invalidSql,
       };
     },
   });

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

@@ -729,6 +729,7 @@ export const typeDefs = gql`
     error: Error
     sql: String
     traceId: String
+    invalidSql: String
   }
 
   type ThreadResponse {