Update nerdgraph-rag.mdx

Author: tejaswaroop-nr

src/content/docs/apis/nerdgraph/examples/nerdgraph-rag.mdx

@@ -24,7 +24,7 @@ To learn more about the knowledge connector, refer to [New Relic AI Knowledge co
 
 To start indexing your content and benefit from the knowledge connector with New Relic AI, follow these mentioned steps:
 
-## Create your RAG tool [#create-your-rag-tool]
+## Task 1: Create your RAG tool [#create-your-rag-tool]
 
 <CollapserGroup>
 <Collapser id ="get-started" title ="Obtain your organization ID">
@@ -33,7 +33,7 @@ Organization ID uniquely identifies your account and ensures that any RAG tool y
 
 Organization ID is required to perform mutations and queries in NerdGraph when setting up and managing RAG tools and documents. Run the following query and keep the organization ID handy for later steps.
 
-#### Sample query
+### Sample query
 ```graphql
 {
   actor {
@@ -52,7 +52,8 @@ To create a new RAG tool in your New Relic account, run the following `entityMan
 
 1. Make sure to replace `${ORGANIZATION_ID}` with your actual organization ID obtained in the previous step.
 2. If successful, you'll receive an `id` for your RAG tool.
-#### Input parameters
+
+### Input parameters
 
 <table>
     <thead>
@@ -103,7 +104,7 @@ To create a new RAG tool in your New Relic account, run the following `entityMan
     </tbody>
 </table>
 
-#### Sample mutation
+### Sample mutation
 ```graphql
 mutation {
   entityManagementCreateRagTool(
@@ -122,9 +123,11 @@ mutation {
 
 You should save the `id` returned as you'll need it in later steps to link documents to the tool, verify relationships, and query the tool for relevant information in New Relic.
 </Collapser>
+</CollapserGroup>
 
-## The indexing phase [#the-indexing-phase]
+## Task 2: Index your documents [#index-your-documents]
 
+<CollapserGroup>
 <Collapser id = "upload-document" title = "Upload a document via the Blob API">
 
 <Callout variant="important">
@@ -144,7 +147,7 @@ You need a valid New Relic API key with permissions to upload documents. To get
 
 Here's an example of how to upload a document using a `curl` `bash` command:
 
-#### Input parameters
+### Input parameters
 
 <table>
     <thead>
@@ -184,7 +187,7 @@ Here's an example of how to upload a document using a `curl` `bash` command:
 </table>
 
 
-#### Sample query
+### Sample query
 ```shell
 curl -X POST https://blob-api.one-service.newrelic.com/v1/e/organizations/$ORGANIZATION_ID/RagDocuments \
      -H 'Api-Key: NRAK-XXXXXXXXXX' \
@@ -193,7 +196,7 @@ curl -X POST https://blob-api.one-service.newrelic.com/v1/e/organizations/$ORGAN
      -d @incidents.json
 ```
 
-#### Sample response
+### Sample response
 
 <table>
     <thead>
@@ -253,7 +256,7 @@ After a document is uploaded via the Blob API, running this query confirms that
     </tbody>
 </table>
 
-#### Sample query
+### Sample query
 
 In the query below, replace the `${RAG_DOCUMENT_GUID}` placeholder with the `entityGuid` you received in the previous step.
 
@@ -293,7 +296,7 @@ Now that you've created a RAG tool, uploaded your document, and verified that th
 1. Replace `${RAG_DOCUMENT_GUID}` with the `entityGuid` from the response of the document upload via the Blob API.
 2. Replace `${RAG_TOOL_GUID}` with the `id` from the response of the RAG tool creation mutation.
 
-#### Input parameters
+### Input parameters
 
 <table>
     <thead>
@@ -356,7 +359,7 @@ Now that you've created a RAG tool, uploaded your document, and verified that th
     </tbody>
 </table>
 
-#### Sample mutation
+### Sample mutation
 
 ```graphql
 mutation {
@@ -388,17 +391,19 @@ mutation {
 }
 ```
 </Collapser>
+</CollapserGroup>
 
-## The retrieval phase [#the-retrieval-phase]
+## Task 3: Retrieve relevant information [#retrieve-relevant-information]
 
+<CollapserGroup>
 <Collapser id="query-to-see-relationships-between-rag-documents-and-rag-tools" title="Query to see relationships between RAG documents and RAG tools">
 
 After creating a relationship between your RAG document and RAG tool, you can verify the association by querying relationships in NerdGraph. This helps ensure your document is properly linked and available for New Relic AI to use.
 
 - Replace `${RAG_DOCUMENT_ID}` with the `entityGuid` of your uploaded document.
 
 
-#### Input parameters
+### Input parameters
 
 <table>
     <thead>
@@ -437,7 +442,7 @@ After creating a relationship between your RAG document and RAG tool, you can ve
     </tbody>
 </table>
 
-#### Sample query
+### Sample query
 
 ```graphql
 {
@@ -464,37 +469,34 @@ After creating a relationship between your RAG document and RAG tool, you can ve
 
 After you've set up your RAG tool and indexed documents, you can query the RAG tool to retrieve relevant information based on your prompt. This allows New Relic AI to surface context-aware answers using your organization's documentation.
 
-#### Input parameters
+### Input parameters
 
 <table>
-    <thead>
-        <tr>
-            <th>Parameter Name</th>
-            <th>Data Type</th>
-            <th>Is it Required?</th>
-            <th>Description</th>
-            <th>Response (If required)</th>
-        </tr>
-    </thead>
-    <tbody>
-        <tr>
-            <td>`prompt`</td>
-            <td>String</td>
-            <td>Yes</td>
-            <td>The natural language query you want the RAG tool to process.</td>
-            <td>(Input parameter)</td>
-        </tr>
-        <tr>
-            <td>`toolId`</td>
-            <td>String</td>
-            <td>Yes</td>
-            <td>The unique GUID of the RAG tool to be queried.</td>
-            <td>(Input parameter)</td>
-        </tr>
-    </tbody>
+  <thead>
+    <tr>
+      <th>Parameter Name</th>
+      <th>Data Type</th>
+      <th>Is it Required?</th>
+      <th>Description</th>
+    </tr>
+  </thead>
+  <tbody>
+    <tr>
+      <td>`prompt`</td>
+      <td>String</td>
+      <td>Yes</td>
+      <td>The natural language query you want the RAG tool to process.</td>
+    </tr>
+    <tr>
+      <td>`toolId`</td>
+      <td>String</td>
+      <td>Yes</td>
+      <td>The unique GUID of the RAG tool to be queried.</td>
+    </tr>
+  </tbody>
 </table>
 
-#### Sample query
+### Sample query
 ```graphql
 {
   actor {