From 9e67fbfaae44cc05df55a1fcfe8b97912d3c0e87 Mon Sep 17 00:00:00 2001
From: dal <dallin@buster.so>
Date: Wed, 23 Jul 2025 13:12:56 -0600
Subject: [PATCH] refactor: update web search tool integration tests and
 improve code formatting

- Changed test query from 'OpenAI GPT' to 'Buster Data' with an increased limit of 10.
- Reorganized import statements in web-search-tool.ts for better readability.
- Enhanced formatting in permission-validator.test.ts for improved code clarity.
---
 .../ai/src/tools/web-tools/web-search-tool.int.test.ts    | 4 ++--
 packages/ai/src/tools/web-tools/web-search-tool.ts        | 8 ++++----
 .../utils/sql-permissions/permission-validator.test.ts    | 5 ++++-
 packages/web-tools/src/services/firecrawl.ts              | 5 +----
 4 files changed, 11 insertions(+), 11 deletions(-)

diff --git a/packages/ai/src/tools/web-tools/web-search-tool.int.test.ts b/packages/ai/src/tools/web-tools/web-search-tool.int.test.ts
index 9f91ceadf..1d3546516 100644
--- a/packages/ai/src/tools/web-tools/web-search-tool.int.test.ts
+++ b/packages/ai/src/tools/web-tools/web-search-tool.int.test.ts
@@ -13,8 +13,8 @@ describe('webSearch tool integration', () => {
     async () => {
       const result = await webSearch.execute({
         context: {
-          query: 'OpenAI GPT',
-          limit: 3,
+          query: 'Buster Data',
+          limit: 10,
           scrapeContent: true,
           formats: ['markdown'],
         },
diff --git a/packages/ai/src/tools/web-tools/web-search-tool.ts b/packages/ai/src/tools/web-tools/web-search-tool.ts
index 8a58d8a8b..ca8b29dfe 100644
--- a/packages/ai/src/tools/web-tools/web-search-tool.ts
+++ b/packages/ai/src/tools/web-tools/web-search-tool.ts
@@ -1,12 +1,12 @@
-import { createTool } from '@mastra/core/tools';
-import type { RuntimeContext } from '@mastra/core/runtime-context';
-import { wrapTraced } from 'braintrust';
-import { z } from 'zod';
 import {
   FirecrawlService,
   type WebSearchOptions,
   type WebSearchResult,
 } from '@buster-tools/web-tools';
+import type { RuntimeContext } from '@mastra/core/runtime-context';
+import { createTool } from '@mastra/core/tools';
+import { wrapTraced } from 'braintrust';
+import { z } from 'zod';
 
 const inputSchema = z.object({
   query: z.string().min(1, 'Search query is required').describe('The search query to execute'),
diff --git a/packages/ai/src/utils/sql-permissions/permission-validator.test.ts b/packages/ai/src/utils/sql-permissions/permission-validator.test.ts
index 1dce6de07..569f2247b 100644
--- a/packages/ai/src/utils/sql-permissions/permission-validator.test.ts
+++ b/packages/ai/src/utils/sql-permissions/permission-validator.test.ts
@@ -51,7 +51,10 @@ describe('Permission Validator', () => {
         },
       ] as any);
 
-      const result = await validateSqlPermissions('SELECT id, user_id FROM public.orders', 'user123');
+      const result = await validateSqlPermissions(
+        'SELECT id, user_id FROM public.orders',
+        'user123'
+      );
 
       expect(result).toEqual({
         isAuthorized: false,
diff --git a/packages/web-tools/src/services/firecrawl.ts b/packages/web-tools/src/services/firecrawl.ts
index aaf9a3d00..afa9f1551 100644
--- a/packages/web-tools/src/services/firecrawl.ts
+++ b/packages/web-tools/src/services/firecrawl.ts
@@ -196,10 +196,7 @@ export class FirecrawlService {
   /**
    * Search the web using Firecrawl's search endpoint
    */
-  async webSearch(
-    query: string,
-    options?: WebSearchOptions
-  ): Promise<WebSearchResponse> {
+  async webSearch(query: string, options?: WebSearchOptions): Promise<WebSearchResponse> {
     try {
       const searchOptions = {
         limit: options?.limit || 5,