@iola_adm/iola-cli 0.1.6 → 0.1.8

package/README.md

@@ -79,6 +79,9 @@ iola agent
 iola ai doctor
 iola ai setup ollama
 iola ai ask "Какие школы есть на улице Петрова?"
+iola ai context "школа 29"
+iola ai key set openai
+iola ai key status
 iola ai setup openai --model gpt-4.1-mini
 iola ai setup openrouter --model openai/gpt-4.1-mini
 iola health
@@ -113,6 +116,11 @@ iola agent
 /search лицей --limit 3
 /mcp-info
 /ai doctor
+/context школа 29
+/use ollama
+/use openai
+/key status
+/key set openai
 /model
 /provider
 /config
@@ -135,7 +143,7 @@ iola ai ask "Какие школы есть на улице Петрова?"
 OpenAI:
 
 ```bash
-set OPENAI_API_KEY=...
+iola ai key set openai
 iola ai setup openai --model gpt-4.1-mini
 iola ai ask "Найди школу 29"
 ```
@@ -143,19 +151,48 @@ iola ai ask "Найди школу 29"
 OpenRouter:
 
 ```bash
-set OPENROUTER_API_KEY=...
+iola ai key set openrouter
 iola ai setup openrouter --model openai/gpt-4.1-mini
 iola ai ask "Покажи контакты лицея"
 ```
 
+Проверить, какие данные попадут в AI-контекст:
+
+```bash
+iola ai context "школа 29"
+iola ai context "1215067180" --json
+iola ai context "улица Петрова"
+```
+
+Поиск контекста учитывает номера учреждений, ИНН и улицы.
+
+Ключи OpenAI/OpenRouter сохраняются локально на компьютере пользователя:
+
+```text
+%USERPROFILE%\.iola\secrets.json
+```
+
+Управление ключами:
+
+```bash
+iola ai key set openai
+iola ai key set openrouter
+iola ai key status
+iola ai key delete openai
+```
+
+Если одновременно задана переменная окружения (`OPENAI_API_KEY` или
+`OPENROUTER_API_KEY`) и сохранен локальный ключ, CLI использует переменную
+окружения как более приоритетную.
+
 AI-ответ строится по контексту из публичного API. Если данных в контексте
 недостаточно, ассистент должен сообщить об этом, а не выдумывать сведения.
 
 ## Назначение
 
-Первый релиз CLI дает прямой терминальный доступ к открытым данным и командам
-подключения MCP/skill. Дальше планируется добавить режим AI-запросов через
-ключ пользователя для OpenAI/OpenRouter и интерактивный агентный режим.
+CLI дает прямой терминальный доступ к открытым данным городского округа,
+командам подключения MCP/skill, AI-запросам через Ollama/OpenAI/OpenRouter и
+интерактивному агентному режиму.
 
 ## Переменные окружения
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@iola_adm/iola-cli",
-  "version": "0.1.6",
+  "version": "0.1.8",
   "description": "CLI и AI-агент для работы с открытыми данными городского округа Йошкар-Ола.",
   "license": "MIT",
   "homepage": "https://github.com/adm-iola/iola-cli#readme",

package/src/cli.js

@@ -9,6 +9,7 @@ const API_BASE_URL = process.env.IOLA_API_BASE_URL || "https://apiiola.yasg.ru/a
 const MCP_BASE_URL = process.env.IOLA_MCP_BASE_URL || "https://apiiola.yasg.ru";
 const CONFIG_DIR = path.join(os.homedir(), ".iola");
 const CONFIG_FILE = path.join(CONFIG_DIR, "config.json");
