diff --git a/logic/classification.js b/logic/classification.js new file mode 100644 index 0000000..4b565b8 --- /dev/null +++ b/logic/classification.js @@ -0,0 +1,44 @@ +const VALID_CLASSIFICATIONS = [ + 'ACTIONABLE_NOW', 'UNASSIGNED_ACTIONABLE', 'WAITING_REQUESTER', 'WAITING_VENDOR', + 'WAITING_APPROVAL', 'SCHEDULED', 'DEPENDENCY_BLOCKED', 'NEEDS_COORDINATION', + 'PROJECT_LONG_TERM', 'DUPLICATE_OR_OBSOLETE', 'INSUFFICIENT_CONTEXT', +]; + +function validateAiClassification(raw) { + const errors = []; + if (!raw || typeof raw !== 'object') { + return { valid: false, data: null, errors: ['resposta nao e um objeto JSON'] }; + } + if (!VALID_CLASSIFICATIONS.includes(raw.classification)) { + errors.push(`classification invalida: ${raw.classification}`); + } + if (typeof raw.actionable_today !== 'boolean') errors.push('actionable_today deve ser boolean'); + const confidence = Number(raw.confidence); + if (!Number.isFinite(confidence) || confidence < 0 || confidence > 1) errors.push('confidence deve estar entre 0 e 1'); + if (typeof raw.next_action !== 'string' || raw.next_action.length > 200) errors.push('next_action invalido'); + if (raw.evidence != null && !Array.isArray(raw.evidence)) errors.push('evidence deve ser array'); + if (typeof raw.reason !== 'string') errors.push('reason deve ser string'); + + if (errors.length) return { valid: false, data: null, errors }; + + return { + valid: true, + errors: [], + data: { + classification: raw.classification, + actionable_today: raw.actionable_today, + confidence, + blocker_type: raw.blocker_type ?? null, + waiting_for: raw.waiting_for ?? null, + next_action: raw.next_action, + suggested_action_type: raw.suggested_action_type ?? 'technical_action', + business_risk: raw.business_risk ?? 'medium', + reason: raw.reason, + evidence: Array.isArray(raw.evidence) ? raw.evidence.slice(0, 5) : [], + is_fallback: false, + fallback_reason: null, + }, + }; +} + +module.exports = { VALID_CLASSIFICATIONS, validateAiClassification }; diff --git a/logic/classification.test.js b/logic/classification.test.js new file mode 100644 index 0000000..004e57b --- /dev/null +++ b/logic/classification.test.js @@ -0,0 +1,54 @@ +const { test } = require('node:test'); +const assert = require('node:assert/strict'); +const { validateAiClassification, VALID_CLASSIFICATIONS } = require('./classification'); + +function validRaw(overrides) { + return Object.assign({ + classification: 'ACTIONABLE_NOW', + actionable_today: true, + confidence: 0.91, + blocker_type: null, + waiting_for: null, + next_action: 'Validar equipamento e registrar resultado.', + suggested_action_type: 'technical_action', + business_risk: 'medium', + reason: 'Existe acao tecnica pendente.', + evidence: ['ultimo apontamento indica necessidade de validacao'], + }, overrides); +} + +test('objeto valido passa e normaliza campos', () => { + const result = validateAiClassification(validRaw({})); + assert.equal(result.valid, true); + assert.equal(result.data.classification, 'ACTIONABLE_NOW'); + assert.equal(result.data.is_fallback, false); +}); + +test('classification invalida falha com erro claro', () => { + const result = validateAiClassification(validRaw({ classification: 'ALGO_INVENTADO' })); + assert.equal(result.valid, false); + assert.ok(result.errors.some((e) => e.includes('classification'))); +}); + +test('JSON invalido/null nao lanca excecao', () => { + assert.doesNotThrow(() => validateAiClassification(null)); + assert.doesNotThrow(() => validateAiClassification('nao e objeto')); + assert.equal(validateAiClassification(null).valid, false); +}); + +test('confidence fora de 0..1 falha', () => { + const result = validateAiClassification(validRaw({ confidence: 1.5 })); + assert.equal(result.valid, false); +}); + +test('evidence e limitado a 5 itens', () => { + const result = validateAiClassification(validRaw({ evidence: ['a', 'b', 'c', 'd', 'e', 'f', 'g'] })); + assert.equal(result.data.evidence.length, 5); +}); + +test('todas as classes do dominio sao aceitas', () => { + for (const classification of VALID_CLASSIFICATIONS) { + const result = validateAiClassification(validRaw({ classification })); + assert.equal(result.valid, true, `classe ${classification} deveria ser valida`); + } +});