[Obs AI Assistant][ES|QL] Add 10s request timeout to ES|QL query execution (#238200)

Closes #234135

## Summary

Add 10s request timeout to ES|QL query execution

---------

Co-authored-by: kibanamachine <42973632+kibanamachine@users.noreply.github.com>
(cherry picked from commit 0f4c777)

# Conflicts:
#	x-pack/solutions/observability/plugins/observability_ai_assistant_app/server/functions/query/index.ts
#	x-pack/solutions/observability/plugins/observability_ai_assistant_app/tsconfig.json

Author: arturoliduena

x-pack/solutions/observability/plugins/observability_ai_assistant_app/server/functions/query/index.test.ts

@@ -0,0 +1,80 @@
+/*
+ * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
+ * or more contributor license agreements. Licensed under the Elastic License
+ * 2.0; you may not use this file except in compliance with the Elastic License
+ * 2.0.
+ */
+
+import type { ElasticsearchClient, Logger } from '@kbn/core/server';
+import type { DeeplyMockedKeys } from '@kbn/utility-types-jest';
+import { ChatFunctionClient } from '@kbn/observability-ai-assistant-plugin/server/service/chat_function_client';
+import { registerExecuteQueryFunction } from '.';
+import { EXECUTE_QUERY_FUNCTION_NAME } from '@kbn/observability-ai-assistant-plugin/server';
+
+describe('executeQuery function', () => {
+  const currentUserEsClientMock: DeeplyMockedKeys<ElasticsearchClient> = {
+    search: jest.fn(),
+    fieldCaps: jest.fn(),
+    esql: {
+      query: () => Promise.resolve({ columns: [], values: [] }),
+    },
+  } as any;
+
+  const consoleOrPassThrough = () => {};
+
+  const loggerMock: DeeplyMockedKeys<Logger> = {
+    log: jest.fn().mockImplementation(consoleOrPassThrough),
+    error: jest.fn().mockImplementation(consoleOrPassThrough),
+    debug: jest.fn().mockImplementation(consoleOrPassThrough),
+    trace: jest.fn().mockImplementation(consoleOrPassThrough),
+    isLevelEnabled: jest.fn().mockReturnValue(true),
+  } as any;
+
+  let result: any;
+  const timeoutError = new Error('Request timed out');
+  beforeEach(async () => {
+    jest.clearAllMocks();
+    const functionClient = new ChatFunctionClient([]);
+
+    // mock the esql query to throw a timeout error
+    currentUserEsClientMock.esql.query = jest.fn().mockImplementation(() => {
+      const error = timeoutError;
+      error.name = 'TimeoutError';
+      throw error;
+    });
+
+    registerExecuteQueryFunction({
+      functions: functionClient,
+      resources: {
+        context: {
+          // @ts-expect-error Type mismatch is expected in test mocks
+          core: Promise.resolve({
+            elasticsearch: { client: { asCurrentUser: currentUserEsClientMock } },
+          }),
+        },
+      },
+      signal: new AbortController().signal,
+    });
+
+    result = await functionClient.executeFunction({
+      chat: jest.fn(),
+      name: EXECUTE_QUERY_FUNCTION_NAME,
+      args: JSON.stringify({ query: 'FROM logs-* | LIMIT 10' }),
+      messages: [],
+      signal: new AbortController().signal,
+      logger: loggerMock,
+      connectorId: 'foo',
+      simulateFunctionCalling: false,
+    });
+  });
+
+  it('return an error message when the execute_query function throws a timeout', () => {
+    expect(result).toEqual({
+      content: {
+        message: 'The query failed to execute',
+        error: timeoutError,
+        errorMessages: undefined,
+      },
+    });
+  });
+});

x-pack/solutions/observability/plugins/observability_ai_assistant_app/server/functions/query/index.ts

@@ -24,12 +24,11 @@ import { runAndValidateEsqlQuery } from './validate_esql_query';
 export const QUERY_FUNCTION_NAME = 'query';
 export const EXECUTE_QUERY_NAME = 'execute_query';
 
-export function registerQueryFunction({
+export const registerExecuteQueryFunction = ({
   functions,
   resources,
-  pluginsStart,
   signal,
-}: FunctionRegistrationParameters) {
+}: FunctionRegistrationParameters) => {
   functions.registerInstruction(({ availableFunctionNames }) => {
     if (!availableFunctionNames.includes(QUERY_FUNCTION_NAME)) {
       return;
@@ -102,6 +101,11 @@ export function registerQueryFunction({
       };
     }
   );
+};
+
+export function registerQueryFunction(params: FunctionRegistrationParameters) {
+  const { functions, resources, pluginsStart } = params;
+  registerExecuteQueryFunction(params);
   functions.registerFunction(
     {
       name: QUERY_FUNCTION_NAME,

x-pack/solutions/observability/plugins/observability_ai_assistant_app/server/functions/query/validate_esql_query.ts

@@ -50,7 +50,10 @@ export async function runAndValidateEsqlQuery({
   });
 
   try {
-    const res = await client.esql.query({ query, drop_null_columns: true }, { signal });
+    const res = await client.esql.query(
+      { query, drop_null_columns: true },
+      { signal, requestTimeout: '10s' }
+    );
     const esqlResponse = res as unknown as ESQLSearchResponse;
 
     const columns =