Fix logging statements and error handling in image search and Searxng API integration

src/lib/searxng.ts

@@ -21,25 +21,36 @@ export const searchSearxng = async (
   query: string,
   opts?: SearxngSearchOptions,
 ) => {
+  try {
     const searxngURL = getSearxngApiEndpoint();
+    console.log('Searxng API endpoint:', searxngURL);
 
     const url = new URL(`${searxngURL}/search?format=json`);
     url.searchParams.append('q', query);
+    console.log('Initial URL with query:', url.toString());
 
     if (opts) {
       Object.keys(opts).forEach((key) => {
         if (Array.isArray(opts[key])) {
           url.searchParams.append(key, opts[key].join(','));
-        return;
-      }
+        } else {
           url.searchParams.append(key, opts[key]);
+        }
+        console.log(`Added search option ${key}:`, opts[key]);
       });
     }
 
+    console.log('Final URL with all parameters:', url.toString());
     const res = await axios.get(url.toString());
+    console.log('API response received');
 
     const results: SearxngSearchResult[] = res.data.results;
     const suggestions: string[] = res.data.suggestions;
+    console.log('Results and suggestions extracted from the response');
 
     return { results, suggestions };
+  } catch (error) {
+    console.error('Error during Searxng search:', error);
+    throw error; // Re-throw the error after logging it
+  }
 };

src/routes/config.ts

