import request from '@/utils/request'
import type {
  LLMProviderInfo,
  LLMConfig,
  LLMConfigUpdate,
  LLMTestResult,
  LLMTestRequest,
  LLMProvidersResponse,
  LLMUsageTypesResponse,
  LLMConfigUpdateResponse,
  LLMAllConfigs
} from '@/types/llm'

export function getLLMProviders(): Promise<LLMProvidersResponse> {
  return request({
    url: '/admin/llm/providers',
    method: 'get'
  })
}

export function getLLMUsageTypes(): Promise<LLMUsageTypesResponse> {
  return request({
    url: '/admin/llm/usage-types',
    method: 'get'
  })
}

export function getLLMConfig(usageType?: string): Promise<LLMConfig | LLMAllConfigs> {
  const params: Record<string, string> = {}
  if (usageType) {
    params.usage_type = usageType
  }
  return request({
    url: '/admin/llm/config',
    method: 'get',
    params
  })
}

export function updateLLMConfig(data: LLMConfigUpdate): Promise<LLMConfigUpdateResponse> {
  return request({
    url: '/admin/llm/config',
    method: 'put',
    data
  })
}

export function testLLM(data: LLMTestRequest): Promise<LLMTestResult> {
  return request({
    url: '/admin/llm/test',
    method: 'post',
    data
  })
}

export type {
  LLMProviderInfo,
  LLMConfig,
  LLMConfigUpdate,
  LLMTestResult,
  LLMTestRequest,
  LLMProvidersResponse,
  LLMUsageTypesResponse,
  LLMConfigUpdateResponse,
  LLMAllConfigs,
  LLMUsageType
}