❤

src/tools/diffbot.ts

@@ -104,9 +104,9 @@ export class DiffbotTool extends BaseTask<DiffbotInput, DiffbotOutput> {
       url: ctx.input!.url
     })
 
-    this._logger.info(res, `Diffbot response for url "${ctx.input!.url}"`)
+    // this._logger.info(res, `Diffbot response for url "${ctx.input!.url}"`)
 
-    const pickedRes = {
+    const output = this.outputSchema.parse({
       type: res.type,
       title: res.title,
       objects: res.objects.map((obj) => ({
@@ -126,12 +126,9 @@ export class DiffbotTool extends BaseTask<DiffbotInput, DiffbotOutput> {
         // tags: obj.tags?.map((tag) => tag.label)
         // images: obj.images?.map((image) => omit(image, 'diffbotUri'))
       }))
-    }
+    })
 
-    this._logger.info(
-      pickedRes,
-      `Diffbot picked response for url "${ctx.input!.url}"`
-    )
-    return this.outputSchema.parse(pickedRes)
+    this._logger.info(output, `Diffbot response for url "${ctx.input!.url}"`)
+    return output
   }
 }

src/tools/search-and-crawl.ts

@@ -166,13 +166,12 @@ export class SearchAndCrawlTool extends BaseTask<
       )
     ).flat()
 
-    const result = {
+    const output = this.outputSchema.parse({
       ...omit(search.result, 'organic_results'),
       scrape_results: scrapeResults
-    }
+    })
 
-    this._logger.info(result, `SearchAndCrawl response for query "${query}"`)
-
-    return this.outputSchema.parse(result)
+    this._logger.info(output, `SearchAndCrawl response for query "${query}"`)
+    return output
   }
 }

src/tools/serpapi.ts

@@ -103,11 +103,6 @@ export class SerpAPITool extends BaseTask<SerpAPIInput, SerpAPIOutput> {
       // results manuall
     })
 
-    this._logger.info(
-      res,
-      `SerpAPI response for query ${JSON.stringify(ctx.input, null, 2)}"`
-    )
-
     const twitterResults = res.twitter_results
       ? {
           ...res.twitter_results,
@@ -120,11 +115,15 @@ export class SerpAPITool extends BaseTask<SerpAPIInput, SerpAPIOutput> {
         }
       : undefined
 
-    return this.outputSchema.parse({
+    const output = this.outputSchema.parse({
       knowledge_graph: res.knowledge_graph,
       answer_box: res.answer_box,
       organic_results: res.organic_results?.slice(0, numResults),
       twitter_results: twitterResults
     })
+
+    this._logger.info(output, `SerpAPI response for query "${query}"`)
+
+    return output
   }
 }