import client from '../client';

export interface DialysisRiskRequest {
  patient_id: string;
  dialysis_session_id?: string;
}

export interface DialysisRiskAssessment {
  id: string;
  patient_id: string;
  risk_level: string;
  risk_factors: string[];
  recommendations: string[];
  kdigo_stage?: string;
  created_at: string;
}

export const dialysisRiskApi = {
  assess: async (data: DialysisRiskRequest) => {
    const resp = await client.post('/ai/dialysis/risk-assessment', data);
    return resp.data.data as DialysisRiskAssessment;
  },
};
