@link-assistant/agent 0.0.8 → 0.0.11

package/src/tool/codesearch.ts

@@ -1,44 +1,44 @@
-import z from "zod"
-import { Tool } from "./tool"
-import DESCRIPTION from "./codesearch.txt"
+import z from 'zod';
+import { Tool } from './tool';
+import DESCRIPTION from './codesearch.txt';
 
 const API_CONFIG = {
-  BASE_URL: "https://mcp.exa.ai",
+  BASE_URL: 'https://mcp.exa.ai',
   ENDPOINTS: {
-    CONTEXT: "/mcp",
+    CONTEXT: '/mcp',
   },
-} as const
+} as const;
 
 interface McpCodeRequest {
-  jsonrpc: string
-  id: number
-  method: string
+  jsonrpc: string;
+  id: number;
+  method: string;
   params: {
-    name: string
+    name: string;
     arguments: {
-      query: string
-      tokensNum: number
-    }
-  }
+      query: string;
+      tokensNum: number;
+    };
+  };
 }
 
 interface McpCodeResponse {
-  jsonrpc: string
+  jsonrpc: string;
   result: {
     content: Array<{
-      type: string
-      text: string
-    }>
-  }
+      type: string;
+      text: string;
+    }>;
+  };
 }
 
-export const CodeSearchTool = Tool.define("codesearch", {
+export const CodeSearchTool = Tool.define('codesearch', {
   description: DESCRIPTION,
   parameters: z.object({
     query: z
       .string()
       .describe(
-        "Search query to find relevant context for APIs, Libraries, and SDKs. For example, 'React useState hook examples', 'Python pandas dataframe filtering', 'Express.js middleware', 'Next js partial prerendering configuration'",
+        "Search query to find relevant context for APIs, Libraries, and SDKs. For example, 'React useState hook examples', 'Python pandas dataframe filtering', 'Express.js middleware', 'Next js partial prerendering configuration'"
       ),
     tokensNum: z
       .number()
@@ -46,78 +46,85 @@ export const CodeSearchTool = Tool.define("codesearch", {
       .max(50000)
       .default(5000)
       .describe(
-        "Number of tokens to return (1000-50000). Default is 5000 tokens. Adjust this value based on how much context you need - use lower values for focused queries and higher values for comprehensive documentation.",
+        'Number of tokens to return (1000-50000). Default is 5000 tokens. Adjust this value based on how much context you need - use lower values for focused queries and higher values for comprehensive documentation.'
       ),
   }),
   async execute(params, ctx) {
     // No restrictions - unrestricted code search
     const codeRequest: McpCodeRequest = {
-      jsonrpc: "2.0",
+      jsonrpc: '2.0',
       id: 1,
-      method: "tools/call",
+      method: 'tools/call',
       params: {
-        name: "get_code_context_exa",
+        name: 'get_code_context_exa',
         arguments: {
           query: params.query,
           tokensNum: params.tokensNum || 5000,
         },
       },
-    }
+    };
 
-    const controller = new AbortController()
-    const timeoutId = setTimeout(() => controller.abort(), 30000)
+    const controller = new AbortController();
+    const timeoutId = setTimeout(() => controller.abort(), 30000);
 
     try {
       const headers: Record<string, string> = {
-        accept: "application/json, text/event-stream",
-        "content-type": "application/json",
-      }
+        accept: 'application/json, text/event-stream',
+        'content-type': 'application/json',
+      };
 
-      const response = await fetch(`${API_CONFIG.BASE_URL}${API_CONFIG.ENDPOINTS.CONTEXT}`, {
-        method: "POST",
-        headers,
-        body: JSON.stringify(codeRequest),
-        signal: AbortSignal.any([controller.signal, ctx.abort]),
-      })
+      const response = await fetch(
+        `${API_CONFIG.BASE_URL}${API_CONFIG.ENDPOINTS.CONTEXT}`,
+        {
+          method: 'POST',
+          headers,
+          body: JSON.stringify(codeRequest),
+          signal: AbortSignal.any([controller.signal, ctx.abort]),
+        }
+      );
 
-      clearTimeout(timeoutId)
+      clearTimeout(timeoutId);
 
       if (!response.ok) {
-        const errorText = await response.text()
-        throw new Error(`Code search error (${response.status}): ${errorText}`)
+        const errorText = await response.text();
+        throw new Error(`Code search error (${response.status}): ${errorText}`);
       }
 
-      const responseText = await response.text()
+      const responseText = await response.text();
 
       // Parse SSE response
-      const lines = responseText.split("\n")
+      const lines = responseText.split('\n');
       for (const line of lines) {
-        if (line.startsWith("data: ")) {
-          const data: McpCodeResponse = JSON.parse(line.substring(6))
-          if (data.result && data.result.content && data.result.content.length > 0) {
+        if (line.startsWith('data: ')) {
+          const data: McpCodeResponse = JSON.parse(line.substring(6));
+          if (
+            data.result &&
+            data.result.content &&
+            data.result.content.length > 0
+          ) {
             return {
               output: data.result.content[0].text,
               title: `Code search: ${params.query}`,
               metadata: {},
-            }
+            };
           }
         }
       }
 
       return {
         output:
-          "No code snippets or documentation found. Please try a different query, be more specific about the library or programming concept, or check the spelling of framework names.",
+          'No code snippets or documentation found. Please try a different query, be more specific about the library or programming concept, or check the spelling of framework names.',
         title: `Code search: ${params.query}`,
         metadata: {},
-      }
+      };
     } catch (error) {
-      clearTimeout(timeoutId)
+      clearTimeout(timeoutId);
 
-      if (error instanceof Error && error.name === "AbortError") {
-        throw new Error("Code search request timed out")
+      if (error instanceof Error && error.name === 'AbortError') {
+        throw new Error('Code search request timed out');
       }
 
-      throw error
+      throw error;
     }
   },
-})
+});