feat(logs): Wire Seer AI visualization params into logs explore

static/app/components/searchQueryBuilder/askSeerCombobox/types.ts

@@ -1,5 +1,21 @@
 import type {ChartType} from 'sentry/views/insights/common/components/chart';
 
+export interface SeerRawResponseItem {
+  end: string | null;
+  group_by: string[];
+  mode: string;
+  query: string;
+  sort: string;
+  start: string | null;
+  stats_period: string;
+  visualization?: Array<{chart_type?: number; y_axes?: string[]}>;
+}
+
+export interface SeerRawResponse {
+  responses: SeerRawResponseItem[];
+  unsupported_reason: string | null;
+}
+
 export interface NoneOfTheseItem {
   key: 'none-of-these';
   label: string;
@@ -18,7 +34,7 @@ export interface QueryTokensProps {
   sort?: string;
   start?: string | null;
   statsPeriod?: string;
-  visualizations?: Array<{chartType: ChartType; yAxes: string[]}>;
+  visualizations?: Array<{yAxes: string[]; chartType?: ChartType}>;
 }
 
 export interface AskSeerSearchQuery extends QueryTokensProps {
@@ -29,7 +45,7 @@ export interface AskSeerSearchQuery extends QueryTokensProps {
   sort: string;
   start: string | null;
   statsPeriod: string;
-  visualizations: Array<{chartType: ChartType; yAxes: string[]}>;
+  visualizations: Array<{yAxes: string[]; chartType?: ChartType}>;
 }
 
 /**

static/app/components/searchQueryBuilder/askSeerCombobox/useSeerComboBoxSetup.spec.tsx

@@ -0,0 +1,328 @@
+import {PageFiltersFixture} from 'sentry-fixture/pageFilters';
+import {ProjectFixture} from 'sentry-fixture/project';
+
+import {renderHookWithProviders} from 'sentry-test/reactTestingLibrary';
+
+import {PageFiltersStore} from 'sentry/components/pageFilters/store';
+import {SearchQueryBuilderProvider} from 'sentry/components/searchQueryBuilder/context';
+import {ProjectsStore} from 'sentry/stores/projectsStore';
+import {ChartType} from 'sentry/views/insights/common/components/chart';
+
+import {
+  buildSeerDateTimeSelection,
+  mapSeerResponseItem,
+  transformSeerResponse,
+  useInitialSeerQuery,
+  useSelectedProjectIds,
+  useSelectedProjectIdsForMutation,
+} from './useSeerComboBoxSetup';
+
+const defaultProviderProps = {
+  enableAISearch: true,
+  filterKeys: {},
+  getTagValues: () => Promise.resolve([]),
+  initialQuery: '',
+  searchSource: 'test' as const,
+};
+
+function makeWrapper(providerProps = {}) {
+  return function Wrapper({children}: {children: React.ReactNode}) {
+    return (
+      <SearchQueryBuilderProvider {...defaultProviderProps} {...providerProps}>
+        {children}
+      </SearchQueryBuilderProvider>
+    );
+  };
+}
+
+describe('transformSeerResponse', () => {
+  const mapItem = (r: any) => ({
+    query: r?.query ?? '',
+    sort: r?.sort ?? '',
+    groupBys: r?.group_by ?? [],
+    statsPeriod: r?.stats_period ?? '',
+    start: r?.start ?? null,
+    end: r?.end ?? null,
+  });
+
+  it('transforms a response with responses array', () => {
+    const rawResponse = {
+      responses: [
+        {
+          query: 'is:unresolved',
+          sort: '-count',
+          group_by: ['project'],
+          stats_period: '24h',
+          start: null,
+          end: null,
+          mode: 'samples',
+        },
+      ],
+    };
+
+    const result = transformSeerResponse(rawResponse as any, mapItem);
+
+    expect(result).toEqual([
+      {
+        query: 'is:unresolved',
+        sort: '-count',
+        groupBys: ['project'],
+        statsPeriod: '24h',
+        start: null,
+        end: null,
+      },
+    ]);
+  });
+
+  it('wraps a single response without responses array', () => {
+    const singleResponse = {
+      query: 'is:unresolved',
+      sort: '',
+      groupBys: [],
+      statsPeriod: '24h',
+      start: null,
+      end: null,
+    };
+
+    const result = transformSeerResponse(singleResponse, mapItem);
+
+    expect(result).toEqual([singleResponse]);
+  });
+
+  it('handles empty responses array', () => {
+    const rawResponse = {responses: []};
+    const result = transformSeerResponse(rawResponse as any, mapItem);
+    expect(result).toEqual([]);
+  });
+
+  it('maps visualization fields when present', () => {
+    const rawResponse = {
+      responses: [
+        {
+          query: 'span.op:db',
+          sort: '',
+          group_by: [],
+          stats_period: '1h',
+          start: null,
+          end: null,
+          mode: 'spans',
+          visualization: [{chart_type: 1, y_axes: ['count()']}],
+        },
+      ],
+    };
+
+    const mapWithViz = (r: any) => ({
+      query: r?.query ?? '',
+      visualizations:
+        r?.visualization?.map((v: any) => ({
+          chartType: v.chart_type,
+          yAxes: v.y_axes ?? [],
+        })) ?? [],
+    });
+
+    const result = transformSeerResponse(rawResponse as any, mapWithViz);
+
+    expect(result).toEqual([
+      {
+        query: 'span.op:db',
+        visualizations: [{chartType: 1, yAxes: ['count()']}],
+      },
+    ]);
+  });
+});
+
+describe('mapSeerResponseItem', () => {
+  it('maps Seer response fields into the shared query shape', () => {
+    expect(
+      mapSeerResponseItem({
+        query: 'span.op:db',
+        sort: '-timestamp',
+        group_by: ['project'],
+        stats_period: '24h',
+        start: null,
+        end: null,
+        mode: 'aggregates',
+        visualization: [{chart_type: ChartType.BAR, y_axes: ['count()']}],
+      })
+    ).toEqual({
+      query: 'span.op:db',
+      sort: '-timestamp',
+      groupBys: ['project'],
+      statsPeriod: '24h',
+      start: null,
+      end: null,
+      mode: 'aggregates',
+      visualizations: [{chartType: ChartType.BAR, yAxes: ['count()']}],
+    });
+  });
+
+  it('leaves missing or invalid chart types undefined', () => {
+    expect(
+      mapSeerResponseItem(
+        {
+          query: 'is:unresolved',
+          sort: '',
+          group_by: [],
+          stats_period: '',
+          start: null,
+          end: null,
+          mode: '',
+          visualization: [{y_axes: []}, {chart_type: 999, y_axes: ['count()']}],
+        },
+        'issues'
+      )
+    ).toEqual({
+      query: 'is:unresolved',
+      sort: '',
+      groupBys: [],
+      statsPeriod: '',
+      start: null,
+      end: null,
+      mode: 'issues',
+      visualizations: [{yAxes: ['count()']}],
+    });
+  });
+});
+
+describe('buildSeerDateTimeSelection', () => {
+  const pageFiltersDatetime = {
+    start: '2024-01-01T00:00:00',
+    end: '2024-01-02T00:00:00',
+    period: '24h',
+    utc: true,
+  };
+
+  it('strips Z suffix and converts to ISO when both start and end provided', () => {
+    const result = buildSeerDateTimeSelection(
+      '2024-06-01T00:00:00Z',
+      '2024-06-02T00:00:00Z',
+      '',
+      pageFiltersDatetime
+    );
+
+    expect(result.start).toBe(new Date('2024-06-01T00:00:00').toISOString());
+    expect(result.end).toBe(new Date('2024-06-02T00:00:00').toISOString());
+    expect(result.period).toBeNull();
+    expect(result.utc).toBe(true);
+  });
+
+  it('falls back to pageFilters datetime when no start/end', () => {
+    const result = buildSeerDateTimeSelection(null, null, '', pageFiltersDatetime);
+
+    expect(result.start).toBe('2024-01-01T00:00:00');
+    expect(result.end).toBe('2024-01-02T00:00:00');
+    expect(result.period).toBe('24h');
+    expect(result.utc).toBe(true);
+  });
+
+  it('uses statsPeriod when no start/end', () => {
+    const result = buildSeerDateTimeSelection(null, null, '7d', pageFiltersDatetime);
+
+    expect(result.start).toBe('2024-01-01T00:00:00');
+    expect(result.end).toBe('2024-01-02T00:00:00');
+    expect(result.period).toBe('7d');
+  });
+
+  it('sets period to null when start and end are provided', () => {
+    const result = buildSeerDateTimeSelection(
+      '2024-06-01T00:00:00',
+      '2024-06-02T00:00:00',
+      '7d',
+      pageFiltersDatetime
+    );
+
+    expect(result.period).toBeNull();
+  });
+
+  it('handles dates without Z suffix', () => {
+    const result = buildSeerDateTimeSelection(
+      '2024-06-01T12:00:00',
+      '2024-06-02T12:00:00',
+      '',
+      pageFiltersDatetime
+    );
+
+    expect(result.start).toBe(new Date('2024-06-01T12:00:00').toISOString());
+    expect(result.end).toBe(new Date('2024-06-02T12:00:00').toISOString());
+  });
+});
+
+describe('useInitialSeerQuery', () => {
+  it('returns empty string when no query or input', () => {
+    const {result} = renderHookWithProviders(() => useInitialSeerQuery(), {
+      additionalWrapper: makeWrapper({initialQuery: ''}),
+    });
+
+    expect(result.current).toBe('');
+  });
+
+  it('uses committed query when available', () => {
+    const {result} = renderHookWithProviders(() => useInitialSeerQuery(), {
+      additionalWrapper: makeWrapper({initialQuery: 'is:unresolved'}),
+    });
+
+    expect(result.current).toBe('is:unresolved');
+  });
+});
+
+describe('useSelectedProjectIds', () => {
+  beforeEach(() => {
+    ProjectsStore.loadInitialData([
+      ProjectFixture({id: '1', isMember: true}),
+      ProjectFixture({id: '2', isMember: true}),
+      ProjectFixture({id: '3', isMember: false}),
+    ]);
+  });
+
+  it('returns pageFilters projects when specified', () => {
+    PageFiltersStore.onInitializeUrlState(PageFiltersFixture({projects: [1, 2]}), false);
+
+    const {result} = renderHookWithProviders(() => useSelectedProjectIds());
+
+    expect(result.current).toEqual([1, 2]);
+  });
+
+  it('falls back to member projects when pageFilters has all projects', () => {
+    PageFiltersStore.onInitializeUrlState(PageFiltersFixture({projects: [-1]}), false);
+
+    const {result} = renderHookWithProviders(() => useSelectedProjectIds());
+
+    expect(result.current).toEqual([1, 2]);
+  });
+
+  it('falls back to member projects when pageFilters has no projects', () => {
+    PageFiltersStore.onInitializeUrlState(PageFiltersFixture({projects: []}), false);
+
+    const {result} = renderHookWithProviders(() => useSelectedProjectIds());
+
+    expect(result.current).toEqual([1, 2]);
+  });
+});
+
+describe('useSelectedProjectIdsForMutation', () => {
+  beforeEach(() => {
+    ProjectsStore.loadInitialData([
+      ProjectFixture({id: '10', isMember: true}),
+      ProjectFixture({id: '20', isMember: false}),
+    ]);
+  });
+
+  it('returns string IDs for member projects', () => {
+    PageFiltersStore.onInitializeUrlState(PageFiltersFixture({projects: [-1]}), false);
+
+    const {result} = renderHookWithProviders(() => useSelectedProjectIdsForMutation());
+
+    expect(result.current).toEqual(['10']);
+  });
+
+  it('returns pageFilters projects when specified', () => {
+    PageFiltersStore.onInitializeUrlState(
+      PageFiltersFixture({projects: [10, 20]}),
+      false
+    );
+
+    const {result} = renderHookWithProviders(() => useSelectedProjectIdsForMutation());
+
+    expect(result.current).toEqual([10, 20]);
+  });
+});