Browse Source

temporarily restrict codesearch and websearch to opencode zen users. need to figure out how to opt out for enterprise users who do not want this

Dax Raad 2 months ago
parent
commit
9c237f0bfb
1 changed files with 10 additions and 5 deletions
  1. 10 5
      packages/opencode/src/tool/registry.ts

+ 10 - 5
packages/opencode/src/tool/registry.ts

@@ -108,13 +108,18 @@ export namespace ToolRegistry {
     return all().then((x) => x.map((t) => t.id))
   }
 
-  export async function tools(_providerID: string, _modelID: string) {
+  export async function tools(providerID: string, _modelID: string) {
     const tools = await all()
     const result = await Promise.all(
-      tools.map(async (t) => ({
-        id: t.id,
-        ...(await t.init()),
-      })),
+      tools
+        .filter((t) => {
+          if (t.id === "codesearch" || t.id === "websearch") return providerID === "opencode"
+          return true
+        })
+        .map(async (t) => ({
+          id: t.id,
+          ...(await t.init()),
+        })),
     )
     return result
   }