import api from './api'
import type { AiCategorizationSettings, OllamaConnectionTestResult, ClassifyUncategorizedResult } from '@/types'

export const aiCategorizationApi = {
  getSettings: () =>
    api.get<AiCategorizationSettings>('/ai-categorization/settings'),

  updateSettings: (data: AiCategorizationSettings) =>
    api.put<AiCategorizationSettings>('/ai-categorization/settings', data),

  testConnection: () =>
    api.post<OllamaConnectionTestResult>('/ai-categorization/test-connection'),

  classifyUncategorized: (data: { accountId?: string; startDate?: string; endDate?: string; searchText?: string; status?: string }) =>
    api.post<ClassifyUncategorizedResult>('/ai-categorization/classify', data),
}