@polka-codes/runner 0.9.62 → 0.9.64

package/dist/index.js

@@ -21331,7 +21331,7 @@ var require_src6 = __commonJS((exports) => {
   var jws = require_jws();
   var path = __require("path");
   var util_1 = __require("util");
-  var readFile = fs.readFile ? (0, util_1.promisify)(fs.readFile) : async () => {
+  var readFile2 = fs.readFile ? (0, util_1.promisify)(fs.readFile) : async () => {
     throw new ErrorWithCode("use key rather than keyFile.", "MISSING_CREDENTIALS");
   };
   var GOOGLE_TOKEN_URL = "https://www.googleapis.com/oauth2/v4/token";
@@ -21402,7 +21402,7 @@ var require_src6 = __commonJS((exports) => {
       const ext = path.extname(keyFile);
       switch (ext) {
         case ".json": {
-          const key = await readFile(keyFile, "utf8");
+          const key = await readFile2(keyFile, "utf8");
           const body = JSON.parse(key);
           const privateKey = body.private_key;
           const clientEmail = body.client_email;
@@ -21414,7 +21414,7 @@ var require_src6 = __commonJS((exports) => {
         case ".der":
         case ".crt":
         case ".pem": {
-          const privateKey = await readFile(keyFile, "utf8");
+          const privateKey = await readFile2(keyFile, "utf8");
           return { privateKey };
         }
         case ".p12":
@@ -22545,7 +22545,7 @@ var require_filesubjecttokensupplier = __commonJS((exports) => {
   exports.FileSubjectTokenSupplier = undefined;
   var util_1 = __require("util");
   var fs = __require("fs");
-  var readFile = (0, util_1.promisify)((_a16 = fs.readFile) !== null && _a16 !== undefined ? _a16 : () => {});
+  var readFile2 = (0, util_1.promisify)((_a16 = fs.readFile) !== null && _a16 !== undefined ? _a16 : () => {});
   var realpath = (0, util_1.promisify)((_b8 = fs.realpath) !== null && _b8 !== undefined ? _b8 : () => {});
   var lstat = (0, util_1.promisify)((_c = fs.lstat) !== null && _c !== undefined ? _c : () => {});
 
@@ -22569,7 +22569,7 @@ var require_filesubjecttokensupplier = __commonJS((exports) => {
         throw err;
       }
       let subjectToken;
-      const rawText = await readFile(parsedFilePath, { encoding: "utf8" });
+      const rawText = await readFile2(parsedFilePath, { encoding: "utf8" });
       if (this.formatType === "text") {
         subjectToken = rawText;
       } else if (this.formatType === "json" && this.subjectTokenFieldName) {
@@ -38418,7 +38418,7 @@ var {
   Help
 } = import__.default;
 // package.json
-var version = "0.9.62";
+var version = "0.9.64";
 
 // src/runner.ts
 import { execSync } from "node:child_process";
@@ -38426,6 +38426,7 @@ import { promises as fs2 } from "node:fs";
 
 // ../cli-shared/src/config.ts
 import { existsSync, readFileSync } from "node:fs";
+import { readFile } from "node:fs/promises";
 import { homedir } from "node:os";
 import { join } from "node:path";
 // ../core/src/Agent/prompts.ts
@@ -50959,6 +50960,18 @@ var providerModelSchema = exports_external.object({
   parameters: exports_external.record(exports_external.string(), exports_external.any()).optional(),
   budget: exports_external.number().positive().optional()
 });
+var ruleSchema = exports_external.union([
+  exports_external.string(),
+  exports_external.object({ path: exports_external.string() }).strict(),
+  exports_external.object({ url: exports_external.string() }).strict(),
+  exports_external.object({
+    repo: exports_external.string(),
+    path: exports_external.string(),
+    tag: exports_external.string().optional(),
+    commit: exports_external.string().optional(),
+    branch: exports_external.string().optional()
+  }).strict()
+]);
 var configSchema = exports_external.object({
   prices: exports_external.record(exports_external.string(), exports_external.record(exports_external.string(), exports_external.object({
     inputPrice: exports_external.number().optional(),
@@ -50987,7 +51000,10 @@ var configSchema = exports_external.object({
     description: exports_external.string()
   }))).optional(),
   commands: exports_external.record(exports_external.string(), providerModelSchema).optional(),
-  rules: exports_external.array(exports_external.string()).optional().or(exports_external.string()).optional(),
+  tools: exports_external.object({
+    search: providerModelSchema.or(exports_external.boolean()).optional()
+  }).optional(),
+  rules: exports_external.array(ruleSchema).optional().or(exports_external.string()).optional(),
   excludeFiles: exports_external.array(exports_external.string()).optional()
 }).strict();
 // ../core/src/tools/askFollowupQuestion.ts
@@ -51967,9 +51983,30 @@ var replaceInFile_default = {
 // ../core/src/tools/search.ts
 var toolInfo14 = {
   name: "search",
-  description: "Search the web for information.",
+  description: "Search the web for information using Google Search. Use this tool to find current information, facts, news, documentation, or research that is not available in your training data. Returns comprehensive search results with relevant content extracted from the web.",
   parameters: exports_external.object({
-    query: exports_external.string().describe("The query to search for")
+    query: exports_external.string().describe("The query to search for").meta({ usageValue: "Your search query here" })
+  }).meta({
+    examples: [
+      {
+        description: "Search for current events or news",
+        input: {
+          query: "latest developments in AI language models 2024"
+        }
+      },
+      {
+        description: "Look up technical documentation",
+        input: {
+          query: "TypeScript advanced type system features"
+        }
+      },
+      {
+        description: "Research specific information",
+        input: {
+          query: "Node.js performance optimization best practices"
+        }
+      }
+    ]
   })
 };
 var handler14 = async (provider, args) => {
@@ -63106,21 +63143,60 @@ function mergeConfigs(configs) {
   }
   const mergedConfig = configs.reduce((acc, config3) => {
     const merged = merge_default({}, acc, config3);
-    let accRules = acc.rules ?? [];
-    if (typeof accRules === "string") {
-      accRules = [accRules];
-    }
-    let configRules = config3.rules ?? [];
-    if (typeof configRules === "string") {
-      configRules = [configRules];
-    }
+    const accRules = acc.rules ? Array.isArray(acc.rules) ? acc.rules : [acc.rules] : undefined;
+    const configRules = config3.rules ? Array.isArray(config3.rules) ? config3.rules : [config3.rules] : undefined;
     merged.rules = mergeArray(accRules, configRules);
     merged.excludeFiles = mergeArray(acc.excludeFiles, config3.excludeFiles);
     return merged;
   });
   return mergedConfig;
 }
-function loadConfig(paths, cwd = process.cwd(), home = homedir()) {
+async function resolveRules(rules) {
+  if (!rules) {
+    return;
+  }
+  if (typeof rules === "string") {
+    return rules;
+  }
+  const resolvedRules = await Promise.all(rules.map(async (rule) => {
+    if (typeof rule === "string") {
+      return rule;
+    }
+    if ("url" in rule) {
+      try {
+        const response = await fetch(rule.url);
+        if (response.ok) {
+          return await response.text();
+        }
+        console.warn(`Failed to fetch rule from ${rule.url}: ${response.statusText}`);
+      } catch (error46) {
+        console.warn(`Error fetching rule from ${rule.url}: ${error46}`);
+      }
+    } else if ("repo" in rule) {
+      const ref = rule.commit ?? rule.tag ?? rule.branch ?? "main";
+      const url2 = `https://raw.githubusercontent.com/${rule.repo}/${ref}/${rule.path}`;
+      try {
+        const response = await fetch(url2);
+        if (response.ok) {
+          return await response.text();
+        }
+        console.warn(`Failed to fetch rule from ${url2}: ${response.statusText}`);
+      } catch (error46) {
+        console.warn(`Error fetching rule from ${url2}: ${error46}`);
+      }
+    } else if ("path" in rule) {
+      if (existsSync(rule.path)) {
+        return await readFile(rule.path, "utf-8");
+      }
+      console.warn(`Rule file not found: ${rule.path}`);
+    }
+    return;
+  }));
+  return resolvedRules.filter((rule) => rule !== undefined).join(`
+
+`);
+}
+async function loadConfig(paths, cwd = process.cwd(), home = homedir()) {
   const configs = [];
   const globalConfigPath = getGlobalConfigPath(home);
   if (existsSync(globalConfigPath)) {
@@ -63157,7 +63233,15 @@ ${error46}`);
       }
     }
   }
-  return configs.length > 0 ? mergeConfigs(configs) : undefined;
+  const mergedConfig = configs.length > 0 ? mergeConfigs(configs) : undefined;
+  if (!mergedConfig) {
+    return;
+  }
+  const resolvedRules = await resolveRules(mergedConfig.rules);
+  return {
+    ...mergedConfig,
+    rules: resolvedRules
+  };
 }
 var readConfig = (path) => {
   const file2 = readFileSync(path, "utf8");
@@ -63166,7 +63250,7 @@ var readConfig = (path) => {
 };
 // ../cli-shared/src/provider.ts
 import { spawn as spawn2 } from "node:child_process";
-import { mkdir, readFile, rename, unlink, writeFile } from "node:fs/promises";
+import { mkdir, readFile as readFile2, rename, unlink, writeFile } from "node:fs/promises";
 import { dirname, normalize, resolve as resolve3 } from "node:path";
 
 // ../../node_modules/.bun/@ai-sdk+google-vertex@3.0.68+5954958163efbb2a/node_modules/@ai-sdk/google-vertex/dist/index.mjs
@@ -66215,6 +66299,7 @@ var getProvider = (options = {}) => {
   const memoryStore = options.memoryStore ?? new InMemoryStore;
   const todoItemStore = options.todoItemStore ?? new InMemoryStore;
   const defaultMemoryTopic = ":default:";
+  const searchModel = options.getModel?.("search");
   const provider2 = {
     listTodoItems: async (id, status) => {
       const todoItems = await todoItemStore.read() ?? [];
@@ -66348,7 +66433,7 @@ ${content}`;
         throw new Error(`Not allow to access file ${path}`);
       }
       try {
-        return await readFile(path, "utf8");
+        return await readFile2(path, "utf8");
       } catch (_e) {
         return;
       }
@@ -66382,7 +66467,7 @@ ${content}`;
         if (!resolvedPath.startsWith(process.cwd())) {
           throw new Error(`Access to file path "${filePath}" is restricted.`);
         }
-        const data2 = await readFile(resolvedPath);
+        const data2 = await readFile2(resolvedPath);
         const mediaType2 = $lookup(resolvedPath) || "application/octet-stream";
         return {
           base64Data: data2.toString("base64"),
@@ -66479,13 +66564,10 @@ ${content}`;
         throw error46;
       }
     },
-    search: options.getModel && (async (query) => {
-      const model = options.getModel?.("search");
-      if (!model) {
-        throw new Error("Unable to get model for search");
-      }
+    search: searchModel && (async (query) => {
       const resp = await generateText({
-        model,
+        model: searchModel,
+        system: "You are a web search assistant. When searching for information, provide comprehensive and detailed results. Include relevant facts, statistics, dates, and key details from the search results. Synthesize information from multiple sources when available. Structure your response clearly with the most relevant information first. Reference or cite sources when presenting specific claims or data.",
         tools: {
           google_search: vertex.tools.googleSearch({})
         },
@@ -66722,9 +66804,8 @@ class Runner {
   wsManager;
   provider;
   availableTools;
-  constructor(options) {
+  constructor(options, config4) {
     this.options = options;
-    const config4 = loadConfig() ?? {};
     this.provider = getProvider({
       command: {
         onStarted(command) {
@@ -66743,7 +66824,7 @@ class Runner {
           console.log(`$ <<<< $ Command error: ${error46}`);
         }
       },
-      excludeFiles: config4.excludeFiles
+      excludeFiles: config4?.excludeFiles
     });
     this.availableTools = {
       [executeCommand_default.name]: executeCommand_default,
@@ -67052,7 +67133,8 @@ async function runRunner(options) {
     console.error("Error: API URL is required.");
     process.exit(1);
   }
-  const runner = new Runner(options);
+  const config4 = await loadConfig();
+  const runner = new Runner(options, config4);
   runner.start();
 }
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@polka-codes/runner",
-  "version": "0.9.62",
+  "version": "0.9.64",
   "license": "AGPL-3.0",
   "author": "github@polka.codes",
   "type": "module",
@@ -18,8 +18,8 @@
   },
   "dependencies": {
     "@ai-sdk/provider": "^2.0.0",
-    "@polka-codes/cli-shared": "0.9.61",
-    "@polka-codes/core": "0.9.61",
+    "@polka-codes/cli-shared": "0.9.63",
+    "@polka-codes/core": "0.9.63",
     "commander": "^14.0.2",
     "dotenv": "^17.2.3",
     "ignore": "^7.0.5",