chore(compass-query-bar): update ai backend response parsing COMPASS-6986 (#4601)

Author: Anemy

packages/compass-query-bar/src/modules/ai-query-request.spec.ts

@@ -43,9 +43,11 @@ describe('#runFetchAIQuery', function () {
       expect(requests[0].req.url).to.equal('/ai/api/v1/mql-query');
 
       expect(response).to.deep.equal({
-        query: {
-          find: {
-            test: 'pineapple',
+        content: {
+          query: {
+            find: {
+              test: 'pineapple',
+            },
           },
         },
       });

packages/compass-query-bar/src/stores/ai-query-reducer.ts

@@ -106,13 +106,13 @@ export const runAIQuery = (
 
     let query;
     try {
-      if (!jsonResponse?.query) {
+      if (!jsonResponse?.content?.query) {
         throw new Error(
           'No query returned. Please try again with a different prompt.'
         );
       }
 
-      query = jsonResponse.query;
+      query = jsonResponse?.content?.query;
     } catch (err: any) {
       dispatch({
         type: AIQueryActionTypes.AIQueryFailed,

packages/compass-query-bar/test/create-mock-ai-endpoint.ts

@@ -11,9 +11,11 @@ export async function startMockAIServer(
     sendError?: boolean;
   } = {
     response: {
-      query: {
-        find: {
-          test: 'pineapple',
+      content: {
+        query: {
+          find: {
+            test: 'pineapple',
+          },
         },
       },
     },