[AI Assistant] Add scopes telemetry to AI Assistant events (#197983)

## Summary

This adds telemetry to the Observability and Search assistant to judge
whether it's search or observability.

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

@@ -18,6 +18,7 @@ import { useLastUsedPrompts } from '../hooks/use_last_used_prompts';
 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';
 
 export interface PromptEditorProps {
   disabled: boolean;
@@ -42,6 +43,7 @@ export function PromptEditor({
   onSendTelemetry,
   onSubmit,
 }: PromptEditorProps) {
+  const scopes = useScopes();
   const containerRef = useRef<HTMLDivElement>(null);
 
   const [mode, setMode] = useState<'prompt' | 'function'>(
@@ -121,16 +123,15 @@ export function PromptEditor({
 
       setInnerMessage(undefined);
       setMode('prompt');
-
       onSendTelemetry({
         type: ObservabilityAIAssistantTelemetryEventType.UserSentPromptInChat,
-        payload: message,
+        payload: { ...message, scopes },
       });
     } catch (_) {
       setInnerMessage(oldMessage);
       setMode(oldMessage.function_call?.name ? 'function' : 'prompt');
     }
-  }, [addLastUsedPrompt, innerMessage, loading, onSendTelemetry, onSubmit]);
+  }, [addLastUsedPrompt, innerMessage, loading, onSendTelemetry, onSubmit, scopes]);
 
   // Submit on Enter
   useEffect(() => {

x-pack/plugins/observability_solution/observability_ai_assistant/public/analytics/index.ts

@@ -6,6 +6,7 @@
  */
 
 import type { AnalyticsServiceSetup, AnalyticsServiceStart } from '@kbn/core-analytics-browser';
+import { AssistantScope } from '@kbn/ai-assistant-common';
 import type { Message } from '../../common';
 import { chatFeedbackEventSchema, ChatFeedback } from './schemas/chat_feedback';
 import { insightFeedbackEventSchema, InsightFeedback } from './schemas/insight_feedback';
@@ -17,7 +18,10 @@ const schemas = [chatFeedbackEventSchema, insightFeedbackEventSchema, userSentPr
 export type TelemetryEventTypeWithPayload =
   | { type: ObservabilityAIAssistantTelemetryEventType.ChatFeedback; payload: ChatFeedback }
   | { type: ObservabilityAIAssistantTelemetryEventType.InsightFeedback; payload: InsightFeedback }
-  | { type: ObservabilityAIAssistantTelemetryEventType.UserSentPromptInChat; payload: Message };
+  | {
+      type: ObservabilityAIAssistantTelemetryEventType.UserSentPromptInChat;
+      payload: Message & { scopes: AssistantScope[] };
+    };
 
 export const registerTelemetryEventTypes = (analytics: AnalyticsServiceSetup) => {
   schemas.forEach((schema) => {

x-pack/plugins/observability_solution/observability_ai_assistant/public/analytics/schemas/common.ts

@@ -6,9 +6,10 @@
  */
 
 import type { RootSchema } from '@kbn/core/public';
+import { AssistantScope } from '@kbn/ai-assistant-common';
 import type { Message } from '../../../common';
 
-export const messageSchema: RootSchema<Message> = {
+export const messageSchema: RootSchema<Message & { scopes: AssistantScope[] }> = {
   '@timestamp': {
     type: 'text',
     _meta: {
@@ -74,4 +75,13 @@ export const messageSchema: RootSchema<Message> = {
       },
     },
   },
+  scopes: {
+    type: 'array',
+    items: {
+      type: 'text',
+      _meta: {
+        description: 'The scopes that were used when generating the message.',
+      },
+    },
+  },
 };