Annotation Agent

async generateAnnotations(
  params: GenerateAnnotationsParams,
  onProgress?: (event: ProgressEvent) => void,
  signal?: AbortSignal
): Promise<GenerateAnnotationsResponse>

async generateAnnotationsStream(
  params: GenerateAnnotationsParams,
  signal?: AbortSignal
): Promise<AsyncIterableIterator<SSEEvent<GenerateAnnotationsResponse>>>

interface GenerateAnnotationsParams {
  config: GraphConfig;
  userPrompt?: string;
  metadata?: Metadata;
}

interface GenerateAnnotationsResponse {
  response: {
    message: string;
    steps: string[];
  };
  config: GraphConfig;
}

import { GraphyAiSdk } from '@graphysdk/agents-sdk';
import type { GraphConfig } from '@graphysdk/core';

const ai = new GraphyAiSdk({
  apiKey: process.env.GRAPHY_API_KEY,
  baseUrl: 'https://agents.graphy.dev',
});

const config: GraphConfig = {
  type: 'column',
  data: {
    columns: [
      { key: 'month', label: 'Month' },
      { key: 'revenue', label: 'Revenue' },
    ],
    rows: [
      { month: 'Jan', revenue: 100 },
      { month: 'Feb', revenue: 140 },
      { month: 'Mar', revenue: 220 },
      { month: 'Apr', revenue: 130 },
    ],
  },
};

const result = await ai.generateAnnotations({
  config,
  userPrompt: 'Highlight the March peak and add a tooltip explaining it',
});

console.log(result.config.annotations); // Annotations applied to the chart

const result = await ai.generateAnnotations(
  {
    config,
    userPrompt: 'Add a difference arrow between the first and last month',
  },
  (progress) => {
    console.log(progress.message);
  }
);

import { isProgressEvent, isCompleteEvent, isErrorEvent } from '@graphysdk/agents-sdk';

const stream = await ai.generateAnnotationsStream({
  config,
  userPrompt: 'Highlight every month above 150',
});

for await (const event of stream) {
  if (isProgressEvent(event)) {
    console.log(event.message);
  }

  if (isCompleteEvent(event)) {
    console.log('Result:', event.data.config);
  }

  if (isErrorEvent(event)) {
    console.error('Error:', event.error);
  }
}

import { isGraphyApiError } from '@graphysdk/agents-sdk';

try {
  const result = await ai.generateAnnotations({
    config,
    userPrompt: 'Highlight the peak value',
  });
} catch (error) {
  if (isGraphyApiError(error)) {
    console.error('API error:', error.message);
  }
}