refactor

pages/api/functions/index.ts

@@ -103,23 +103,29 @@ const serpEmbedApi = async ({ input }: any) => {
   const queryResult = await vectorStore.similaritySearch(input, VECTOR_STORE_SIZE);
   return queryResult;
 }
+const handlers: any = {
+  searchApi: serpEmbedApi,
+  surferEmbedApi: surferEmbedApi
+};
 
 export default async function handler(req: NextApiRequest, res: NextApiResponse) {
   const args = req.body.args as string;
   const functionName = req.body.name as string;
   const functionInput = JSON.parse(args);
   
+  const functionHandler = handlers[functionName];
+  
+  if (!functionHandler) {
+    console.error(`Function "${functionName}" is not supported.`);
+    return res.status(500).json({ error: `Function "${functionName}" is not supported.` });
+  }
+  
   try {
-    if (functionName === 'searchApi') {
-      const result = await serpEmbedApi(functionInput);
-      return res.status(200).send(result);
-    } else {
-      const result = await surferEmbedApi(functionInput)
-      return res.status(200).send(result);
-    }
+    const result = await functionHandler(functionInput);
+    return res.status(200).send(result);
   } catch (error) {
     console.error(error);
     // @ts-ignore
     return res.status(500).json({ error: error.message });
   }
-}
+}