lcbp3/frontend/lib/services/admin-ai.service.ts

// File: lib/services/admin-ai.service.ts
// Change Log
// - 2026-06-14: เพิ่ม methods สำหรับจัดการ Prompt Versions และ RAG Prep Sandbox (T020, T030, T039)
// - 2026-05-21: เพิ่ม service สำหรับ AI Admin Console toggle API.
// - 2026-05-21: เพิ่ม service method `getHealth` สำหรับดึงข้อมูลสุขภาพของระบบ AI (T028).
// - 2026-05-21: เพิ่ม API service สำหรับ Superadmin Sandbox RAG (T037).
// - 2026-05-21: เพิ่ม service method `submitSandboxExtract` สำหรับอัปโหลดไฟล์ใน OCR Sandbox (T043).
// - 2026-05-25: เพิ่ม methods สำหรับจัดการโมเดล AI แบบไดนามิก (ADR-027).
// - 2026-05-29: เพิ่ม ocr field ใน AiSystemHealth interface ตาม OcrService.checkHealth()
// - 2026-05-29: เพิ่ม ocrText, ocrUsed, promptVersionUsed ใน AiSandboxJobResult
// - 2026-06-06: เพิ่ม llmPrompt ใน AiSandboxJobResult เพื่อแสดง prompt ที่ส่งไป LLM
// - 2026-05-30: เพิ่มเมธอด getOcrEngines และ selectOcrEngine สำหรับจัดการ OCR engines (T017, T018, US1)
// - 2026-05-30: เพิ่ม getVramStatus และปรับปรุง getAvailableModels/setActiveModel/addModel ให้เรียกใช้ endpoints ใหม่ที่มี VRAM capacity check (T031-T034, US2)
// - 2026-06-03: ADR-034 — เพิ่ม activeModels field (หลัก+OCR) ใน AiSystemHealth interface
// - 2026-06-02: แก้ endpoint getAvailableModels ให้ตรงกับ backend admin route (/ai/admin/models)
// - 2026-06-02: normalize VRAM response ให้รองรับ field names จาก backend ปัจจุบันและรูปแบบ loadedModels แบบเดิม
// - 2026-06-13: T027-T029 — เพิ่ม getSandboxProfile, saveSandboxProfile, resetSandboxProfile สำหรับ sandbox parameter management
// - 2026-06-13: T042-T043 — เพิ่ม applyProfile และ getProductionDefaults สำหรับปรับใช้และดึงค่า production parameters
// - 2026-06-13: US4 — อัปเดต submitSandboxExtract และ submitSandboxAiExtract ให้รองรับ project/contract publicId


import api from '../api/client';
import { AiJobResponse } from '../../types/ai';
import { PromptType, PromptVersion, ContextConfig } from '../types/ai-prompts';

export interface AiAdminSettings {
  aiFeaturesEnabled: boolean;
}

export interface QueueMetrics {
  active?: number;
  waiting?: number;
  failed?: number;
  completed?: number;
  isPaused?: boolean;
  error?: string;
}

export interface AiSystemHealth {
  activeModels?: {
    main: string;
    ocr: string;
  };
  ollama: {
    status: 'HEALTHY' | 'DEGRADED' | 'DOWN';
    latencyMs: number;
    models: string[];
    error?: string;
  };
  qdrant: {
    status: 'HEALTHY' | 'DEGRADED' | 'DOWN';
    latencyMs: number;
    collections?: string[];
    error?: string;
  };
  ocr: {
    status: 'HEALTHY' | 'DOWN';
    latencyMs: number;
    url: string;
    error?: string;
  };
  queues: {
    realtime: QueueMetrics;
    batch: QueueMetrics;
  };
  timestamp: string;
}

export interface AiRagCitation {
  pointId: string | number;
  score: number;
  docType?: string;
  docNumber?: string;
  snippet?: string;
}

export interface AiSandboxJobResult {
  requestPublicId: string;
  status: 'pending' | 'processing' | 'completed' | 'failed' | 'cancelled' | 'not_found';
  answer?: string;
  ocrText?: string;
  ocrUsed?: boolean;
  engineUsed?: string;
  fallbackUsed?: boolean;
  promptVersionUsed?: number;
  llmPrompt?: string;
  citations?: AiRagCitation[];
  confidence?: number;
  usedFallbackModel?: boolean;
  errorMessage?: string;
  completedAt?: string;
}

export interface LoadedModelInfo {
  modelId: string;
  modelName: string;
  vramUsageMB: number;
}

