Add connector and scope information to Obs AI Assistant EBT events (elastic#234550)

Closes elastic#232546
## Summary

Add connector and scope information to Obs AI Assistant EBT events

### Checklist

Check the PR satisfies following conditions.

Reviewers should verify this PR satisfies this list as well.

- [ ] Any text added follows [EUI's writing
guidelines](https://elastic.github.io/eui/#/guidelines/writing), uses
sentence case text and includes [i18n
support](https://github.com/elastic/kibana/blob/main/src/platform/packages/shared/kbn-i18n/README.md)
- [ ]
[Documentation](https://www.elastic.co/guide/en/kibana/master/development-documentation.html)
was added for features that require explanation or tutorials
- [ ] [Unit or functional
tests](https://www.elastic.co/guide/en/kibana/master/development-tests.html)
were updated or added to match the most common scenarios
- [ ] If a plugin configuration key changed, check if it needs to be
allowlisted in the cloud and added to the [docker
list](https://github.com/elastic/kibana/blob/main/src/dev/build/tasks/os_packages/docker_generator/resources/base/bin/kibana-docker)
- [ ] This was checked for breaking HTTP API changes, and any breaking
changes have been approved by the breaking-change committee. The
`release_note:breaking` label should be applied in these situations.
- [ ] [Flaky Test
Runner](https://ci-stats.kibana.dev/trigger_flaky_test_runner/1) was
used on any tests changed
- [ ] The PR description includes the appropriate Release Notes section,
and the correct `release_note:*` label is applied per the
[guidelines](https://www.elastic.co/guide/en/kibana/master/contributing.html#kibana-release-notes-process)
- [ ] Review the [backport
guidelines](https://docs.google.com/document/d/1VyN5k91e5OVumlc0Gb9RPa3h1ewuPE705nRtioPiTvY/edit?usp=sharing)
and apply applicable `backport:*` labels.

### Identify risks

Does this PR introduce any risks? For example, consider risks like hard
to test bugs, performance regression, potential of data loss.

Describe the risk, its severity, and mitigation for each identified
risk. Invite stakeholders and evaluate how to proceed before merging.

- [ ] [See some risk
examples](https://github.com/elastic/kibana/blob/main/RISK_MATRIX.mdx)
- [ ] ...

(cherry picked from commit b3a75dc)

# Conflicts:
#	packages/kbn-optimizer/limits.yml
#	x-pack/platform/packages/shared/kbn-ai-assistant/src/chat/chat_body.tsx
#	x-pack/platform/packages/shared/kbn-ai-assistant/src/chat/chat_header.test.tsx
#	x-pack/platform/packages/shared/kbn-ai-assistant/src/chat/welcome_message.test.tsx
#	x-pack/platform/plugins/shared/observability_ai_assistant/public/analytics/schemas/user_sent_prompt.ts
#	x-pack/platform/plugins/shared/observability_ai_assistant/server/analytics/recall_ranking.ts
#	x-pack/platform/plugins/shared/observability_ai_assistant/server/functions/context.ts
#	x-pack/platform/plugins/shared/observability_ai_assistant/server/functions/context/utils/recall_and_score.test.ts
#	x-pack/platform/plugins/shared/observability_ai_assistant/server/functions/context/utils/recall_and_score.ts
#	x-pack/platform/plugins/shared/observability_ai_assistant/server/routes/chat/route.ts

Author: arturoliduena

x-pack/platform/packages/shared/kbn-ai-assistant/src/chat/chat_body.tsx

@@ -50,6 +50,7 @@ import { useLicense } from '../hooks/use_license';
 import { PromptEditor } from '../prompt_editor/prompt_editor';
 import { deserializeMessage } from '../utils/deserialize_message';
 import { useKibana } from '../hooks/use_kibana';
+import { useScopes } from '../hooks';
 
 const fullHeightClassName = css`
   height: 100%;
@@ -145,6 +146,8 @@ export function ChatBody({
     false
   );
 
+  const scopes = useScopes();
+
   const { conversation, messages, next, state, stop, saveTitle } = useConversation({
     initialConversationId,
     initialMessages,
@@ -216,11 +219,15 @@ export function ChatBody({
         conversation: { id, last_updated: lastUpdated, token_count: tokenCount },
       };
 
+      const connector = connectors.getConnector(connectors.selectedConnector || '');
+
       chatService.sendAnalyticsEvent({
         type: ObservabilityAIAssistantTelemetryEventType.ChatFeedback,
         payload: {
           feedback,
           conversation: conversationWithoutMessagesAndTitle,
+          connector,
+          scopes,
         },
       });
     }

x-pack/platform/packages/shared/kbn-ai-assistant/src/chat/chat_header.test.tsx

@@ -74,6 +74,7 @@ describe('ChatHeader', () => {
           error: undefined,
           selectConnector: (id: string) => {},
           reloadConnectors: () => {},
+          getConnector: () => undefined,
         }}
       />
     );
@@ -96,6 +97,7 @@ describe('ChatHeader', () => {
           error: undefined,
           selectConnector: (id: string) => {},
           reloadConnectors: () => {},
+          getConnector: () => undefined,
         }}
       />
     );

x-pack/platform/packages/shared/kbn-ai-assistant/src/hooks/__storybook_mocks__/use_genai_connectors.ts

@@ -15,5 +15,6 @@ export function useGenAIConnectors(): UseGenAIConnectorsResult {
     selectedConnector: 'foo',
     selectConnector: (id: string) => {},
     reloadConnectors: () => {},
+    getConnector: () => undefined,
   };
 }

x-pack/platform/packages/shared/kbn-ai-assistant/src/prompt_editor/prompt_editor.tsx

@@ -19,6 +19,7 @@ import { FunctionListPopover } from '../chat/function_list_popover';
 import { PromptEditorFunction } from './prompt_editor_function';
 import { PromptEditorNaturalLanguage } from './prompt_editor_natural_language';
 import { useScopes } from '../hooks/use_scopes';
+import { useGenAIConnectors } from '../hooks';
 
 export interface PromptEditorProps {
   disabled: boolean;
@@ -44,6 +45,7 @@ export function PromptEditor({
   onSubmit,
 }: PromptEditorProps) {
   const scopes = useScopes();
+  const connectors = useGenAIConnectors();
   const containerRef = useRef<HTMLDivElement>(null);
 
   const [mode, setMode] = useState<'prompt' | 'function'>(
@@ -123,15 +125,19 @@ export function PromptEditor({
 
       setInnerMessage(undefined);
       setMode('prompt');
+      const connector = connectors.getConnector(connectors.selectedConnector || '');
       onSendTelemetry({
         type: ObservabilityAIAssistantTelemetryEventType.UserSentPromptInChat,
-        payload: { scopes },
+        payload: {
+          scopes,
+          connector,
+        },
       });
     } catch (_) {
       setInnerMessage(oldMessage);
       setMode(oldMessage.function_call?.name ? 'function' : 'prompt');
     }
-  }, [addLastUsedPrompt, innerMessage, loading, onSendTelemetry, onSubmit, scopes]);
+  }, [addLastUsedPrompt, innerMessage, loading, onSendTelemetry, onSubmit, scopes, connectors]);
 
   // Submit on Enter
   useEffect(() => {

x-pack/platform/plugins/shared/actions/server/index.ts

@@ -25,7 +25,10 @@ export type {
   ActionsApiRequestHandlerContext,
 } from './types';
 
-export type { ConnectorWithExtraFindData as FindActionResult } from './application/connector/types';
+export type {
+  ConnectorWithExtraFindData as FindActionResult,
+  Connector,
+} from './application/connector/types';
 
 export type { PluginSetupContract, PluginStartContract } from './plugin';
 

x-pack/platform/plugins/shared/observability_ai_assistant/common/analytics/index.ts

@@ -0,0 +1,9 @@
+/*
+ * 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.
+ */
+
+export { type Connector, connectorSchema } from './schemas/connector';
+export { type Scope, scopeSchema } from './schemas/scope';

x-pack/platform/plugins/shared/observability_ai_assistant/common/analytics/schemas/connector.ts

@@ -0,0 +1,53 @@
+/*
+ * 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 { RootSchema } from '@kbn/core/public';
+import type { InferenceConnector } from '../../utils/get_inference_connector';
+
+export interface Connector {
+  connector: InferenceConnector | undefined;
+}
+
+export const connectorSchema: RootSchema<Connector['connector']> = {
+  connectorId: {
+    type: 'text',
+    _meta: {
+      description: 'The id of the connector.',
+    },
+  },
+  name: {
+    type: 'text',
+    _meta: {
+      description: 'The name of the connector.',
+    },
+  },
+  type: {
+    type: 'text',
+    _meta: {
+      description: 'The action type id of the connector.',
+    },
+  },
+  modelFamily: {
+    type: 'text',
+    _meta: {
+      description: 'The model family of the connector.',
+    },
+  },
+  modelProvider: {
+    type: 'text',
+    _meta: {
+      description: 'The model provider of the connector.',
+    },
+  },
+  modelId: {
+    type: 'text',
+    _meta: {
+      description: 'The model id of the connector, if applicable.',
+      optional: true,
+    },
+  },
+};

x-pack/platform/plugins/shared/observability_ai_assistant/common/analytics/schemas/scope.ts

@@ -0,0 +1,23 @@
+/*
+ * 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 { SchemaValue } from '@kbn/core/public';
+import type { AssistantScope } from '@kbn/ai-assistant-common';
+
+export interface Scope {
+  scopes: AssistantScope[];
+}
+
+export const scopeSchema: SchemaValue<AssistantScope[]> = {
+  type: 'array',
+  items: {
+    type: 'text',
+    _meta: {
+      description: 'Scope of the AI Assistant',
+    },
+  },
+};

x-pack/platform/plugins/shared/observability_ai_assistant/common/utils/get_inference_connector.ts

@@ -0,0 +1,46 @@
+/*
+ * 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 { Connector } from '@kbn/actions-plugin/server';
+import {
+  getConnectorProvider,
+  getConnectorFamily,
+  getConnectorModel,
+  connectorToInference,
+  type InferenceConnectorType,
+  type ModelFamily,
+  type ModelProvider,
+} from '@kbn/inference-common';
+
+export interface InferenceConnector {
+  connectorId: string;
+  name: string;
+  type: InferenceConnectorType;
+  modelFamily: ModelFamily;
+  modelProvider: ModelProvider;
+  modelId: string | undefined;
+}
+
+export const getInferenceConnectorInfo = (
+  connector?: Connector
+): InferenceConnector | undefined => {
+  if (!connector) {
+    return;
+  }
+  const inferenceConnector = connectorToInference(connector);
+  const modelFamily = getConnectorFamily(inferenceConnector);
+  const modelProvider = getConnectorProvider(inferenceConnector);
+  const modelId = getConnectorModel(inferenceConnector);
+  return {
+    connectorId: inferenceConnector.connectorId,
+    name: inferenceConnector.name,
+    type: inferenceConnector.type,
+    modelFamily,
+    modelProvider,
+    modelId,
+  };
+};

x-pack/platform/plugins/shared/observability_ai_assistant/public/analytics/schemas/chat_feedback.ts

@@ -9,8 +9,14 @@ import type { EventTypeOpts } from '@kbn/core/public';
 import type { Conversation } from '../../../common';
 import type { Feedback } from '../../components/buttons/feedback_buttons';
 import { ObservabilityAIAssistantTelemetryEventType } from '../telemetry_event_type';
+import {
+  type Connector,
+  type Scope,
+  connectorSchema,
+  scopeSchema,
+} from '../../../common/analytics';
 
-export interface ChatFeedback {
+export interface ChatFeedback extends Connector, Scope {
   feedback: Feedback;
   conversation: Omit<Omit<Conversation, 'messages'>, 'conversation'> & {
     conversation: Omit<Conversation['conversation'], 'title'>;
@@ -115,5 +121,9 @@ export const chatFeedbackEventSchema: EventTypeOpts<ChatFeedback> = {
         },
       },
     },
+    connector: {
+      properties: connectorSchema,
+    },
+    scopes: scopeSchema,
   },
 };