+const SECRETS_FILE = path.join(CONFIG_DIR, "secrets.json");
 const DEFAULT_AI_CONFIG = {
   ai: {
     provider: "ollama",
@@ -64,6 +65,11 @@ Usage:
   iola banner
   iola agent
   iola ai ask TEXT [--provider ollama|openai|openrouter] [--model MODEL]
+  iola ai context TEXT [--json]
+  iola ai key set openai
+  iola ai key set openrouter
+  iola ai key status
+  iola ai key delete openai|openrouter
   iola ai doctor [--json]
   iola ai setup
   iola ai setup ollama [--yes] [--model MODEL]
@@ -158,6 +164,21 @@ async function handleAgentLine(line, state) {
     return false;
   }
 
+  if (command === "context") {
+    await aiContext(args);
+    return false;
+  }
+
+  if (command === "use") {
+    await useAiProvider(args);
+    return false;
+  }
+
+  if (command === "key") {
+    await handleAiKey(args);
+    return false;
+  }
+
   if (command === "provider") {
     await printAiConfigField("provider");
     return false;
@@ -210,8 +231,13 @@ function printAgentHelp() {
   /kindergartens get --inn 1215077421
   /search лицей --limit 3
   /mcp-info
+  /context школа 29
   /ai doctor
   /ai setup ollama
+  /use openai
+  /use ollama
+  /key status
+  /key set openai
   /config
   /provider
   /model
@@ -285,6 +311,11 @@ async function handleAi(args) {
     showBanner();
     console.log(`AI-команды:
   iola ai ask TEXT [--provider ollama|openai|openrouter] [--model MODEL]
+  iola ai context TEXT [--json]
+  iola ai key set openai
+  iola ai key set openrouter
+  iola ai key status
+  iola ai key delete openai|openrouter
   iola ai doctor [--json]
   iola ai setup
   iola ai setup ollama [--yes] [--model MODEL]
@@ -300,6 +331,16 @@ async function handleAi(args) {
     return;
   }
 
+  if (subcommand === "context") {
+    await aiContext(rest);
+    return;
+  }
+
+  if (subcommand === "key") {
+    await handleAiKey(rest);
+    return;
+  }
+
   if (subcommand === "doctor") {
     await aiDoctor(rest);
     return;
@@ -364,6 +405,133 @@ async function aiSetup(args) {
   throw new Error(`Unknown AI provider: ${provider}`);
 }
 
+async function handleAiKey(args) {
+  const [action, provider] = args;
+
+  if (action === "set") {
+    await setAiKey(provider);
+    return;
+  }
+
+  if (action === "status") {
+    await printAiKeyStatus();
+    return;
+  }
+
+  if (action === "delete") {
+    await deleteAiKey(provider);
+    return;
+  }
+
+  throw new Error(`Unknown key command. Use:
+  iola ai key set openai
+  iola ai key set openrouter
+  iola ai key status
+  iola ai key delete openai|openrouter`);
+}
+
+async function useAiProvider(args) {
+  const [provider] = args;
+
+  if (provider !== "ollama" && provider !== "openai" && provider !== "openrouter") {
+    throw new Error("Провайдер должен быть ollama, openai или openrouter.");
+  }
+
+  const config = await loadConfig();
+  const defaultModel = {
+    ollama: config.ai.provider === "ollama" ? config.ai.model : "llama3.2:1b",
+    openai: config.ai.provider === "openai" ? config.ai.model : "gpt-4.1-mini",
+    openrouter: config.ai.provider === "openrouter" ? config.ai.model : "openai/gpt-4.1-mini",
+  }[provider];
+
+  await saveConfig({
+    ai: {
+      provider,
+      model: defaultModel,
+      baseUrl: provider === "ollama"
+        ? "http://127.0.0.1:11434"
+        : provider === "openai"
+          ? "https://api.openai.com/v1"
+          : "https://openrouter.ai/api/v1",
+    },
+  });
+
+  console.log(`AI-провайдер переключен: ${provider}, модель: ${defaultModel}`);
+}
+
+async function aiContext(args) {
+  const options = parseOptions(args);
+  const query = options._.join(" ").trim();
+
+  if (!query) {
+    throw new Error('Текст запроса обязателен. Пример: iola ai context "школа 29"');
+  }
+
+  const context = await buildDataContext(query);
+
+  if (options.json) {
+    printJson(context);
+    return;
+  }
+
+  printContext(context);
+}
+
+async function setAiKey(provider) {
+  assertKeyProvider(provider);
+
+  if (!process.stdin.isTTY) {
+    throw new Error("Для сохранения ключа запустите команду в интерактивном терминале.");
+  }
+
+  const envName = provider === "openai" ? "OPENAI_API_KEY" : "OPENROUTER_API_KEY";
+  const rl = readline.createInterface({ input, output });
+
+  try {
+    const key = (await rl.question(`Введите ${envName}: `)).trim();
+
+    if (!key) {
+      throw new Error("Ключ пустой, сохранение отменено.");
+    }
+
+    const secrets = await loadSecrets();
+    secrets[provider] = { apiKey: key };
+    await saveSecrets(secrets);
+    console.log(`Ключ ${provider} сохранен локально: ${SECRETS_FILE}`);
+  } finally {
+    rl.close();
+  }
+}
+
+async function printAiKeyStatus() {
+  const secrets = await loadSecrets();
+  const rows = ["openai", "openrouter"].map((provider) => ({
+    provider,
+    env: provider === "openai" ? (process.env.OPENAI_API_KEY ? "yes" : "no") : (process.env.OPENROUTER_API_KEY ? "yes" : "no"),
+    local: secrets[provider]?.apiKey ? "yes" : "no",
+  }));
+
+  printTable(rows, [
+    ["provider", "Провайдер"],
+    ["env", "Env"],
+    ["local", "Локально"],
+  ]);
+}
+
+async function deleteAiKey(provider) {
+  assertKeyProvider(provider);
+  const secrets = await loadSecrets();
+  delete secrets[provider];
+  await saveSecrets(secrets);
+  console.log(`Локальный ключ ${provider} удален.`);
+}
+
+function assertKeyProvider(provider) {
+  if (provider !== "openai" && provider !== "openrouter") {
+    throw new Error("Провайдер должен быть openai или openrouter.");
+  }
+}
+
 async function chooseAiProvider() {
   console.log("Выберите режим AI:");
   console.log("1. Локальная модель через Ollama");
@@ -456,11 +624,23 @@ async function buildDataContext(question) {
     fetchJson(`${API_BASE_URL}/kindergartens?limit=100&offset=0`),
   ]);
   const queryTerms = extractSearchTerms(question);
-  const schoolItems = findRelevantItems(normalizeItems(schools), queryTerms).slice(0, 8);
-  const kindergartenItems = findRelevantItems(normalizeItems(kindergartens), queryTerms).slice(0, 8);
+  const patterns = extractStructuredPatterns(question);
+  const includeSchools = patterns.targetLayers.length === 0 || patterns.targetLayers.includes("schools");
+  const includeKindergartens = patterns.targetLayers.length === 0 || patterns.targetLayers.includes("kindergartens");
+  const schoolItems = includeSchools
+    ? findRelevantItems(normalizeItems(schools), queryTerms, patterns, "schools").slice(0, 8)
+    : [];
+  const kindergartenItems = includeKindergartens
+    ? findRelevantItems(normalizeItems(kindergartens), queryTerms, patterns, "kindergartens").slice(0, 8)
+    : [];
 
   return {
     layers: layers.data_layers || [],
+    query: {
+      text: question,
+      terms: queryTerms,
+      patterns,
+    },
     schools: schoolItems.map(selectPublicSummary),
     kindergartens: kindergartenItems.map(selectPublicSummary),
   };
@@ -478,20 +658,83 @@ function extractSearchTerms(question) {
   return normalized.length > 0 ? normalized : [question];
 }
 
-function findRelevantItems(items, terms) {
+function extractStructuredPatterns(question) {
+  const normalized = question.toLocaleLowerCase("ru-RU");
+  const numbers = [...new Set([...normalized.matchAll(/\b\d{1,3}\b/g)].map((match) => match[0]))];
+  const inns = [...new Set([...normalized.matchAll(/\b\d{10,12}\b/g)].map((match) => match[0]))];
+  const targetLayers = [];
+  if (/(^|[^а-яёa-z])(школа|школы|лицей|лицея|гимназия|гимназии)(?=$|[^а-яёa-z])/iu.test(normalized)) {
+    targetLayers.push("schools");
+  }
+  if (/(^|[^а-яёa-z])(сад|сады|детсад|детский|детские|доу|мбдоу)(?=$|[^а-яёa-z])/iu.test(normalized)) {
+    targetLayers.push("kindergartens");
+  }
+  const streetMatches = [
+    ...normalized.matchAll(/(?:улица|ул\.?)\s+([а-яёa-z0-9 .-]+)/giu),
+    ...normalized.matchAll(/([а-яёa-z0-9 .-]+)\s+(?:улица|ул\.?)/giu),
+  ];
+  const streets = [...new Set(streetMatches.map((match) => cleanupPattern(match[1])).filter(Boolean))];
+
+  return { numbers, inns, streets, targetLayers: [...new Set(targetLayers)] };
+}
+
+function cleanupPattern(value) {
+  return value
+    .replace(/\b(школа|школы|сад|детский|детские|сады|лицей|гимназия|контакты|телефон|адрес|найди|покажи)\b/giu, " ")
+    .replace(/\s+/g, " ")
+    .trim();
+}
+
+function findRelevantItems(items, terms, patterns, layer) {
   return items
     .map((item) => ({
       item,
-      score: scoreItem(item, terms),
+      score: scoreItem(item, terms, patterns, layer),
     }))
     .filter((entry) => entry.score > 0)
     .sort((left, right) => right.score - left.score)
     .map((entry) => entry.item);
 }
 
-function scoreItem(item, terms) {
-  const text = JSON.stringify(selectPublicSummary(item)).toLocaleLowerCase("ru-RU");
-  return terms.reduce((score, term) => score + (text.includes(term.toLocaleLowerCase("ru-RU")) ? 1 : 0), 0);
+function scoreItem(item, terms, patterns, layer) {
+  const summary = selectPublicSummary(item);
+  const text = JSON.stringify(summary).toLocaleLowerCase("ru-RU");
+  const name = String(summary.name || "").toLocaleLowerCase("ru-RU");
+  const address = String(summary.address || "").toLocaleLowerCase("ru-RU");
+  const generalTerms = terms.filter((term) => !/^\d+$/.test(term));
+  let score = generalTerms.reduce((value, term) => value + (text.includes(term.toLocaleLowerCase("ru-RU")) ? 1 : 0), 0);
+
+  for (const inn of patterns.inns) {
+    if (String(summary.inn) === inn) {
+      score += 20;
+    }
+  }
+
+  for (const number of patterns.numbers) {
+    const numberPatterns = [
+      `№ ${number}`,
+      `№${number}`,
+      `школа ${number}`,
+      `сад ${number}`,
+      `лицей ${number}`,
+      `гимназия ${number}`,
+    ];
+
+    if (numberPatterns.some((pattern) => name.includes(pattern))) {
+      score += 12;
+      if (patterns.targetLayers.length === 0 || patterns.targetLayers.includes(layer)) {
+        score += 5;
+      }
+    }
+  }
+
+  for (const street of patterns.streets) {
+    if (street && address.includes(street)) {
+      score += 8;
+    }
+  }
+
+  return score;
 }
 
 function buildAiMessages(question, dataContext, history) {
@@ -522,11 +765,11 @@ async function callAiProvider(config, messages) {
   }
 
   if (config.provider === "openai") {
-    return callOpenAiCompatible(config, messages, process.env.OPENAI_API_KEY, "OpenAI");
+    return callOpenAiCompatible(config, messages, await getApiKey("openai"), "OpenAI");
   }
 
   if (config.provider === "openrouter") {
-    return callOpenAiCompatible(config, messages, process.env.OPENROUTER_API_KEY, "OpenRouter");
+    return callOpenAiCompatible(config, messages, await getApiKey("openrouter"), "OpenRouter");
   }
 
   throw new Error(`Неизвестный AI-провайдер: ${config.provider}`);
@@ -586,6 +829,19 @@ async function callOpenAiCompatible(config, messages, apiKey, providerName) {
   return payload.choices?.[0]?.message?.content || "";
 }
 
+async function getApiKey(provider) {
+  if (provider === "openai" && process.env.OPENAI_API_KEY) {
+    return process.env.OPENAI_API_KEY;
+  }
+
+  if (provider === "openrouter" && process.env.OPENROUTER_API_KEY) {
+    return process.env.OPENROUTER_API_KEY;
+  }
+
+  const secrets = await loadSecrets();
+  return secrets[provider]?.apiKey || "";
+}
+
 async function listLayers(args) {
   const options = parseOptions(args);
   const info = await fetchJson(`${MCP_BASE_URL}/mcp-version`);
@@ -972,6 +1228,19 @@ function mergeConfig(base, override) {
   };
 }
 
+async function loadSecrets() {
+  try {
+    return JSON.parse(await readFile(SECRETS_FILE, "utf8"));
+  } catch {
+    return {};
+  }
+}
+
+async function saveSecrets(value) {
+  await mkdir(CONFIG_DIR, { recursive: true });
+  await writeFile(SECRETS_FILE, `${JSON.stringify(value, null, 2)}\n`, { encoding: "utf8", mode: 0o600 });
+}
+
 async function printAiConfig() {
   const config = await loadConfig();
   printJson({
@@ -980,6 +1249,45 @@ async function printAiConfig() {
   });
 }
 
+function printContext(context) {
+  const layerNames = context.layers.map((layer) => layer.name || layer.id || String(layer));
+  console.log(`Запрос: ${context.query.text}`);
+  console.log(`Слова поиска: ${context.query.terms.length > 0 ? context.query.terms.join(", ") : "-"}`);
+  console.log(`Номера: ${context.query.patterns.numbers.length > 0 ? context.query.patterns.numbers.join(", ") : "-"}`);
+  console.log(`ИНН: ${context.query.patterns.inns.length > 0 ? context.query.patterns.inns.join(", ") : "-"}`);
+  console.log(`Улицы: ${context.query.patterns.streets.length > 0 ? context.query.patterns.streets.join(", ") : "-"}`);
+  console.log(`Целевые слои: ${context.query.patterns.targetLayers.length > 0 ? context.query.patterns.targetLayers.join(", ") : "все"}`);
+  console.log("");
+  console.log(`Слои данных: ${layerNames.length > 0 ? layerNames.join(", ") : "-"}`);
+  console.log("");
+
+  if (context.schools.length > 0) {
+    console.log("Школы в контексте:");
+    printTable(context.schools, [
+      ["name", "Название"],
+      ["address", "Адрес"],
+      ["phone", "Телефон"],
+      ["inn", "ИНН"],
+    ]);
+  } else {
+    console.log("Школы в контексте: нет совпадений");
+  }
+
+  console.log("");
+
+  if (context.kindergartens.length > 0) {
+    console.log("Детские сады в контексте:");
+    printTable(context.kindergartens, [
+      ["name", "Название"],
+      ["address", "Адрес"],
+      ["phone", "Телефон"],
+      ["inn", "ИНН"],
+    ]);
+  } else {
+    console.log("Детские сады в контексте: нет совпадений");
+  }
+}
+
 async function printAiConfigField(field) {
   const config = await loadConfig();
   console.log(config.ai[field] || "-");