improve types

elastic · Jun 7, 2024 · aabe6f7 · aabe6f7
1 parent b0c8821
commit aabe6f7
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/x-pack/plugins/observability_solution/observability_ai_assistant/server/functions/context.ts b/x-pack/plugins/observability_solution/observability_ai_assistant/server/functions/context.ts
@@ -70,8 +70,10 @@ export function registerContextFunction({
           messages.filter((message) => message.message.role === MessageRole.User)
         );
 
-        const userPrompt = userMessage?.message.content!;
-        const queries = compact([{ text: userPrompt, boost: 3 }, { text: screenDescription }]);
+        const userPrompt = userMessage?.message.content;
+        const queries = [{ text: userPrompt, boost: 3 }, { text: screenDescription }].filter(
+          ({ text }) => text
+        ) as Array<{ text: string; boost?: number }>;
 
         const suggestions = await retrieveSuggestions({ client, queries });
         if (suggestions.length === 0) {
@@ -90,7 +92,7 @@ export function registerContextFunction({
           });
 
           analytics.reportEvent<RecallRanking>(RecallRankingEventType, {
-            prompt: `${screenDescription} | ${userPrompt}`,
+            prompt: queries.map((query) => query.text).join('|'),
             scoredDocuments: suggestions.map((suggestion) => {
               const llmScore = scores.find((score) => score.id === suggestion.id);
               return {