export interface VramStatusResponse {
  totalVRAMMB: number;
  usedVRAMMB: number;
  usagePercent: number;
  thresholdPercent: number;
  loadedModels: LoadedModelInfo[];
  canLoadModel: boolean;
  lastUpdated: string;
}

interface RawVramStatusResponse {
  totalVRAMMB?: number;
  usedVRAMMB?: number;
  usagePercent?: number;
  thresholdPercent?: number;
  loadedModels?: Array<string | LoadedModelInfo>;
  canLoadModel?: boolean;
  lastUpdated?: string;
  totalVramMb?: number;
  usedVramMb?: number;
  freeVramMb?: number;
  hasCapacity?: boolean;
}

export interface AiAvailableModel {
  id?: number;
  modelId?: string;
  modelName: string;
  modelVersion: string;
  description?: string;
  vramGb?: number;
  vramRequirementMB?: number;
  isActive: boolean;
  isDefault: boolean;
  createdAt: string;
  updatedAt: string;
}

export interface AiModelsResponse {
  models: AiAvailableModel[];
  activeModel: string;
}

export interface AiActiveModelResponse {
  activeModel: string;
}

/** พารามิเตอร์ sandbox draft สำหรับ profile (ADR-036) */
export interface SandboxProfileParams {
  canonicalModel: 'np-dms-ai' | 'np-dms-ocr';
  temperature: number;
  topP: number;
  maxTokens: number | null;
  numCtx: number | null;
  repeatPenalty: number;
  keepAliveSeconds: number;
}

const extractData = <T>(value: unknown): T => {
  if (value && typeof value === 'object' && 'data' in value) {
    return (value as { data: T }).data;
  }
  return value as T;
};

const normalizeLoadedModels = (
  models: Array<string | LoadedModelInfo> | undefined
): LoadedModelInfo[] => {
  if (!Array.isArray(models)) {
    return [];
  }
  return models.map((model, index) => {
    if (typeof model === 'string') {
      return {
        modelId: `${model}-${index}`,
        modelName: model,
        vramUsageMB: 0,
      };
    }
    return model;
  });
};

const normalizeVramStatus = (value: unknown): VramStatusResponse => {
  const raw = extractData<RawVramStatusResponse>(value);
  const totalVRAMMB = raw.totalVRAMMB ?? raw.totalVramMb ?? 0;
  const usedVRAMMB = raw.usedVRAMMB ?? raw.usedVramMb ?? 0;
  const usagePercent =
    raw.usagePercent ??
    (totalVRAMMB > 0 ? Math.round((usedVRAMMB / totalVRAMMB) * 100) : 0);

  return {
    totalVRAMMB,
    usedVRAMMB,
    usagePercent,
    thresholdPercent: raw.thresholdPercent ?? 90,
    loadedModels: normalizeLoadedModels(raw.loadedModels),
    canLoadModel: raw.canLoadModel ?? raw.hasCapacity ?? false,
    lastUpdated: raw.lastUpdated ?? new Date().toISOString(),
  };
};

