embed serpapi

pages/api/functions/index.ts

@@ -64,7 +64,7 @@ const handleContentText = async (targetUrl: string) => {
   return content;
 }
 
-const surferApi = async ({ input }: any) => {
+const surferEmbedApi = async ({ input }: any) => {
   const urls = input.match(urlRegex);
   const targetUrl = urls ? urls[0] : null;
   const promptWithoutUrl = urls ? input.replace(urlRegex, '').trim() : input;
@@ -87,16 +87,30 @@ const surferApi = async ({ input }: any) => {
   return `Here is the context: ${JSON.stringify(queryResult.map(result => result.pageContent))} from using the prompt to lookup relevant information. Here is the prompt: ${promptWithoutUrl}`;
 }
 
+const serpEmbedApi = async ({ input }: any) => {
+  const content: string = await serpApi({ input })
+  const documents = await textSplitter.createDocuments([content]);
+  const vectorStore = await MemoryVectorStore.fromTexts(
+    // @ts-ignore
+    [...documents.map(doc => doc.pageContent)],
+    // @ts-ignore
+    [...documents.map((v, k) => k)],
+    model
+  )
+  const queryResult = await vectorStore.similaritySearch(input, VECTOR_STORE_SIZE);
+  return queryResult;
+}
+
 export default async function handler(req: NextApiRequest, res: NextApiResponse) {
   const prompt = req.body.prompt as string;
   const functionName = req.body.name as string;
   
   try {
     if (functionName === 'serpApi') {
-      const result = await serpApi({ input: prompt });
+      const result = await serpEmbedApi({ input: prompt });
       return res.status(200).send(result);
     } else {
-      const result = await surferApi({ input: prompt })
+      const result = await surferEmbedApi({ input: prompt })
       return res.status(200).send(result);
     }
   } catch (error) {