rird 1.0.200

package/src/tool/codesearch.ts

@@ -0,0 +1,168 @@
+import z from "zod"
+import { Tool } from "./tool"
+import DESCRIPTION from "./codesearch.txt"
+import { Config } from "../config/config"
+import { Permission } from "../permission"
+
+const API_CONFIG = {
+  BASE_URL: "https://mcp.exa.ai",
+  ENDPOINTS: {
+    CONTEXT: "/mcp",
+  },
+} as const
+
+interface McpCodeRequest {
+  jsonrpc: string
+  id: number
+  method: string
+  params: {
+    name: string
+    arguments: {
+      query: string
+      tokensNum: number
+    }
+  }
+}
+
+interface McpCodeResponse {
+  jsonrpc: string
+  result: {
+    content: Array<{
+      type: string
+      text: string
+    }>
+  }
+}
+
+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'",
+      ),
+    tokensNum: z
+      .number()
+      .min(1000)
+      .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.",
+      ),
+  }),
+  async execute(params, ctx) {
+    const cfg = await Config.get()
+    if (cfg.permission?.webfetch === "ask")
+      await Permission.ask({
+        type: "codesearch",
+        sessionID: ctx.sessionID,
+        messageID: ctx.messageID,
+        callID: ctx.callID,
+        title: "Search code for: " + params.query,
+        metadata: {
+          query: params.query,
+          tokensNum: params.tokensNum,
+        },
+      })
+
+    const codeRequest: McpCodeRequest = {
+      jsonrpc: "2.0",
+      id: 1,
+      method: "tools/call",
+      params: {
+        name: "get_code_context_exa",
+        arguments: {
+          query: params.query,
+          tokensNum: params.tokensNum || 5000,
+        },
+      },
+    }
+
+    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",
+      }
+
+      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)
+
+      if (!response.ok) {
+        const errorText = await response.text()
+        throw new Error(`Code search error (${response.status}): ${errorText}`)
+      }
+
+      const responseText = await response.text()
+
+      // Parse SSE response
+      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) {
+            return {
+              output: data.result.content[0].text,
+              title: `RIRD Code Index: ${params.query}`,
+              metadata: { source: "rird-code" },
+            }
+          }
+        }
+      }
+
+      return {
+        output:
+          "No code snippets found in RIRD index. Please try a different query.",
+        title: `RIRD Code Index: ${params.query}`,
+        metadata: { source: "rird-code" },
+      }
+    } catch (error) {
+      clearTimeout(timeoutId)
+
+      // Fallback to local RIRD Index (Option B)
+      try {
+        console.log("Primary code index unavailable, attempting local RIRD index...")
+        const searxUrl = "http://localhost:8080/search"
+        const searxParams = new URLSearchParams({
+          q: params.query + " code examples", // Optimize for code
+          format: "json",
+          categories: "it", // Focus on IT/Tech category if available
+          safesearch: "0"
+        })
+        
+        const searxResponse = await fetch(`${searxUrl}?${searxParams.toString()}`)
+        if (searxResponse.ok) {
+          const data = await searxResponse.json()
+          if (data.results && data.results.length > 0) {
+            const formattedResults = data.results.map((r: any) => 
+              `[${r.title}](${r.url})\n${r.content}`
+            ).join("\n\n")
+            
+            return {
+              output: formattedResults,
+              title: `RIRD Code Index: ${params.query}`,
+              metadata: { source: "rird-local-code" }
+            }
+          }
+        }
+      } catch (fallbackError) {
+        console.error("RIRD code index fallback failed:", fallbackError)
+      }
+
+      if (error instanceof Error && error.name === "AbortError") {
+        throw new Error("Code search request timed out")
+      }
+
+      throw error
+    }
+  },
+})

package/src/tool/codesearch.txt

@@ -0,0 +1,12 @@
+- Search and get relevant context for any task using Exa API
+- Provides the highest quality and freshest context for tools, services, and APIs
+- Use this tool for ANY question or task that requires external knowledge
+- Returns comprehensive examples, documentation, and API references
+- Optimized for finding specific patterns and solutions
+
+Usage notes:
+  - Adjustable token count (1000-50000) for focused or comprehensive results
+  - Default 5000 tokens provides balanced context for most queries
+  - Use lower values for specific questions, higher values for comprehensive documentation
+  - Supports queries about tools, services, APIs, and general concepts
+  - Examples: 'browser automation examples', 'data extraction patterns', 'API integration methods'