Yi.Admin/Yi.Ai.Vue3/src/api/model/index.ts

import type { GetSessionListVO } from './types';
import { del, get, post, put } from '@/utils/request';

// 获取当前用户的模型列表
export function getModelList() {
  return get<GetSessionListVO[]>('/ai-chat/model').json();
}
// 申请ApiKey
export function applyApiKey() {
  return post<any>('/token').json();
}
// 获取ApiKey
export function getApiKey() {
  return get<any>('/token').json();
}

// 查询充值记录
export function getRechargeLog() {
  return get<any>('/recharge/account').json();
}

// 查询用户近7天token消耗
// tokenId: 可选，传入则查询该token的用量，不传则查询全部
export function getLast7DaysTokenUsage(tokenId?: string) {
  const url = tokenId
    ? `/usage-statistics/last7Days-token-usage?tokenId=${tokenId}`
    : '/usage-statistics/last7Days-token-usage';
  return get<any>(url).json();
}
// 查询用户token消耗各模型占比
// tokenId: 可选，传入则查询该token的用量，不传则查询全部
export function getModelTokenUsage(tokenId?: string) {
  const url = tokenId
    ? `/usage-statistics/model-token-usage?tokenId=${tokenId}`
    : '/usage-statistics/model-token-usage';
  return get<any>(url).json();
}

// 获取当前用户得token列表
export function getTokenList() {
  return get<any>('/token/list').json();
}

// 创建token
export function createToken(data: any) {
  return post<any>('/token', data).json();
}

// 编辑token
export function editToken(data: any) {
  return put('/token', data).json();
}

// 删除token
export function deleteToken(id: string) {
  return del(`/token/${id}`).json();
}

// 启用token
export function enableToken(id: string) {
  return post(`/token/${id}/enable`).json();
}

// 禁用token
export function disableToken(id: string) {
  return post(`/token/${id}/disable`).json();
}

// 新增接口2
// 获取可选择的token信息
export function getSelectableTokenInfo() {
  return get<any>('/token/select-list').json();
}
/*
返回数据
  [
  {
    "tokenId": "3fa85f64-5717-4562-b3fc-2c963f66afa6",
    "name": "string",
    "isDisabled": true
  }
  ] */
// 获取当前用户尊享包不同token用量占比（饼图）
export function getPremiumPackageTokenUsage() {
  return get<any>('/usage-statistics/premium-token-usage/by-token').json();
}
/* 返回数据
  [
  {
    "tokenId": "3fa85f64-5717-4562-b3fc-2c963f66afa6",
    "tokenName": "string",
    "tokens": 0,
    "percentage": 0
  }
  ] */