diff --git a/src/agent.ts b/src/agent.ts index c491f2a..e03f25e 100644 --- a/src/agent.ts +++ b/src/agent.ts @@ -267,7 +267,7 @@ async function updateReferences(thisStep: AnswerAction, allURLs: Record { const authHeader = req.headers.authorization; if (!authHeader || !authHeader.startsWith('Bearer ') || authHeader.split(' ')[1] !== secret) { - console.log('[chat/completions] Unauthorized request'); + logError('[chat/completions] Unauthorized request'); res.status(401).json({ error: 'Unauthorized' }); return; } @@ -380,7 +380,7 @@ app.post('/v1/chat/completions', (async (req: Request, res: Response) => { if (secret) { const authHeader = req.headers.authorization; if (!authHeader || !authHeader.startsWith('Bearer ') || authHeader.split(' ')[1] !== secret) { - console.log('[chat/completions] Unauthorized request'); + logError('[chat/completions] Unauthorized request'); res.status(401).json({ error: 'Unauthorized' }); return; } @@ -392,7 +392,7 @@ app.post('/v1/chat/completions', (async (req: Request, res: Response) => { req.socket.remoteAddress || 'unknown'; // Log request details (excluding sensitive data) - console.log('[chat/completions] Request:', { + logInfo('[chat/completions] Request:', { model: req.body.model, stream: req.body.stream, messageCount: req.body.messages?.length, @@ -410,7 +410,7 @@ app.post('/v1/chat/completions', (async (req: Request, res: Response) => { return res.status(400).json({ error: 'Last message must be from user' }); } - console.log('messages', JSON.stringify(body.messages)); + logDebug('Input messages', { messages: body.messages }); // clean from all assistant messages body.messages = body.messages?.filter(message => { diff --git a/src/tools/jina-dedup.ts b/src/tools/jina-dedup.ts index 26f4e73..8cc9fd6 100644 --- a/src/tools/jina-dedup.ts +++ b/src/tools/jina-dedup.ts @@ -1,6 +1,7 @@ -import {TokenTracker} from "../utils/token-tracker"; -import {cosineSimilarity} from "./cosine"; -import {getEmbeddings} from "./embeddings"; +import { logDebug, logError, logInfo } from "../logging"; +import { TokenTracker } from "../utils/token-tracker"; +import { cosineSimilarity } from "./cosine"; +import { getEmbeddings } from "./embeddings"; const SIMILARITY_THRESHOLD = 0.86; // Adjustable threshold for cosine similarity @@ -20,7 +21,7 @@ export async function dedupQueries( // Get embeddings for all queries in one batch const allQueries = [...newQueries, ...existingQueries]; - const {embeddings: allEmbeddings} = await getEmbeddings(allQueries, tracker); + const { embeddings: allEmbeddings } = await getEmbeddings(allQueries, tracker); // If embeddings is empty (due to 402 error), return all new queries if (!allEmbeddings.length) { @@ -66,12 +67,12 @@ export async function dedupQueries( usedIndices.add(i); } } - console.log('Dedup:', uniqueQueries); + logInfo('Dedup:', { uniqueQueries }); return { unique_queries: uniqueQueries, }; } catch (error) { - console.error('Error in deduplication analysis:', error); + logError('Error in deduplication analysis:', { error }); // return all new queries if there is an error return {