teleton 0.5.2 → 0.6.0

package/dist/cli/index.js

@@ -1,33 +1,45 @@
+import {
+  CONFIGURABLE_KEYS,
+  deleteNestedValue,
+  getNestedValue,
+  readRawConfig,
+  setNestedValue,
+  writeRawConfig
+} from "../chunk-2QUJLHCZ.js";
+import {
+  TelegramUserClient,
+  ensureWorkspace,
+  isNewWorkspace,
+  main
+} from "../chunk-6L6KGATM.js";
+import "../chunk-WL2Q3VRD.js";
+import "../chunk-EHEV7FJ7.js";
+import "../chunk-U7FQYCBQ.js";
 import {
   ConfigSchema,
   DealsConfigSchema,
-  TelegramUserClient,
   configExists,
-  ensureWorkspace,
   generateWallet,
   getDefaultConfigPath,
   getProviderMetadata,
   getSupportedProviders,
   importWallet,
-  isNewWorkspace,
   loadWallet,
-  main,
   saveWallet,
   validateApiKeyFormat,
   walletExists
-} from "../chunk-WOXBZOQX.js";
-import "../chunk-WL2Q3VRD.js";
-import "../chunk-EHEV7FJ7.js";
-import "../chunk-U7FQYCBQ.js";
-import "../chunk-5WWR4CU3.js";
+} from "../chunk-4IPJ25HE.js";
 import "../chunk-YBA6IBGT.js";
+import "../chunk-ADCMUNYU.js";
+import "../chunk-D5I7GBV7.js";
+import "../chunk-ECSCVEQQ.js";
 import {
   fetchWithTimeout
-} from "../chunk-WUTMT6DW.js";
+} from "../chunk-GDCODBNO.js";
 import "../chunk-4DU3C27M.js";
 import {
   TELEGRAM_MAX_MESSAGE_LENGTH
-} from "../chunk-O4R7V5Y2.js";
+} from "../chunk-RO62LO6Z.js";
 import {
   TELETON_ROOT
 } from "../chunk-EYWNOHMJ.js";
