[Obs AI Assistant][Bug] Display results and Visualize query actions on an ES|QL query doesn't work with Bedrock

When an ES|QL is generated, we present 2 options
- `Visualize Query`
- `Display Results`

None of these options work as expected with Bedrock and it throws a 400 error. Sometimes it displays the result, but still throws an error.

Error:
```
Encountered error in Bedrock stream of type validationException messages.8: Did not find 1 `tool_result` block(s) at the beginning of this message. Messages following `tool_use` blocks must begin with a matching number of `tool_result` blocks.
```

Possible cause:
Sending a tool call in an assistant message without the corresponding tool response in the next message

https://github.com/user-attachments/assets/26a78370-6bcf-46ab-aea3-bcbea5ea54c1

LoC: https://github.com/elastic/kibana/blob/323cbdb9cdfed351445a9e24523b78b8fd23dbb5/x-pack/platform/packages/shared/kbn-ai-assistant/src/chat/chat_body.tsx#L288-L305

	case ChatActionClickType.executeEsqlQuery:
	next(
	messages.concat({
	'@timestamp': new Date().toISOString(),
	message: {
	role: MessageRole.Assistant,
	content: '',
	function_call: {
	name: 'execute_query',
	arguments: JSON.stringify({
	query: payload.query,
	}),
	trigger: MessageRole.User,
	},
	},
	})
	);
	break;

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

[Obs AI Assistant][Bug] Display results and Visualize query actions on an ES|QL query doesn't work with Bedrock #214005

Metadata

Assignees

Labels

Type

Fields

Projects

Milestone

Relationships

Development

[Obs AI Assistant][Bug] Display results and Visualize query actions on an ES|QL query doesn't work with Bedrock #214005

Description

Metadata

Metadata

Assignees

Labels

Type

Fields

Projects

Milestone

Relationships

Development

Issue actions