[8.17] Fix calling Sonnet 3.7 (#212372) (#212411)

# Backport

This will backport the following commits from `main` to `8.17`:
- [Fix calling Sonnet 3.7
(#212372)](https://github.com/elastic/kibana/pull/212372)

<!--- Backport version: 9.6.6 -->

### Questions ?
Please refer to the [Backport tool
documentation](https://github.com/sorenlouv/backport)

<!--BACKPORT [{"author":{"name":"Patryk
Kopyciński","email":"contact@patrykkopycinski.com"},"sourceCommit":{"committedDate":"2025-02-25T15:41:55Z","message":"Fix
calling Sonnet 3.7 (#212372)\n\n## Summary\n\nFixes \n<img
width=\"1509\"
alt=\"image\"\nsrc=\"https://github.com/user-attachments/assets/ba409fde-f52b-4864-a81a-70025ef61f2f\"\n/>","sha":"8c456d1e1eb172a2550455caa42cc503cf589e27","branchLabelMapping":{"^v9.1.0$":"main","^v8.19.0$":"8.x","^v(\\d+).(\\d+).\\d+$":"$1.$2"}},"sourcePullRequest":{"labels":["bug","release_note:skip","v9.0.0","backport:version","v8.18.0","v9.1.0","v8.17.3"],"title":"Fix
calling Sonnet
3.7","number":212372,"url":"https://github.com/elastic/kibana/pull/212372","mergeCommit":{"message":"Fix
calling Sonnet 3.7 (#212372)\n\n## Summary\n\nFixes \n<img
width=\"1509\"
alt=\"image\"\nsrc=\"https://github.com/user-attachments/assets/ba409fde-f52b-4864-a81a-70025ef61f2f\"\n/>","sha":"8c456d1e1eb172a2550455caa42cc503cf589e27"}},"sourceBranch":"main","suggestedTargetBranches":["8.17"],"targetPullRequestStates":[{"branch":"9.0","label":"v9.0.0","branchLabelMappingKey":"^v(\\d+).(\\d+).\\d+$","isSourceBranch":false,"url":"https://github.com/elastic/kibana/pull/212399","number":212399,"state":"OPEN"},{"branch":"8.18","label":"v8.18.0","branchLabelMappingKey":"^v(\\d+).(\\d+).\\d+$","isSourceBranch":false,"url":"https://github.com/elastic/kibana/pull/212398","number":212398,"state":"OPEN"},{"branch":"main","label":"v9.1.0","branchLabelMappingKey":"^v9.1.0$","isSourceBranch":true,"state":"MERGED","url":"https://github.com/elastic/kibana/pull/212372","number":212372,"mergeCommit":{"message":"Fix
calling Sonnet 3.7 (#212372)\n\n## Summary\n\nFixes \n<img
width=\"1509\"
alt=\"image\"\nsrc=\"https://github.com/user-attachments/assets/ba409fde-f52b-4864-a81a-70025ef61f2f\"\n/>","sha":"8c456d1e1eb172a2550455caa42cc503cf589e27"}},{"branch":"8.17","label":"v8.17.3","branchLabelMappingKey":"^v(\\d+).(\\d+).\\d+$","isSourceBranch":false,"state":"NOT_CREATED"}]}]
BACKPORT-->
This commit is contained in:
Patryk Kopyciński 2025-02-25 19:32:48 +01:00 committed by GitHub
parent 13cfa3ef48
commit 0ccc106cac
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -50,6 +50,18 @@ export const BedrockToolChoiceSchema = schema.object({
name: schema.maybe(schema.string()),
});
export const BedrockUsageSchema = schema.maybe(
schema.object(
{
input_tokens: schema.number(),
output_tokens: schema.number(),
// added with Sonnet 3.7
cache_creation_input_tokens: schema.maybe(schema.number()),
},
{ unknowns: 'allow' }
)
);
export const InvokeAIActionParamsSchema = schema.object({
messages: schema.arrayOf(BedrockMessageSchema),
model: schema.maybe(schema.string()),
@ -75,6 +87,7 @@ export const InvokeAIActionParamsSchema = schema.object({
export const InvokeAIActionResponseSchema = schema.object({
message: schema.string(),
usage: BedrockUsageSchema,
});
export const InvokeAIRawActionParamsSchema = schema.object({
@ -110,10 +123,7 @@ export const InvokeAIRawActionResponseSchema = schema.object({}, { unknowns: 'al
export const RunApiLatestResponseSchema = schema.object(
{
stop_reason: schema.maybe(schema.string()),
usage: schema.object({
input_tokens: schema.number(),
output_tokens: schema.number(),
}),
usage: BedrockUsageSchema,
content: schema.arrayOf(
schema.object(
{ type: schema.string(), text: schema.maybe(schema.string()) },
@ -128,12 +138,7 @@ export const RunActionResponseSchema = schema.object(
{
completion: schema.string(),
stop_reason: schema.maybe(schema.string()),
usage: schema.maybe(
schema.object({
input_tokens: schema.number(),
output_tokens: schema.number(),
})
),
usage: BedrockUsageSchema,
},
{ unknowns: 'ignore' }
);