@@ -325,7 +337,7 @@ Get it at: ${providerMeta.consoleUrl}`,
       prompter.warn(`TELETON_API_KEY env var found but invalid: ${validationError}`);
       apiKey = await prompter.password({
         message: `${providerMeta.displayName} API Key (${providerMeta.keyHint})`,
-        validate: (value) => validateApiKeyFormat(selectedProvider, value)
+        validate: (value = "") => validateApiKeyFormat(selectedProvider, value)
       });
     } else {
       prompter.log(`Using API key from TELETON_API_KEY env var`);
@@ -334,7 +346,7 @@ Get it at: ${providerMeta.consoleUrl}`,
   } else {
     apiKey = await prompter.password({
       message: `${providerMeta.displayName} API Key (${providerMeta.keyHint})`,
-      validate: (value) => validateApiKeyFormat(selectedProvider, value)
+      validate: (value = "") => validateApiKeyFormat(selectedProvider, value)
     });
   }
   const MODEL_OPTIONS = {
@@ -442,7 +454,7 @@ Get it at: ${providerMeta.consoleUrl}`,
     maxAgenticIterations = await prompter.text({
       message: "Max agentic iterations (tool call loops per message)",
       initialValue: "5",
-      validate: (v) => {
+      validate: (v = "") => {
         const n = parseInt(v, 10);
         if (isNaN(n) || n < 1 || n > 50) return "Must be a number between 1 and 50";
       }
@@ -461,7 +473,7 @@ Get it at: ${providerMeta.consoleUrl}`,
       const buyInput = await prompter.text({
         message: "Max buy price (% of floor price)",
         initialValue: "100",
-        validate: (v) => {
+        validate: (v = "") => {
           const n = parseInt(v, 10);
           if (isNaN(n) || n < 50 || n > 150) return "Must be between 50 and 150";
         }
@@ -470,7 +482,7 @@ Get it at: ${providerMeta.consoleUrl}`,
       const sellInput = await prompter.text({
         message: "Min sell price (% of floor price)",
         initialValue: "105",
-        validate: (v) => {
+        validate: (v = "") => {
           const n = parseInt(v, 10);
           if (isNaN(n) || n < 100 || n > 200) return "Must be between 100 and 200";
         }
@@ -530,6 +542,24 @@ Get it at: ${providerMeta.consoleUrl}`,
     });
     tonapiKey = keyInput;
   }
+  let tavilyApiKey;
+  const setupTavily = await prompter.confirm({
+    message: "Enable web search? (requires a free Tavily API key \u2014 1,000 req/month, no credit card)",
+    initialValue: false
+  });
+  if (setupTavily) {
+    prompter.note(
+      "Web search lets your agent search the internet and read web pages.\n\nTo get your free API key (takes 30 seconds):\n\n  1. Go to https://app.tavily.com/sign-in\n  2. Create an account (email or Google/GitHub)\n  3. Your API key is displayed on the dashboard\n     (starts with tvly-)\n\nFree plan: 1,000 requests/month \u2014 no credit card required.",
+      "Tavily \u2014 Web Search API"
+    );
+    const keyInput = await prompter.text({
+      message: "Tavily API key (starts with tvly-)",
+      validate: (v) => {
+        if (!v || !v.startsWith("tvly-")) return "Invalid key (should start with tvly-)";
+      }
+    });
+    tavilyApiKey = keyInput;
+  }
   const config = {
     meta: {
       version: "1.0.0",
@@ -593,8 +623,24 @@ Get it at: ${providerMeta.consoleUrl}`,
       log_requests: false
     },
     dev: { hot_reload: false },
+    tool_rag: {
+      enabled: true,
+      top_k: 25,
+      always_include: [
+        "telegram_send_message",
+        "telegram_reply_message",
+        "telegram_send_photo",
+        "telegram_send_document",
+        "journal_*",
+        "workspace_*",
+        "web_*"
+      ],
+      skip_unlimited_providers: false
+    },
+    mcp: { servers: {} },
     plugins: {},
-    tonapi_key: tonapiKey
+    tonapi_key: tonapiKey,
+    tavily_api_key: tavilyApiKey
   };
   spinner2.start("Saving configuration...");
   const configYaml = YAML.stringify(config);
@@ -619,7 +665,7 @@ Get it at: ${providerMeta.consoleUrl}`,
     } else if (walletAction === "import") {
       const mnemonicInput = await prompter.text({
         message: "Enter your 24-word mnemonic (space-separated)",
-        validate: (value) => {
+        validate: (value = "") => {
           const words = value.trim().split(/\s+/);
           if (words.length !== 24) return `Expected 24 words, got ${words.length}`;
         }
@@ -683,7 +729,8 @@ Telegram: ${phone} (API ID: ${apiId})${telegramConnected ? " \u2713 connected" :
 Admin: User ID ${userId}
 Provider: ${providerMeta.displayName}
 Model: ${selectedModel}
-TON Wallet: ${wallet.address}`,
+TON Wallet: ${wallet.address}
+Web Search: ${tavilyApiKey ? "Tavily \u2713" : "disabled (no Tavily key)"}`,
     "Setup complete"
   );
   if (telegramConnected) {
@@ -771,7 +818,23 @@ async function runNonInteractiveOnboarding(options, prompter) {
       log_requests: false
     },
     dev: { hot_reload: false },
-    plugins: {}
+    tool_rag: {
+      enabled: true,
+      top_k: 25,
+      always_include: [
+        "telegram_send_message",
+        "telegram_reply_message",
+        "telegram_send_photo",
+        "telegram_send_document",
+        "journal_*",
+        "workspace_*",
+        "web_*"
+      ],
+      skip_unlimited_providers: false
+    },
+    mcp: { servers: {} },
+    plugins: {},
+    tavily_api_key: options.tavilyApiKey
   };
   const configYaml = YAML.stringify(config);
   writeFileSync(workspace.configPath, configYaml, "utf-8");
@@ -1159,6 +1222,224 @@ ${blue}  \u250C\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u250
   console.log("");
 }
 
+// src/cli/commands/mcp.ts
+function ensureMcpSection(raw) {
+  if (!raw.mcp || typeof raw.mcp !== "object") {
+    raw.mcp = { servers: {} };
+  }
+  const mcp2 = raw.mcp;
+  if (!mcp2.servers || typeof mcp2.servers !== "object") {
+    mcp2.servers = {};
+  }
+  return mcp2.servers;
+}
+function deriveServerName(pkg) {
+  const unscoped = pkg.includes("/") ? pkg.split("/").pop() : pkg;
+  return unscoped.replace(/^server-/, "").replace(/^mcp-server-/, "").replace(/^mcp-/, "");
+}
+async function mcpAddCommand(pkg, extraArgs, options) {
+  const configPath = options.config || getDefaultConfigPath();
+  const raw = readRawConfig(configPath);
+  const servers = ensureMcpSection(raw);
+  const serverName = options.name || deriveServerName(pkg);
+  if (servers[serverName]) {
+    console.error(`\u274C MCP server "${serverName}" already exists. Remove it first or use --name.`);
+    process.exit(1);
+  }
+  const entry = {};
+  if (options.url) {
+    entry.url = pkg;
+  } else {
+    const cmdParts = ["npx", "-y", pkg, ...extraArgs];
+    entry.command = cmdParts.join(" ");
+  }
+  if (options.scope && options.scope !== "always") {
+    entry.scope = options.scope;
+  }
+  if (options.env && options.env.length > 0) {
+    const envMap = {};
+    for (const pair of options.env) {
+      const eq = pair.indexOf("=");
+      if (eq === -1) {
+        console.error(`\u274C Invalid --env format: "${pair}" (expected KEY=VALUE)`);
+        process.exit(1);
+      }
+      envMap[pair.slice(0, eq)] = pair.slice(eq + 1);
+    }
+    entry.env = envMap;
+  }
+  servers[serverName] = entry;
+  writeRawConfig(raw, configPath);
+  console.log(`\u2705 Added MCP server "${serverName}"`);
+  if (entry.command) {
+    console.log(`   Command: ${entry.command}`);
+  } else {
+    console.log(`   URL: ${entry.url}`);
+  }
+  console.log(`
+   Restart teleton to connect: teleton start`);
+}
+async function mcpRemoveCommand(name, options) {
+  const configPath = options.config || getDefaultConfigPath();
+  const raw = readRawConfig(configPath);
+  const servers = ensureMcpSection(raw);
+  if (!servers[name]) {
+    console.error(`\u274C MCP server "${name}" not found.`);
+    const names = Object.keys(servers);
+    if (names.length > 0) {
+      console.error(`   Available: ${names.join(", ")}`);
+    }
+    process.exit(1);
+  }
+  delete servers[name];
+  writeRawConfig(raw, configPath);
+  console.log(`\u2705 Removed MCP server "${name}"`);
+}
+async function mcpListCommand(options) {
+  const configPath = options.config || getDefaultConfigPath();
+  const raw = readRawConfig(configPath);
+  const servers = ensureMcpSection(raw);
+  const entries = Object.entries(servers);
+  if (entries.length === 0) {
+    console.log("No MCP servers configured.");
+    console.log("\n  Add one: teleton mcp add @modelcontextprotocol/server-filesystem /tmp");
+    return;
+  }
+  console.log(`MCP servers (${entries.length}):
+`);
+  for (const [name, cfg] of entries) {
+    const type = cfg.command ? "stdio" : "sse";
+    const target = cfg.command || cfg.url || "?";
+    const scope = cfg.scope || "always";
+    const enabled = cfg.enabled !== false ? "\u2713" : "\u2717";
+    console.log(`  ${enabled} ${name} (${type}, ${scope})`);
+    console.log(`    ${target}`);
+    if (cfg.env && typeof cfg.env === "object") {
+      const keys = Object.keys(cfg.env);
+      console.log(`    env: ${keys.join(", ")}`);
+    }
+  }
+}
+
+// src/cli/commands/config.ts
+function requireWhitelisted(key) {
+  const meta = CONFIGURABLE_KEYS[key];
+  if (!meta) {
+    const allowed = Object.keys(CONFIGURABLE_KEYS).join(", ");
+    console.error(`Key "${key}" is not configurable.
+   Allowed keys: ${allowed}`);
+    process.exit(1);
+  }
+  return meta;
+}
+async function actionSet(key, value, configPath) {
+  const meta = requireWhitelisted(key);
+  if (!value) {
+    const prompter = createPrompter();
+    try {
+      if (meta.sensitive) {
+        value = await prompter.password({
+          message: `Enter value for ${key}:`,
+          validate: (v) => {
+            if (!v) return "Value is required";
+            const err2 = meta.validate(v);
+            return err2 ? new Error(err2) : void 0;
+          }
+        });
+      } else {
+        value = await prompter.text({
+          message: `Enter value for ${key}:`,
+          validate: (v) => {
+            if (!v) return "Value is required";
+            const err2 = meta.validate(v);
+            return err2 ? new Error(err2) : void 0;
+          }
+        });
+      }
+    } catch (e) {
+      if (e instanceof CancelledError) {
+        console.log("Cancelled.");
+        return;
+      }
+      throw e;
+    }
+  }
+  const err = meta.validate(value);
+  if (err) {
+    console.error(`Invalid value for ${key}: ${err}`);
+    process.exit(1);
+  }
+  const raw = readRawConfig(configPath);
+  setNestedValue(raw, key, meta.parse(value));
+  writeRawConfig(raw, configPath);
+  console.log(`\u2713 ${key} = ${meta.mask(value)}`);
+}
+function actionGet(key, configPath) {
+  const meta = requireWhitelisted(key);
+  const raw = readRawConfig(configPath);
+  const value = getNestedValue(raw, key);
+  if (value == null || value === "") {
+    console.log(`\u2717 ${key}  (not set)`);
+  } else {
+    const display = meta.sensitive ? meta.mask(String(value)) : String(value);
+    console.log(`\u2713 ${key} = ${display}`);
+  }
+}
+function actionList(configPath) {
+  const raw = readRawConfig(configPath);
+  console.log("\nConfigurable keys:\n");
+  for (const [key, meta] of Object.entries(CONFIGURABLE_KEYS)) {
+    const value = getNestedValue(raw, key);
+    if (value != null && value !== "") {
+      const display = meta.sensitive ? meta.mask(String(value)) : String(value);
+      console.log(`  \u2713 ${key.padEnd(24)} = ${display}`);
+    } else {
+      console.log(`  \u2717 ${key.padEnd(24)}   (not set)`);
+    }
+  }
+  console.log();
+}
+function actionUnset(key, configPath) {
+  requireWhitelisted(key);
+  const raw = readRawConfig(configPath);
+  deleteNestedValue(raw, key);
+  writeRawConfig(raw, configPath);
+  console.log(`\u2713 ${key} unset`);
+}
+async function configCommand(action, key, value, options) {
+  const configPath = options.config ?? getDefaultConfigPath();
+  switch (action) {
+    case "list":
+      actionList(configPath);
+      break;
+    case "get":
+      if (!key) {
+        console.error("Usage: teleton config get <key>");
+        process.exit(1);
+      }
+      actionGet(key, configPath);
+      break;
+    case "set":
+      if (!key) {
+        console.error("Usage: teleton config set <key> [value]");
+        process.exit(1);
+      }
+      await actionSet(key, value, configPath);
+      break;
+    case "unset":
+      if (!key) {
+        console.error("Usage: teleton config unset <key>");
+        process.exit(1);
+      }
+      actionUnset(key, configPath);
+      break;
+    default:
+      console.error(`Unknown action: ${action}
+Available: set, get, list, unset`);
+      process.exit(1);
+  }
+}
+
 // src/cli/index.ts
 import { readFileSync as readFileSync3, existsSync as existsSync3 } from "fs";
 import { dirname, join as join3 } from "path";
@@ -1177,7 +1458,7 @@ function findPackageJson() {
 var packageJson = findPackageJson();
 var program = new Command();
 program.name("teleton").description("Teleton Agent - Personal AI Agent for Telegram").version(packageJson.version);
-program.command("setup").description("Interactive wizard to set up Teleton").option("--workspace <dir>", "Workspace directory").option("--non-interactive", "Non-interactive mode").option("--api-id <id>", "Telegram API ID").option("--api-hash <hash>", "Telegram API Hash").option("--phone <number>", "Phone number").option("--api-key <key>", "Anthropic API key").option("--user-id <id>", "Telegram User ID").action(async (options) => {
+program.command("setup").description("Interactive wizard to set up Teleton").option("--workspace <dir>", "Workspace directory").option("--non-interactive", "Non-interactive mode").option("--api-id <id>", "Telegram API ID").option("--api-hash <hash>", "Telegram API Hash").option("--phone <number>", "Phone number").option("--api-key <key>", "Anthropic API key").option("--user-id <id>", "Telegram User ID").option("--tavily-api-key <key>", "Tavily API key for web search").action(async (options) => {
   try {
     await onboardCommand({
       workspace: options.workspace,
@@ -1186,7 +1467,8 @@ program.command("setup").description("Interactive wizard to set up Teleton").opt
       apiHash: options.apiHash,
       phone: options.phone,
       apiKey: options.apiKey,
-      userId: options.userId ? parseInt(options.userId) : void 0
+      userId: options.userId ? parseInt(options.userId) : void 0,
+      tavilyApiKey: options.tavilyApiKey
     });
   } catch (error) {
     console.error("Error:", error instanceof Error ? error.message : String(error));
@@ -1221,6 +1503,46 @@ program.command("doctor").description("Run system health checks").action(async (
     process.exit(1);
   }
 });
+var mcp = program.command("mcp").description("Manage MCP (Model Context Protocol) servers");
+mcp.command("add <package> [args...]").description(
+  "Add an MCP server (e.g. teleton mcp add @modelcontextprotocol/server-filesystem /tmp)"
+).option("-n, --name <name>", "Server name (auto-derived from package if omitted)").option("-s, --scope <scope>", "Tool scope: always | dm-only | group-only | admin-only", "always").option(
+  "-e, --env <KEY=VALUE...>",
+  "Environment variables (repeatable)",
+  (v, prev) => [...prev, v],
+  []
+).option("--url", "Treat <package> as an SSE/HTTP URL instead of an npx package").option("-c, --config <path>", "Config file path").action(async (pkg, args, options) => {
+  try {
+    await mcpAddCommand(pkg, args, options);
+  } catch (error) {
+    console.error("Error:", error instanceof Error ? error.message : String(error));
+    process.exit(1);
+  }
+});
+mcp.command("remove <name>").description("Remove an MCP server by name").option("-c, --config <path>", "Config file path").action(async (name, options) => {
+  try {
+    await mcpRemoveCommand(name, options);
+  } catch (error) {
+    console.error("Error:", error instanceof Error ? error.message : String(error));
+    process.exit(1);
+  }
+});
+mcp.command("list").description("List configured MCP servers").option("-c, --config <path>", "Config file path").action(async (options) => {
+  try {
+    await mcpListCommand(options);
+  } catch (error) {
+    console.error("Error:", error instanceof Error ? error.message : String(error));
+    process.exit(1);
+  }
+});
+program.command("config").description("Manage configuration keys (set, get, list, unset)").argument("<action>", "set | get | list | unset").argument("[key]", "Config key (e.g., tavily_api_key, telegram.bot_token)").argument("[value]", "Value to set (prompts interactively if omitted)").option("-c, --config <path>", "Config file path").action(async (action, key, value, options) => {
+  try {
+    await configCommand(action, key, value, options);
+  } catch (error) {
+    console.error("Error:", error instanceof Error ? error.message : String(error));
+    process.exit(1);
+  }
+});
 program.action(() => {
   program.help();
 });

package/dist/index.js

@@ -1,15 +1,18 @@
 import {
   TeletonApp,
   main
-} from "./chunk-WOXBZOQX.js";
+} from "./chunk-6L6KGATM.js";
 import "./chunk-WL2Q3VRD.js";
 import "./chunk-EHEV7FJ7.js";
 import "./chunk-U7FQYCBQ.js";
-import "./chunk-5WWR4CU3.js";
+import "./chunk-4IPJ25HE.js";
 import "./chunk-YBA6IBGT.js";
-import "./chunk-WUTMT6DW.js";
+import "./chunk-ADCMUNYU.js";
+import "./chunk-D5I7GBV7.js";
+import "./chunk-ECSCVEQQ.js";
+import "./chunk-GDCODBNO.js";
 import "./chunk-4DU3C27M.js";
-import "./chunk-O4R7V5Y2.js";
+import "./chunk-RO62LO6Z.js";
 import "./chunk-EYWNOHMJ.js";
 import "./chunk-NUGDTPE4.js";
 import "./chunk-QUAPFI2N.js";

package/dist/{memory-Y5J7CXAR.js → memory-TVDOGQXS.js}

@@ -1,32 +1,36 @@
 import {
-  AnthropicEmbeddingProvider,
   CURRENT_SCHEMA_VERSION,
-  CachedEmbeddingProvider,
   ChatStore,
   ContextBuilder,
   HybridSearch,
   KnowledgeIndexer,
-  LocalEmbeddingProvider,
   MemoryDatabase,
   MessageStore,
-  NoopEmbeddingProvider,
   SessionStore,
   UserStore,
   closeDatabase,
-  createEmbeddingProvider,
-  deserializeEmbedding,
   ensureSchema,
   ensureVectorTables,
   getDatabase,
   getSchemaVersion,
-  hashText,
   initializeMemory,
   runMigrations,
-  serializeEmbedding,
   setSchemaVersion
-} from "./chunk-WUTMT6DW.js";
+} from "./chunk-ADCMUNYU.js";
+import {
+  AnthropicEmbeddingProvider,
+  CachedEmbeddingProvider,
+  LocalEmbeddingProvider,
+  NoopEmbeddingProvider,
+  createEmbeddingProvider,
+  deserializeEmbedding,
+  hashText,
+  serializeEmbedding
+} from "./chunk-D5I7GBV7.js";
+import "./chunk-ECSCVEQQ.js";
+import "./chunk-GDCODBNO.js";
 import "./chunk-4DU3C27M.js";
-import "./chunk-O4R7V5Y2.js";
+import "./chunk-RO62LO6Z.js";
 import "./chunk-EYWNOHMJ.js";
 import {
   TaskStore,

package/dist/{migrate-UEQCDWL2.js → migrate-QIEMPOMT.js}

@@ -1,8 +1,11 @@
 import {
   getDatabase
-} from "./chunk-WUTMT6DW.js";
+} from "./chunk-ADCMUNYU.js";
+import "./chunk-D5I7GBV7.js";
+import "./chunk-ECSCVEQQ.js";
+import "./chunk-GDCODBNO.js";
 import "./chunk-4DU3C27M.js";
-import "./chunk-O4R7V5Y2.js";
+import "./chunk-RO62LO6Z.js";
 import {
   TELETON_ROOT
 } from "./chunk-EYWNOHMJ.js";