@@ -11,6 +11,7 @@ import {
 const router = express.Router();
 
 router.get('/', async (_, res) => {
+  try {
     const config = {};
 
     const providers = await getAvailableProviders();
@@ -32,9 +33,14 @@ router.get('/', async (_, res) => {
     config['ollamaApiUrl'] = getOllamaApiEndpoint();
 
     res.status(200).json(config);
+  } catch (error) {
+    console.error('Failed to retrieve configuration:', error);
+    res.status(500).json({ message: 'Internal Server Error' });
+  }
 });
 
 router.post('/', async (req, res) => {
+  try {
     const config = req.body;
 
     const updatedConfig = {
@@ -53,6 +59,10 @@ router.post('/', async (req, res) => {
     updateConfig(updatedConfig);
 
     res.status(200).json({ message: 'Config updated' });
+  } catch (error) {
+    console.error('Failed to update configuration:', error);
+    res.status(500).json({ message: 'Internal Server Error' });
+  }
 });
 
 export default router;

src/routes/images.ts

@@ -11,6 +11,8 @@ router.post('/', async (req, res) => {
   try {
     let { query, chat_history } = req.body;
 
+    console.log('Received request with query:', query);
+
     chat_history = chat_history.map((msg: any) => {
       if (msg.role === 'user') {
         return new HumanMessage(msg.content);
@@ -19,10 +21,14 @@ router.post('/', async (req, res) => {
       }
     });
 
+    console.log('Processed chat history:', chat_history);
+
     const models = await getAvailableProviders();
     const provider = getChatModelProvider();
     const chatModel = getChatModel();
 
+    console.log(`Using provider: ${provider} and model: ${chatModel}`);
+
     let llm: BaseChatModel | undefined;
 
     if (models[provider] && models[provider][chatModel]) {
@@ -30,16 +36,22 @@ router.post('/', async (req, res) => {
     }
 
     if (!llm) {
+      console.error('Invalid LLM model selected');
       res.status(500).json({ message: 'Invalid LLM model selected' });
       return;
     }
 
+    try {
       const images = await handleImageSearch({ query, chat_history }, llm);
-
       res.status(200).json({ images });
+      console.log('Image search successful:', images);
+    } catch (error) {
+      console.error('Error during image search:', error);
+      res.status(500).json({ message: 'Error during image search' });
+    }
   } catch (err) {
+    console.error('An error occurred in the main request handler:', err);
     res.status(500).json({ message: 'An error has occurred.' });
-    console.log(err.message);
   }
 });
 

src/websocket/connectionManager.ts

@@ -6,6 +6,7 @@ import { BaseChatModel } from '@langchain/core/language_models/chat_models';
 import type { Embeddings } from '@langchain/core/embeddings';
 
 export const handleConnection = async (ws: WebSocket) => {
+  try {
     const models = await getAvailableProviders();
     const provider = getChatModelProvider();
     const chatModel = getChatModel();
@@ -18,21 +19,61 @@ export const handleConnection = async (ws: WebSocket) => {
       embeddings = models[provider].embeddings as Embeddings | undefined;
     }
 
-  if (!llm || !embeddings) {
+    // Separate checks for llm and embeddings
+    if (!llm && !embeddings) {
       ws.send(
         JSON.stringify({
           type: 'error',
-        data: 'Invalid LLM or embeddings model selected',
+          data: 'Invalid LLM and embeddings model selected',
+        }),
+      );
+      console.error('Invalid LLM and embeddings model selected');
+    } else if (!llm) {
+      ws.send(
+        JSON.stringify({
+          type: 'error',
+          data: 'Invalid LLM model selected',
+        }),
+      );
+      console.error('Invalid LLM model selected');
+    } else if (!embeddings) {
+      ws.send(
+        JSON.stringify({
+          type: 'error',
+          data: 'Invalid embeddings model selected',
+        }),
+      );
+      console.error('Invalid embeddings model selected');
+    }
+
+    if (!llm || !embeddings) {
+      ws.close();
+      return;
+    }
+
+    ws.on('message', async (message) => {
+      try {
+        await handleMessage(message.toString(), ws, llm, embeddings);
+      } catch (error) {
+        console.error('Error handling message:', error);
+        ws.send(
+          JSON.stringify({
+            type: 'error',
+            data: 'Error processing your message',
+          }),
+        );
+      }
+    });
+
+    ws.on('close', () => console.log('Connection closed'));
+  } catch (error) {
+    console.error('Failed to establish a connection:', error);
+    ws.send(
+      JSON.stringify({
+        type: 'error',
+        data: 'Failed to establish a connection',
       }),
     );
     ws.close();
   }
-
-  ws.on(
-    'message',
-    async (message) =>
-      await handleMessage(message.toString(), ws, llm, embeddings),
-  );
-
-  ws.on('close', () => console.log('Connection closed'));
 };

src/websocket/messageHandler.ts

@@ -66,14 +66,26 @@ export const handleMessage = async (
   llm: BaseChatModel,
   embeddings: Embeddings,
 ) => {
+  let parsedMessage: Message;
   try {
-    const parsedMessage = JSON.parse(message) as Message;
-    const id = Math.random().toString(36).substring(7);
-
-    if (!parsedMessage.content)
+    parsedMessage = JSON.parse(message) as Message;
+    console.log('Message parsed successfully:', parsedMessage);
+  } catch (error) {
+    console.error('Error parsing message:', message, error);
     return ws.send(
       JSON.stringify({ type: 'error', data: 'Invalid message format' }),
     );
+  }
+
+  const id = Math.random().toString(36).substring(7);
+  console.log('Generated message ID:', id);
+
+  if (!parsedMessage.content) {
+    console.log('Message content is empty');
+    return ws.send(
+      JSON.stringify({ type: 'error', data: 'Invalid message format' }),
+    );
+  }
 
   const history: BaseMessage[] = parsedMessage.history.map((msg) => {
     if (msg[0] === 'human') {
@@ -90,6 +102,8 @@ export const handleMessage = async (
   if (parsedMessage.type === 'message') {
     const handler = searchHandlers[parsedMessage.focusMode];
     if (handler) {
+      console.log('Handling message with focus mode:', parsedMessage.focusMode);
+      try {
         const emitter = handler(
           parsedMessage.content,
           history,
@@ -97,12 +111,17 @@ export const handleMessage = async (
           embeddings,
         );
         handleEmitterEvents(emitter, ws, id);
+      } catch (error) {
+        console.error(
+          'Error handling message with handler:',
+          parsedMessage.focusMode,
+          error,
+        );
+        ws.send(JSON.stringify({ type: 'error', data: 'Handler failure' }));
+      }
     } else {
+      console.log('Invalid focus mode:', parsedMessage.focusMode);
       ws.send(JSON.stringify({ type: 'error', data: 'Invalid focus mode' }));
     }
   }
-  } catch (error) {
-    console.error('Failed to handle message', error);
-    ws.send(JSON.stringify({ type: 'error', data: 'Invalid message format' }));
-  }
 };