Merge pull request #26543 from brave/leo-search-p3a

Add SERP entry point to Leo metrics
brave · Nov 15, 2024 · 7097006 · 7097006
2 parents 77b46de + cdd9854
commit 7097006
Show file tree

Hide file tree

Showing 4 changed files with 21 additions and 5 deletions.
diff --git a/components/ai_chat/core/browser/ai_chat_metrics.cc b/components/ai_chat/core/browser/ai_chat_metrics.cc
@@ -67,6 +67,7 @@ constexpr char kContextMenuEntryPointKey[] = "context_menu";
 constexpr char kToolbarButtonEntryPointKey[] = "toolbar_button";
 constexpr char kMenuItemEntryPointKey[] = "menu_item";
 constexpr char kOmniboxCommandEntryPointKey[] = "omnibox_command";
+constexpr char kBraveSearchEntryPointKey[] = "brave_search";
 
 constexpr auto kContextMenuActionKeys =
     base::MakeFixedFlatMap<ContextMenuAction, const char*>(
@@ -86,7 +87,8 @@ constexpr auto kEntryPointKeys =
          {EntryPoint::kContextMenu, kContextMenuEntryPointKey},
          {EntryPoint::kToolbarButton, kToolbarButtonEntryPointKey},
          {EntryPoint::kMenuItem, kMenuItemEntryPointKey},
-         {EntryPoint::kOmniboxCommand, kOmniboxCommandEntryPointKey}});
+         {EntryPoint::kOmniboxCommand, kOmniboxCommandEntryPointKey},
+         {EntryPoint::kBraveSearch, kBraveSearchEntryPointKey}});
 
 #endif  // !BUILDFLAG(IS_ANDROID) && !BUILDFLAG(IS_IOS)
 
@@ -234,7 +236,7 @@ void AIChatMetrics::RecordReset() {
   UMA_HISTOGRAM_EXACT_LINEAR(kEnabledHistogramName,
                              std::numeric_limits<int>::max() - 1, 3);
   UMA_HISTOGRAM_EXACT_LINEAR(kAcquisitionSourceHistogramName,
-                             std::numeric_limits<int>::max() - 1, 6);
+                             std::numeric_limits<int>::max() - 1, 7);
 }
 
 void AIChatMetrics::OnPremiumStatusUpdated(bool is_new_user,

diff --git a/components/ai_chat/core/browser/ai_chat_metrics.h b/components/ai_chat/core/browser/ai_chat_metrics.h
@@ -75,7 +75,8 @@ enum class EntryPoint {
   kToolbarButton = 3,
   kMenuItem = 4,
   kOmniboxCommand = 5,
-  kMaxValue = kOmniboxCommand
+  kBraveSearch = 6,
+  kMaxValue = kBraveSearch
 };
 
 enum class ContextMenuAction {

diff --git a/components/ai_chat/core/browser/ai_chat_metrics_unittest.cc b/components/ai_chat/core/browser/ai_chat_metrics_unittest.cc
@@ -350,11 +350,19 @@ TEST_F(AIChatMetricsUnitTest, MostUsedEntryPoint) {
       kMostUsedEntryPointHistogramName,
       static_cast<int>(EntryPoint::kToolbarButton), 1);
 
+  for (size_t i = 0; i < 4; i++) {
+    ai_chat_metrics_->HandleOpenViaEntryPoint(EntryPoint::kBraveSearch);
+  }
+
+  histogram_tester_.ExpectBucketCount(
+      kMostUsedEntryPointHistogramName,
+      static_cast<int>(EntryPoint::kBraveSearch), 1);
+
   task_environment_.FastForwardBy(base::Days(7));
-  histogram_tester_.ExpectTotalCount(kMostUsedEntryPointHistogramName, 10);
+  histogram_tester_.ExpectTotalCount(kMostUsedEntryPointHistogramName, 14);
 
   task_environment_.FastForwardBy(base::Days(7));
-  histogram_tester_.ExpectTotalCount(kMostUsedEntryPointHistogramName, 10);
+  histogram_tester_.ExpectTotalCount(kMostUsedEntryPointHistogramName, 14);
 }
 
 #endif  // !BUILDFLAG(IS_ANDROID) && !BUILDFLAG(IS_IOS)

diff --git a/components/ai_chat/core/browser/ai_chat_service.cc b/components/ai_chat/core/browser/ai_chat_service.cc
@@ -446,6 +446,11 @@ void AIChatService::OpenConversationWithStagedEntries(
       associated_content->GetContentId(), associated_content);
   CHECK(conversation);
 
+#if !BUILDFLAG(IS_ANDROID) && !BUILDFLAG(IS_IOS)
+  if (ai_chat_metrics_ != nullptr) {
+    ai_chat_metrics_->HandleOpenViaEntryPoint(EntryPoint::kBraveSearch);
+  }
+#endif
   // Open AI Chat and trigger a fetch of staged conversations from Brave Search.
   std::move(open_ai_chat).Run();
   conversation->MaybeFetchOrClearContentStagedConversation();