/** Service สำหรับเรียก AI Admin Console API ผ่าน DMS Backend เท่านั้น */
export const adminAiService = {
  getStatus: async (): Promise<AiAdminSettings> => {
    const { data } = await api.get('/ai/status');
    return extractData<AiAdminSettings>(data);
  },
  getSettings: async (): Promise<AiAdminSettings> => {
    const { data } = await api.get('/ai/admin/settings');
    return extractData<AiAdminSettings>(data);
  },
  toggleFeatures: async (enabled: boolean): Promise<AiAdminSettings> => {
    const { data } = await api.post('/ai/admin/toggle', { enabled });
    return extractData<AiAdminSettings>(data);
  },
  getHealth: async (): Promise<AiSystemHealth> => {
    const { data } = await api.get('/ai/admin/health');
    return extractData<AiSystemHealth>(data);
  },
  submitSandboxRag: async (
    projectPublicId: string,
    question: string
  ): Promise<{ requestPublicId: string; jobId: string; status: string }> => {
    const { data } = await api.post('/ai/admin/sandbox/rag', {
      projectPublicId,
      question,
    });
    return extractData<{ requestPublicId: string; jobId: string; status: string }>(data);
  },
  getSandboxJobStatus: async (id: string): Promise<AiSandboxJobResult> => {
    const { data } = await api.get(`/ai/admin/sandbox/job/${id}`);
    return extractData<AiSandboxJobResult>(data);
  },
  submitSandboxExtract: async (
    file: File,
    projectPublicId: string,
    contractPublicId?: string
  ): Promise<{ requestPublicId: string; jobId: string; status: string }> => {
    const formData = new FormData();
    formData.append('file', file);
    formData.append('projectPublicId', projectPublicId);
    if (contractPublicId) {
      formData.append('contractPublicId', contractPublicId);
    }
    const { data } = await api.post('/ai/admin/sandbox/extract', formData, {
      headers: {
        'Content-Type': 'multipart/form-data',
      },
    });
    return extractData<{ requestPublicId: string; jobId: string; status: string }>(data);
  },

  // --- Step 1: OCR Only (สำหรับตรวจคุณภาพ OCR ก่อนทดสอบ AI) ---

  submitSandboxOcr: async (
    file: File,
    engineType: string = 'auto',
    typhoonOptions?: { temperature?: number; topP?: number; repeatPenalty?: number }
  ): Promise<{ requestPublicId: string; jobId: string; status: string }> => {
    const formData = new FormData();
    formData.append('file', file);
    formData.append('engineType', engineType);
    if (typhoonOptions?.temperature !== undefined) {
      formData.append('temperature', String(typhoonOptions.temperature));
    }
    if (typhoonOptions?.topP !== undefined) {
      formData.append('topP', String(typhoonOptions.topP));
    }
    if (typhoonOptions?.repeatPenalty !== undefined) {
      formData.append('repeatPenalty', String(typhoonOptions.repeatPenalty));
    }
    const { data } = await api.post('/ai/admin/sandbox/ocr', formData, {
      headers: {
        'Content-Type': 'multipart/form-data',
      },
    });
    return extractData<{ requestPublicId: string; jobId: string; status: string }>(data);
  },

  // --- Step 2: AI Extraction (ใช้ OCR text ที่ cache จาก Step 1) ---

  submitSandboxAiExtract: async (
    requestPublicId: string,
    promptVersion: number | undefined,
    projectPublicId: string,
    contractPublicId?: string
  ): Promise<{ requestPublicId: string; jobId: string; status: string }> => {
    const { data } = await api.post('/ai/admin/sandbox/ai-extract', {
      requestPublicId,
      promptVersion,
      projectPublicId,
      contractPublicId,
    });
    return extractData<{ requestPublicId: string; jobId: string; status: string }>(data);
  },

  // --- AI Model Management (ADR-027, US2) ---

  getAvailableModels: async (): Promise<AiModelsResponse> => {
    const { data } = await api.get('/ai/admin/models');
    return extractData<AiModelsResponse>(data);
  },

  getActiveModel: async (): Promise<AiActiveModelResponse> => {
    const { data } = await api.get('/ai/admin/models/active');
    return extractData<AiActiveModelResponse>(data);
  },

  setActiveModel: async (modelId: string): Promise<AiActiveModelResponse> => {
    const { data } = await api.patch(`/ai/models/${encodeURIComponent(modelId)}/activate`, {});
    return extractData<AiActiveModelResponse>(data);
  },

  getVramStatus: async (): Promise<VramStatusResponse> => {
    const { data } = await api.get('/ai/vram/status');
    return normalizeVramStatus(data);
  },

  addModel: async (
    model: Omit<AiAvailableModel, 'id' | 'createdAt' | 'updatedAt'>
  ): Promise<{ model: AiAvailableModel }> => {
    const { data } = await api.post('/ai/models', model);
    return extractData<{ model: AiAvailableModel }>(data);
  },

  toggleModelActive: async (modelName: string): Promise<{ model: AiAvailableModel }> => {
    const { data } = await api.patch(`/ai/admin/models/${encodeURIComponent(modelName)}/toggle`);
    return extractData<{ model: AiAvailableModel }>(data);
  },

  removeModel: async (modelName: string): Promise<void> => {
    await api.delete(`/ai/admin/models/${encodeURIComponent(modelName)}`);
  },

  // --- OCR Engine Management (ADR-032) ---

  getOcrEngines: async (): Promise<OcrEngineResponse[]> => {
    const { data } = await api.get('/ai/ocr-engines');
    return extractData<OcrEngineResponse[]>(data);
  },

  selectOcrEngine: async (engineId: string): Promise<{ activeEngineName: string }> => {
    const { data } = await api.post(`/ai/ocr-engines/${encodeURIComponent(engineId)}/select`, {});
    return extractData<{ activeEngineName: string }>(data);
  },

  // --- Sandbox Parameter Management (ADR-036, T027-T029) ---

  getSandboxProfile: async (profileName: string): Promise<SandboxProfileParams> => {
    const { data } = await api.get(`/ai/sandbox-profiles/${encodeURIComponent(profileName)}`);
    return extractData<SandboxProfileParams>(data);
  },

  saveSandboxProfile: async (
    profileName: string,
    updates: Partial<SandboxProfileParams>,
    idempotencyKey: string
  ): Promise<SandboxProfileParams> => {
    const { data } = await api.put(
      `/ai/sandbox-profiles/${encodeURIComponent(profileName)}`,
      updates,
      { headers: { 'Idempotency-Key': idempotencyKey } }
    );
    return extractData<SandboxProfileParams>(data);
  },

  resetSandboxProfile: async (profileName: string): Promise<SandboxProfileParams> => {
    const { data } = await api.post(
      `/ai/sandbox-profiles/${encodeURIComponent(profileName)}/reset`,
      {}
    );
    return extractData<SandboxProfileParams>(data);
  },

  applyProfile: async (
    profileName: string,
    idempotencyKey: string
  ): Promise<SandboxProfileParams> => {
    const { data } = await api.post(
      `/ai/profiles/${encodeURIComponent(profileName)}/apply`,
      {},
      { headers: { 'Idempotency-Key': idempotencyKey } }
    );
    return extractData<SandboxProfileParams>(data);
  },

  getProductionDefaults: async (profileName: string): Promise<SandboxProfileParams> => {
    const { data } = await api.get(`/ai/profiles/${encodeURIComponent(profileName)}`);
    return extractData<SandboxProfileParams>(data);
  },

  submitAiJob: async (
    type: string,
    documentPublicId?: string,
    attachmentPublicId?: string,
    payload?: Record<string, unknown>,
    projectPublicId?: string
  ): Promise<AiJobResponse> => {
    const { data } = await api.post('/ai/jobs', {
      type,
      documentPublicId,
      attachmentPublicId,
      payload,
      projectPublicId,
    });
    return extractData<AiJobResponse>(data);
  },

  listPrompts: async (type: PromptType): Promise<PromptVersion[]> => {
    const { data } = await api.get(`/ai/prompts/${type}`);
    return extractData<PromptVersion[]>(data);
  },

  createPrompt: async (
    type: PromptType,
    updates: { template: string; contextConfig?: ContextConfig | null; manualNote?: string }
  ): Promise<PromptVersion> => {
    const { data } = await api.post(`/ai/prompts/${type}`, updates);
    return extractData<PromptVersion>(data);
  },

  deletePrompt: async (type: PromptType, versionNumber: number): Promise<void> => {
    await api.delete(`/ai/prompts/${type}/${versionNumber}`);
  },

  activatePrompt: async (type: PromptType, versionNumber: number): Promise<PromptVersion> => {
    const { data } = await api.post(`/ai/prompts/${type}/${versionNumber}/activate`);
    return extractData<PromptVersion>(data);
  },

  updatePromptNote: async (
    type: PromptType,
    versionNumber: number,
    manualNote: string
  ): Promise<PromptVersion> => {
    const { data } = await api.patch(`/ai/prompts/${type}/${versionNumber}/note`, { manualNote });
    return extractData<PromptVersion>(data);
  },

  getContextConfig: async (type: PromptType, versionNumber: number): Promise<ContextConfig> => {
    const { data } = await api.get(`/ai/prompts/${type}/${versionNumber}/context-config`);
    return extractData<ContextConfig>(data);
  },

  updateContextConfig: async (
    type: PromptType,
    versionNumber: number,
    contextConfig: ContextConfig
  ): Promise<ContextConfig> => {
    const { data } = await api.put(`/ai/prompts/${type}/${versionNumber}/context-config`, contextConfig);
    return extractData<ContextConfig>(data);
  },

  submitSandboxRagPrep: async (
    text: string,
    profileId?: string | null
  ): Promise<{ jobId: string; status: string }> => {
    const { data } = await api.post('/ai/admin/sandbox/rag-prep', { text, profileId });
    return extractData<{ jobId: string; status: string }>(data);
  },
};

export interface OcrEngineResponse {
  engineId: string;
  engineName: string;
  engineType: string;
  isActive: boolean;
  isCurrentActive: boolean;
  vramRequirementMB: number;
  processingTimeLimitSeconds: number;
  concurrentLimit: number;
  fallbackEngineId?: string | null;
  createdAt: string;
  updatedAt: string;
}