diff --git a/x-pack/plugins/observability_solution/observability_ai_assistant/server/service/client/index.test.ts b/x-pack/plugins/observability_solution/observability_ai_assistant/server/service/client/index.test.ts
index a94138d7a9241..89e7aa4cbb4de 100644
--- a/x-pack/plugins/observability_solution/observability_ai_assistant/server/service/client/index.test.ts
+++ b/x-pack/plugins/observability_solution/observability_ai_assistant/server/service/client/index.test.ts
@@ -63,6 +63,7 @@ function createLlmSimulator(subscriber: any) {
       subscriber.next({
         type: InferenceChatCompletionEventType.ChatCompletionMessage,
         content: msg.content,
+        toolCalls: msg.function_call ? [{ function: msg.function_call }] : [],
       });
     },
     tokenCount: async ({