From 7a490ba6a7f640b8dd945399131ef4b206b88346 Mon Sep 17 00:00:00 2001
From: Kibana Machine <42973632+kibanamachine@users.noreply.github.com>
Date: Thu, 14 Sep 2023 16:20:05 -0400
Subject: [PATCH] [8.10] [Security solution] Another GenAI response schema
 update (#166495) (#166498)

# Backport

This will backport the following commits from `main` to `8.10`:
- [[Security solution] Another GenAI response schema update
(#166495)](https://github.com/elastic/kibana/pull/166495)

<!--- Backport version: 8.9.7 -->

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

<!--BACKPORT [{"author":{"name":"Steph
Milovic","email":"stephanie.milovic@elastic.co"},"sourceCommit":{"committedDate":"2023-09-14T19:06:11Z","message":"[Security
solution] Another GenAI response schema update (#166495)\n\n##
Summary\r\n\r\nFollow up from
https://github.com/elastic/kibana/pull/166300\r\n\r\nWe are only using
`messages` from `choices`, so I made the other\r\n`choices` properties
`maybe` as
well","sha":"6e4ecff7c8776e6d446ba9d8b784cee3d9cf1d3f","branchLabelMapping":{"^v8.11.0$":"main","^v(\\d+).(\\d+).\\d+$":"$1.$2"}},"sourcePullRequest":{"labels":["release_note:skip","Team:Threat
Hunting","Team:
SecuritySolution","v8.11.0","v8.10.1"],"number":166495,"url":"https://github.com/elastic/kibana/pull/166495","mergeCommit":{"message":"[Security
solution] Another GenAI response schema update (#166495)\n\n##
Summary\r\n\r\nFollow up from
https://github.com/elastic/kibana/pull/166300\r\n\r\nWe are only using
`messages` from `choices`, so I made the other\r\n`choices` properties
`maybe` as
well","sha":"6e4ecff7c8776e6d446ba9d8b784cee3d9cf1d3f"}},"sourceBranch":"main","suggestedTargetBranches":["8.10"],"targetPullRequestStates":[{"branch":"main","label":"v8.11.0","labelRegex":"^v8.11.0$","isSourceBranch":true,"state":"MERGED","url":"https://github.com/elastic/kibana/pull/166495","number":166495,"mergeCommit":{"message":"[Security
solution] Another GenAI response schema update (#166495)\n\n##
Summary\r\n\r\nFollow up from
https://github.com/elastic/kibana/pull/166300\r\n\r\nWe are only using
`messages` from `choices`, so I made the other\r\n`choices` properties
`maybe` as
well","sha":"6e4ecff7c8776e6d446ba9d8b784cee3d9cf1d3f"}},{"branch":"8.10","label":"v8.10.1","labelRegex":"^v(\\d+).(\\d+).\\d+$","isSourceBranch":false,"state":"NOT_CREATED"}]}]
BACKPORT-->

Co-authored-by: Steph Milovic <stephanie.milovic@elastic.co>
---
 x-pack/plugins/stack_connectors/common/gen_ai/schema.ts | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/x-pack/plugins/stack_connectors/common/gen_ai/schema.ts b/x-pack/plugins/stack_connectors/common/gen_ai/schema.ts
index 860976834076b..f82b89ee7c2b6 100644
--- a/x-pack/plugins/stack_connectors/common/gen_ai/schema.ts
+++ b/x-pack/plugins/stack_connectors/common/gen_ai/schema.ts
@@ -60,8 +60,8 @@ export const GenAiRunActionResponseSchema = schema.object(
             },
             { unknowns: 'ignore' }
           ),
-          finish_reason: schema.string(),
-          index: schema.number(),
+          finish_reason: schema.maybe(schema.string()),
+          index: schema.maybe(schema.number()),
         },
         { unknowns: 'ignore' }
       )