logger updates

This commit is contained in:
Storme-bit
2026-04-26 22:28:54 -07:00
parent c86b565eed
commit 86e78cc4c6
23 changed files with 87 additions and 67 deletions

View File

@@ -1,5 +1,6 @@
const { Router } = require('express');
const { complete, completeStream } = require('../infer');
const { logger } = require('@nexusai/shared');
const router = Router();
@@ -15,7 +16,7 @@ router.post('/complete', async (req, res) => {
const result = await complete (prompt, {model, temperature, maxTokens, topP, topK, repeatPenalty});
res.json(result);
} catch (error) {
console.error('[Inference] Completion error:', error.message);
logger.error('[Inference] Completion error:', error.message);
res.status(500).json({ error: error.message });
}
});
@@ -42,7 +43,7 @@ router.post('/complete/stream', async (req, res) => {
// capture final metadata from the done signal
lastModel = chunk.model ?? lastModel;
tokenCount = chunk.tokenCount ?? tokenCount;
console.log('[inference router] tokenCount from chunk:', chunk.tokenCount, '→', tokenCount);
logger.info('[inference router] tokenCount from chunk:', chunk.tokenCount, '→', tokenCount);
}
}
@@ -51,7 +52,7 @@ router.post('/complete/stream', async (req, res) => {
res.write('data: [DONE]\n\n');
} catch (err) {
console.error('[Inference] Streaming error:', err.message);
logger.error('[Inference] Streaming error:', err.message);
res.write(`data: ${JSON.stringify({ error: err.message })}\n\n`);
} finally {
res.end();