sk/Perplexica
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat(logging): add logger

Browse source
This commit is contained in:
ItzCrazyKns 2024-04-30 12:18:18 +05:30
parent 7c84025f3c
commit aae85cd767
No known key found for this signature in database
GPG key ID: 8162927C7CCE3065
15 changed files with 240 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

3
src/websocket/connectionManager.ts
View file

@ -4,6 +4,7 @@ import { getChatModel, getChatModelProvider } from '../config';
import { getAvailableProviders } from '../lib/providers';
import { BaseChatModel } from '@langchain/core/language_models/chat_models';
import type { Embeddings } from '@langchain/core/embeddings';
import logger from '../utils/logger';
export const handleConnection = async (ws: WebSocket) => {
const models = await getAvailableProviders();
@ -34,5 +35,5 @@ export const handleConnection = async (ws: WebSocket) => {
await handleMessage(message.toString(), ws, llm, embeddings),
);
ws.on('close', () => console.log('Connection closed'));
ws.on('close', () => logger.debug('Connection closed'));
};

5
src/websocket/messageHandler.ts
View file

@ -8,6 +8,7 @@ import handleYoutubeSearch from '../agents/youtubeSearchAgent';
import handleRedditSearch from '../agents/redditSearchAgent';
import type { BaseChatModel } from '@langchain/core/language_models/chat_models';
import type { Embeddings } from '@langchain/core/embeddings';
import logger from '../utils/logger';
type Message = {
type: string;
@ -101,8 +102,8 @@ export const handleMessage = async (
ws.send(JSON.stringify({ type: 'error', data: 'Invalid focus mode' }));
}
}
} catch (error) {
console.error('Failed to handle message', error);
} catch (err) {
ws.send(JSON.stringify({ type: 'error', data: 'Invalid message format' }));
logger.error(`Failed to handle message: ${err}`);
}
};

3
src/websocket/websocketServer.ts
View file

@ -2,6 +2,7 @@ import { WebSocketServer } from 'ws';
import { handleConnection } from './connectionManager';
import http from 'http';
import { getPort } from '../config';
import logger from '../utils/logger';
export const initServer = (
server: http.Server<typeof http.IncomingMessage, typeof http.ServerResponse>,
@ -13,5 +14,5 @@ export const initServer = (
handleConnection(ws);
});
console.log(`WebSocket server started on port ${port}`);
logger.info(`WebSocket server started on port ${port}`);
};