opencode-v2 1.1.53

package/src/tool/chat.ts

@@ -0,0 +1,217 @@
+import z from "zod"
+import { Tool } from "./tool"
+import { Question } from "../question"
+import { Session } from "../session"
+import { MessageV2 } from "../session/message-v2"
+import { Identifier } from "../id/id"
+import { Provider } from "../provider/provider"
+import CHAT_EXIT_DESCRIPTION from "./chat-exit.txt"
+import CHAT_ENTER_DESCRIPTION from "./chat-enter.txt"
+
+async function getLastModel(sessionID: string) {
+  for await (const item of MessageV2.stream(sessionID)) {
+    if (item.info.role === "user" && item.info.model) return item.info.model
+  }
+  return Provider.defaultModel()
+}
+
+export const ChatExitTool = Tool.define("chat_exit", {
+  description: CHAT_EXIT_DESCRIPTION,
+  parameters: z.object({}),
+  async execute(_params, ctx) {
+    const session = await Session.get(ctx.sessionID)
+
+    const answers = await Question.ask({
+      sessionID: ctx.sessionID,
+      questions: [
+        {
+          question: "You are currently in chat mode. Which mode would you like to switch to?",
+          header: "Exit Chat Mode",
+          custom: false,
+          options: [
+            { label: "Agent", description: "Switch to agent mode for unified planning and execution" },
+            { label: "Research", description: "Switch to research agent for systematic web research" },
+          ],
+        },
+      ],
+      tool: ctx.callID ? { messageID: ctx.messageID, callID: ctx.callID } : undefined,
+    })
+
+    const answer = answers[0]?.[0]
+    if (!answer) throw new Question.RejectedError()
+
+    const targetAgent = answer.toLowerCase()
+    const model = await getLastModel(ctx.sessionID)
+
+    const userMsg: MessageV2.User = {
+      id: Identifier.ascending("message"),
+      sessionID: ctx.sessionID,
+      role: "user",
+      time: {
+        created: Date.now(),
+      },
+      agent: targetAgent,
+      model,
+    }
+    await Session.updateMessage(userMsg)
+    await Session.updatePart({
+      id: Identifier.ascending("part"),
+      messageID: userMsg.id,
+      sessionID: ctx.sessionID,
+      type: "text",
+      text: `User has requested to exit chat mode and switch to ${targetAgent} mode.`,
+      synthetic: true,
+    } satisfies MessageV2.TextPart)
+
+    return {
+      title: `Switching to ${targetAgent} agent`,
+      output: `User confirmed to switch to ${targetAgent} mode. A new message has been created to switch you to ${targetAgent} mode.`,
+      metadata: {},
+    }
+  },
+})
+
+export const ChatEnterTool = Tool.define("chat_enter", {
+  description: CHAT_ENTER_DESCRIPTION,
+  parameters: z.object({}),
+  async execute(_params, ctx) {
+    const session = await Session.get(ctx.sessionID)
+    const messages = await Session.messages({ sessionID: session.id })
+
+    const hasAssistantMessages = messages.some((msg) => msg.info.role === "assistant")
+
+    if (hasAssistantMessages) {
+      return {
+        title: "Cannot enter chat mode",
+        output:
+          "Chat mode can only be entered when starting a new conversation, not in an existing thread. You can use chat_exit to leave chat mode if you're already in it.",
+        metadata: {},
+      }
+    }
+
+    const answers = await Question.ask({
+      sessionID: ctx.sessionID,
+      questions: [
+        {
+          question: "Would you like to switch to chat mode for general conversation without file system access?",
+          header: "Chat Mode",
+          custom: false,
+          options: [
+            { label: "Yes", description: "Switch to chat agent for general conversation" },
+            { label: "No", description: "Stay with current agent" },
+          ],
+        },
+      ],
+      tool: ctx.callID ? { messageID: ctx.messageID, callID: ctx.callID } : undefined,
+    })
+
+    const answer = answers[0]?.[0]
+
+    if (answer === "No") throw new Question.RejectedError()
+
+    const model = await getLastModel(ctx.sessionID)
+
+    const userMsg: MessageV2.User = {
+      id: Identifier.ascending("message"),
+      sessionID: ctx.sessionID,
+      role: "user",
+      time: {
+        created: Date.now(),
+      },
+      agent: "chat",
+      model,
+    }
+    await Session.updateMessage(userMsg)
+    await Session.updatePart({
+      id: Identifier.ascending("part"),
+      messageID: userMsg.id,
+      sessionID: ctx.sessionID,
+      type: "text",
+      text: "User has requested to enter chat mode. Switch to chat mode and begin general conversation. You do not have access to the file system or any file-related tools. You are a general-purpose chatbot.",
+      synthetic: true,
+    } satisfies MessageV2.TextPart)
+
+    return {
+      title: "Switching to chat agent",
+      output: "User confirmed to switch to chat mode. A new message has been created to switch you to chat mode.",
+      metadata: {},
+    }
+  },
+})
+
+export const ModeCycleTool = Tool.define("mode_cycle", {
+  description:
+    "Cycle through available modes: Agent -> Chat -> Research. This tool is only available when starting a new conversation (not in an existing thread).",
+  parameters: z.object({}),
+  async execute(_params, ctx) {
+    const session = await Session.get(ctx.sessionID)
+    const messages = await Session.messages({ sessionID: session.id })
+
+    const lastAssistant = messages.findLast((msg) => msg.info.role === "assistant")
+    const currentMode = lastAssistant?.info.agent ?? "build"
+
+    if (messages.some((msg) => msg.info.role === "assistant")) {
+      return {
+        title: "Mode cycling unavailable",
+        output:
+          "Mode cycling is only available when starting a new conversation. Once a conversation has started, use chat_enter to enter chat mode or chat_exit to leave it.",
+        metadata: {},
+      }
+    }
+
+    const cycle = {
+      agent: "chat",
+      chat: "research",
+      research: "agent",
+    } as const
+
+    const nextMode = cycle[currentMode as keyof typeof cycle] ?? "agent"
+
+    const answers = await Question.ask({
+      sessionID: ctx.sessionID,
+      questions: [
+        {
+          question: `Cycle from ${currentMode} mode to ${nextMode} mode?`,
+          header: "Mode Cycle",
+          custom: false,
+          options: [
+            { label: "Yes", description: `Switch to ${nextMode} mode` },
+            { label: "No", description: "Stay in current mode" },
+          ],
+        },
+      ],
+      tool: ctx.callID ? { messageID: ctx.messageID, callID: ctx.callID } : undefined,
+    })
+
+    const answer = answers[0]?.[0]
+    if (answer === "No") throw new Question.RejectedError()
+
+    const model = await getLastModel(ctx.sessionID)
+
+    const userMsg: MessageV2.User = {
+      id: Identifier.ascending("message"),
+      sessionID: ctx.sessionID,
+      role: "user",
+      time: {
+        created: Date.now(),
+      },
+      agent: nextMode,
+      model,
+    }
+    await Session.updateMessage(userMsg)
+    await Session.updatePart({
+      id: Identifier.ascending("part"),
+      messageID: userMsg.id,
+      sessionID: ctx.sessionID,
+      type: "text",
+      text: `User has cycled to ${nextMode} mode.`,
+      synthetic: true,
+    } satisfies MessageV2.TextPart)
+
+    return {
+      title: `Switched to ${nextMode} mode`,
+      output: `Successfully switched to ${nextMode} mode.`,
+      metadata: {},
+    }
+  },
+})

package/src/tool/codesearch.ts

@@ -0,0 +1,132 @@
+import z from "zod"
+import { Tool } from "./tool"
+import DESCRIPTION from "./codesearch.txt"
+import { abortAfterAny } from "../util/abort"
+
+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) {
+    await ctx.ask({
+      permission: "codesearch",
+      patterns: [params.query],
+      always: ["*"],
+      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 { signal, clearTimeout } = abortAfterAny(30000, ctx.abort)
+
+    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,
+      })
+
+      clearTimeout()
+
+      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: `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.",
+        title: `Code search: ${params.query}`,
+        metadata: {},
+      }
+    } catch (error) {
+      clearTimeout()
+
+      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 programming task using Exa Code API
+- Provides the highest quality and freshest context for libraries, SDKs, and APIs
+- Use this tool for ANY question or task related to programming
+- Returns comprehensive code examples, documentation, and API references
+- Optimized for finding specific programming 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 frameworks, libraries, APIs, and programming concepts
+  - Examples: 'React useState hook examples', 'Python pandas dataframe filtering', 'Express.js middleware'