From 44e25c076997aa7ab4d4d0a180ebfb24f373ed8c Mon Sep 17 00:00:00 2001
From: Predrag Klepic <60631598+Klepic95@users.noreply.github.com>
Date: Thu, 27 Jul 2023 19:42:17 +0200
Subject: [PATCH] [Query Copilot] New Query button added on Items section
 (#1552)

Co-authored-by: Predrag Klepic <v-prklepic@microsoft.com>
---
 src/Explorer/Tree/SampleDataTree.tsx | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/Explorer/Tree/SampleDataTree.tsx b/src/Explorer/Tree/SampleDataTree.tsx
index cb824bbd5..6d521d7c4 100644
--- a/src/Explorer/Tree/SampleDataTree.tsx
+++ b/src/Explorer/Tree/SampleDataTree.tsx
@@ -17,7 +17,7 @@ export const SampleDataTree = ({
   const buildSampleDataTree = (): TreeNode => {
     const updatedSampleTree: TreeNode = {
       label: sampleDataResourceTokenCollection.databaseId,
-      isExpanded: true,
+      isExpanded: false,
       iconSrc: CosmosDBIcon,
       className: "databaseHeader",
       children: [
@@ -47,6 +47,7 @@ export const SampleDataTree = ({
             {
               label: "Items",
               onClick: () => sampleDataResourceTokenCollection.onDocumentDBDocumentsClick(),
+              contextMenu: ResourceTreeContextMenuButtonFactory.createSampleCollectionContextMenuButton(),
               isSelected: () =>
                 useSelectedNode
                   .getState()