Merge pull request #44 from grafana/matryer/create-dashboards

added create_dashboard tool

Author: matryer

public/app/features/dash/agent/tools/index.ts

@@ -17,6 +17,7 @@ import { prometheusRangeQueryTool } from './prometheusRangeQuery';
 import { readDashboardPanelsTool } from './tool_read_dashboard_panels';
 import { simulateToolError } from './tool_simulate_tool_error';
 import { updateDashboardPanelsTool } from './tool_update_dashboard_panels';
+import { createDashboardTool } from './tool_create_dashboard';
 
 const grafanaComSearch = new TavilySearchResults({
   apiKey: process.env.TAVILY_API_KEY,
@@ -60,6 +61,7 @@ export const tools = [
   updateDashboardPanelsTool,
   getDrilldownLogToSummarizeTool,
   simulateToolError,
+  createDashboardTool,
 ];
 
 export const toolsByName = tools.reduce(

public/app/features/dash/agent/tools/tool_create_dashboard.ts

@@ -0,0 +1,69 @@
+import { tool } from '@langchain/core/tools';
+import { z } from 'zod';
+import { locationService } from '@grafana/runtime';
+
+const dashboardSchema = z.object({
+  title: z.string().describe('The title of the dashboard'),
+  uid: z.string().optional().describe('The unique identifier for the dashboard'),
+  folderUid: z.string().optional().describe('The folder UID to place the dashboard in'),
+  data: z.any().describe('The dashboard data object containing panels and other configuration'),
+});
+
+export const createDashboardTool = tool(
+  async (input): Promise<string> => {
+    // Validate input
+    const validatedInput = dashboardSchema.parse(input);
+
+    // Create dashboard data
+    const dashboardData = {
+      dashboard: {
+        ...validatedInput.data,
+        title: validatedInput.title,
+        uid: validatedInput.uid,
+        folderUid: validatedInput.folderUid,
+      },
+      overwrite: false,
+    };
+
+    // Create dashboard
+    const response = await fetch('/api/dashboards/db', {
+      method: 'POST',
+      headers: {
+        'Content-Type': 'application/json',
+      },
+      body: JSON.stringify(dashboardData),
+    });
+
+    if (!response.ok) {
+      return JSON.stringify({
+        error: 'Failed to create dashboard',
+        details: response.statusText,
+      });
+    }
+
+    const result = await response.json();
+
+    if (!result.uid || !result.url) {
+      return JSON.stringify({
+        error: 'Invalid response from Grafana API',
+        details: 'Response missing required fields',
+      });
+    }
+
+    // Navigate to the new dashboard
+    locationService.push({ pathname: result.url });
+
+    return JSON.stringify({
+      success: true,
+      uid: result.uid,
+      url: result.url,
+      message: 'Dashboard created and navigated to successfully',
+    });
+  },
+  {
+    name: 'create_dashboard',
+    description:
+      'Creates a new dashboard with the specified title, optional UID, folder, and dashboard data. Automatically navigates to the new dashboard after creation.',
+    schema: dashboardSchema,
+  }
+);

tool_create_dashboard.ts

@@ -0,0 +1,48 @@
+import { z } from 'zod';
+import { GrafanaClient } from './grafana-client';
+
+const dashboardSchema = z.object({
+  title: z.string(),
+  uid: z.string().optional(),
+  folderUid: z.string().optional(),
+  data: z.any(),
+});
+
+export type CreateDashboardInput = z.infer<typeof dashboardSchema>;
+
+export async function create_dashboard(
+  client: GrafanaClient,
+  input: CreateDashboardInput
+): Promise<{ uid: string; url: string }> {
+  // Validate input
+  const validatedInput = dashboardSchema.parse(input);
+
+  // Create dashboard data
+  const dashboardData = {
+    dashboard: {
+      ...validatedInput.data,
+      title: validatedInput.title,
+      uid: validatedInput.uid,
+      folderUid: validatedInput.folderUid,
+    },
+    overwrite: false,
+  };
+
+  // Create dashboard
+  const response = await client.post('/api/dashboards/db', dashboardData);
+
+  if (!response.ok) {
+    throw new Error(`Failed to create dashboard: ${response.statusText}`);
+  }
+
+  const result = await response.json();
+
+  if (!result.uid || !result.url) {
+    throw new Error('Invalid response from Grafana API');
+  }
+
+  return {
+    uid: result.uid,
+    url: result.url,
+  };
+}