@mytegroupinc/myte-core 0.0.14 → 0.0.15

package/README.md

@@ -3,6 +3,8 @@
 Internal implementation package for the `myte` CLI.
 
 Most users should install the unscoped wrapper instead:
+- `npm install myte` then `npx myte ai "Explain this repository"`
+- `npm install myte` then `npx myte ai "Return a JSON object with risks and next_steps" --json-response`
 - `npm install myte` then `npx myte bootstrap`
 - `npm install myte` then `npx myte run-qaqc --mission-ids M001 --wait --sync`
 - `npm install myte` then `npx myte sync-qaqc`
@@ -48,6 +50,7 @@ Requirements:
 - Node `18+`
 - macOS, Linux, or Windows
 - `git` in `PATH` for `--with-diff`
+- `MYTEAI_API_KEY=<inference_api_key>` in env or `.env` for `myte ai`
 - `MYTE_API_KEY=<project_api_key>` in env or `.env`
 - repo folder names must match the project repo names configured in Myte, including casing on case-sensitive filesystems
 
@@ -114,6 +117,8 @@ Deterministic `create-prd` contract:
 - Optional structured fields: `priority`, `status`, `tags`, `assigned_user_email`, `assigned_user_id`, `due_date`, `repo_name`, `repo_id`, `preview_url`, `source`.
 
 Examples:
+- `npx myte ai "Explain what this project does"`
+- `npx myte ai "Return a JSON object with risks and next_steps" --json-response`
 - `npx myte bootstrap`
 - `npx myte run-qaqc --mission-ids "M001,M002" --wait --sync`
 - `npx myte sync-qaqc`

package/cli.js

@@ -11,6 +11,15 @@ const fs = require("fs");
 const path = require("path");
 const { createHash } = require("crypto");
 const { spawnSync } = require("child_process");
+const {
+  DEFAULT_MYTEAI_BASE,
+  buildSimpleAiPayload,
+  callMyteAiChat,
+  extractAssistantTextFromChatCompletion,
+  getMyteAiKey,
+  normalizeJsonAssistantText,
+  normalizeMyteAiBase,
+} = require("./lib/ai-gateway");
 
 const DEFAULT_API_BASE = "https://api.myte.dev";
 const REMOVED_COMMAND_MESSAGES = {
@@ -59,6 +68,7 @@ function loadEnv() {
 function splitCommand(argv) {
   const known = new Set([
     "query",
+    "ai",
     "ask",
     "chat",
     "config",
@@ -91,10 +101,11 @@ function parseArgs(argv) {
   try {
     // eslint-disable-next-line global-require
     const parsed = require("minimist")(argv, {
-      boolean: ["with-diff", "diff", "print-context", "dry-run", "fetch", "json", "stdin", "with-prd-text", "wait", "sync", "force"],
+      boolean: ["with-diff", "diff", "print-context", "dry-run", "fetch", "json", "stdin", "with-prd-text", "wait", "sync", "force", "json-response"],
       string: [
         "query",
         "q",
+        "payload-file",
         "context",
         "ctx",
         "file",
@@ -102,6 +113,9 @@ function parseArgs(argv) {
         "base-url",
         "timeout-ms",
         "diff-limit",
+        "max-output-tokens",
+        "max-tokens",
+        "temperature",
         "title",
         "description",
         "feedback-text",
@@ -171,6 +185,7 @@ function printHelp() {
     "",
     "Usage:",
     "  myte query \"<text>\" [--with-diff] [--context \"...\"]",
+    "  myte ai \"<text>\" [--json-response] [--max-output-tokens 500]",
     "  myte config [--json]",
     "  myte bootstrap [--output-dir ./MyteCommandCenter] [--json]",
     "  myte run-qaqc --mission-ids \"M001[,M002...]\" [--wait] [--sync] [--force] [--json]",
@@ -190,12 +205,17 @@ function printHelp() {
     "",
     "Run forms:",
     "  npm install myte   then npx myte query \"...\" --with-diff",
+    "  npm install myte   then npx myte ai \"Explain this code path\"",
     "  npm install myte   then npm exec myte -- query \"...\" --with-diff",
+    "  npm install myte   then npm exec myte -- ai \"Return JSON only\" --json-response",
     "  npm i -g myte      then myte query \"...\" --with-diff",
+    "  npm i -g myte      then myte ai \"Summarize this file\"",
     "  npx myte@latest query \"What changed in logging?\" --with-diff",
+    "  npx myte@latest ai \"Return a JSON checklist\" --json-response",
     "",
     "Auth:",
     "  - Set MYTE_API_KEY in a workspace .env (or env var)",
+    "  - Set MYTEAI_API_KEY in a workspace .env (or env var) for `myte ai`",
     "",
     "bootstrap contract:",
     "  - Run from the wrapper root that contains the project's configured repo folders",
@@ -253,6 +273,10 @@ function printHelp() {
     "  --diff-limit <chars>  Truncate diff context to N chars (default: 200000)",
     "  --timeout-ms <ms>     Request timeout (default: 300000)",
     "  --base-url <url>      API base (default: https://api.myte.dev)",
+    "  --payload-file <path> Raw OpenAI-style chat-completions payload for `myte ai`",
+    "  --json-response       Ask the Myte AI gateway to return clean JSON only",
+    "  --max-output-tokens   Output token cap for `myte ai` simple queries",
+    "  --temperature <num>   Temperature for `myte ai` simple queries",
     "  --output-dir <path>   Command Center output directory (default: <wrapper-root>/MyteCommandCenter)",
     "  --file <path>         YAML/JSON payload file for suggestions create/revise/review",
     "  --stdin               Read supported command content from stdin instead of inline text or a file path",
@@ -278,6 +302,8 @@ function printHelp() {
     "",
     "Examples:",
     "  myte query \"What changed in logging?\" --with-diff",
+    "  myte ai \"Explain what this repository does\"",
+    "  myte ai \"Return a JSON object with risks and next_steps\" --json-response",
     "  myte bootstrap",
     "  myte suggestions sync",
     "  myte suggestions create",
@@ -3509,6 +3535,111 @@ async function runSuggestions(args) {
   process.exit(1);
 }
 
+async function runAi(args) {
+  const key = getMyteAiKey(process.env);
+  if (!key) {
+    console.error("Missing MYTEAI_API_KEY in environment/.env");
+    process.exit(1);
+  }
+
+  const printContext = Boolean(args["print-context"] || args.printContext || args["dry-run"] || args.dryRun);
+  const timeoutMs = resolveTimeoutMs(args);
+  const baseRaw = args["base-url"] || args.baseUrl || args.base_url || process.env.MYTEAI_API_BASE || process.env.MYTE_AI_API_BASE || DEFAULT_MYTEAI_BASE;
+  const apiBase = normalizeMyteAiBase(baseRaw);
+  const jsonResponse = Boolean(args["json-response"] || args.jsonResponse || args.json_response);
+  const payloadFile = firstNonEmptyString(args["payload-file"], args.payloadFile, args.payload_file);
+
+  let payload;
+  if (payloadFile) {
+    const absPath = resolveInputFile(payloadFile, "AI payload");
+    try {
+      payload = JSON.parse(String(fs.readFileSync(absPath, "utf8") || ""));
+    } catch (err) {
+      console.error(`AI payload must be valid JSON: ${err?.message || err}`);
+      process.exit(1);
+    }
+    if (!payload || typeof payload !== "object" || Array.isArray(payload)) {
+      console.error("AI payload file must contain one JSON object.");
+      process.exit(1);
+    }
+    if (jsonResponse && payload.myte_json_response === undefined) {
+      payload.myte_json_response = true;
+    }
+  } else {
+    const inlineQuery = firstNonEmptyString(args.query, args.q, Array.isArray(args._) ? args._.join(" ") : args._);
+    const useStdin = Boolean(args.stdin || (!process.stdin.isTTY && !inlineQuery));
+    let query = inlineQuery;
+    if (useStdin) {
+      query = String((await readStdinText()) || "").trim();
+    }
+    if (!query) {
+      console.error("Missing AI query text.");
+      printHelp();
+      process.exit(1);
+    }
+    payload = buildSimpleAiPayload({
+      query,
+      jsonResponse,
+      maxOutputTokens: firstNonEmptyString(args["max-output-tokens"], args.maxOutputTokens, args.max_output_tokens, args["max-tokens"], args.maxTokens, args.max_tokens),
+      temperature: firstNonEmptyString(args.temperature),
+    });
+  }
+
+  if (printContext) {
+    console.log(JSON.stringify(payload, null, 2));
+    return;
+  }
+
+  const fetchFn = await getFetch();
+  let responseBody;
+  try {
+    responseBody = await callMyteAiChat({
+      fetchFn,
+      apiBase,
+      apiKey: key,
+      payload,
+      timeoutMs,
+    });
+  } catch (err) {
+    if (err?.name === "AbortError") {
+      console.error(`Request timed out after ${timeoutMs}ms`);
+    } else {
+      console.error("Myte AI request failed:", err?.message || err);
+    }
+    process.exit(1);
+  }
+
+  let content = extractAssistantTextFromChatCompletion(responseBody);
+  let normalizedJson = null;
+  if (jsonResponse) {
+    try {
+      normalizedJson = normalizeJsonAssistantText(content);
+      content = normalizedJson.text;
+    } catch {
+      content = String(content || "").trim();
+    }
+  }
+
+  if (args.json) {
+    console.log(
+      JSON.stringify(
+        {
+          id: responseBody.id || null,
+          model: responseBody.model || null,
+          usage: responseBody.usage || null,
+          content,
+          parsed_json: normalizedJson ? normalizedJson.parsed : null,
+        },
+        null,
+        2
+      )
+    );
+    return;
+  }
+
+  console.log(content || "");
+}
+
 async function runQuery(args) {
   const key = (process.env.MYTE_API_KEY || process.env.MYTE_PROJECT_API_KEY || "").trim();
   if (!key) {
@@ -3629,6 +3760,11 @@ async function main() {
     return;
   }
 
+  if (command === "ai") {
+    await runAi(args);
+    return;
+  }
+
   if (command === "bootstrap") {
     await runBootstrap(args);
     return;

package/lib/ai-gateway.js

@@ -0,0 +1,112 @@
+"use strict";
+
+const DEFAULT_MYTEAI_BASE = "https://api.myte.ai/v1";
+
+function normalizeMyteAiBase(baseRaw) {
+  const baseTrim = String(baseRaw || "").trim().replace(/\/+$/, "");
+  const base = baseTrim || DEFAULT_MYTEAI_BASE;
+  return /\/v1$/i.test(base) ? base : `${base}/v1`;
+}
+
+function getMyteAiKey(env = process.env) {
+  return String(env.MYTEAI_API_KEY || env.MYTE_AI_API_KEY || "").trim();
+}
+
+function buildSimpleAiPayload({ query, jsonResponse = false, maxOutputTokens, temperature }) {
+  const payload = {
+    messages: [{ role: "user", content: String(query || "").trim() }],
+  };
+  if (jsonResponse) payload.myte_json_response = true;
+  if (Number.isFinite(Number(maxOutputTokens)) && Number(maxOutputTokens) > 0) {
+    payload.max_tokens = Number(maxOutputTokens);
+  }
+  if (temperature !== undefined && temperature !== null && temperature !== "") {
+    const parsed = Number(temperature);
+    if (Number.isFinite(parsed)) payload.temperature = parsed;
+  }
+  return payload;
+}
+
+function extractAssistantTextFromChatCompletion(payload) {
+  const choices = Array.isArray(payload?.choices) ? payload.choices : [];
+  const first = choices[0] && typeof choices[0] === "object" ? choices[0] : {};
+  const message = first.message && typeof first.message === "object" ? first.message : {};
+  const content = message.content;
+  if (typeof content === "string") return content.trim();
+  if (Array.isArray(content)) {
+    return content
+      .map((item) => {
+        if (typeof item === "string") return item;
+        if (!item || typeof item !== "object") return "";
+        if (typeof item.text === "string") return item.text;
+        if (typeof item.content === "string") return item.content;
+        return "";
+      })
+      .filter(Boolean)
+      .join("\n")
+      .trim();
+  }
+  return "";
+}
+
+function stripJsonFences(text) {
+  const raw = String(text || "").trim();
+  const match = raw.match(/^```(?:json)?\s*([\s\S]*?)\s*```$/i);
+  if (match) return String(match[1] || "").trim();
+  return raw;
+}
+
+function normalizeJsonAssistantText(text) {
+  const stripped = stripJsonFences(text);
+  const parsed = JSON.parse(stripped);
+  return {
+    parsed,
+    text: JSON.stringify(parsed, null, 2),
+  };
+}
+
+async function callMyteAiChat({ fetchFn, apiBase, apiKey, payload, timeoutMs }) {
+  const controller = typeof AbortController !== "undefined" ? new AbortController() : undefined;
+  const timeoutId =
+    controller && timeoutMs > 0 ? setTimeout(() => controller.abort(), timeoutMs) : undefined;
+  try {
+    const response = await fetchFn(`${apiBase}/chat/completions`, {
+      method: "POST",
+      headers: {
+        "Content-Type": "application/json",
+        Authorization: `Bearer ${apiKey}`,
+      },
+      signal: controller?.signal,
+      body: JSON.stringify(payload),
+    });
+    const text = await response.text();
+    let body;
+    try {
+      body = JSON.parse(text);
+    } catch (error) {
+      const err = new Error(`Non-JSON response (${response.status}): ${text.slice(0, 500)}`);
+      err.status = response.status;
+      throw err;
+    }
+    if (!response.ok) {
+      const err = new Error(body?.error?.message || body?.message || `Request failed (${response.status})`);
+      err.status = response.status;
+      err.body = body;
+      throw err;
+    }
+    return body;
+  } finally {
+    if (timeoutId) clearTimeout(timeoutId);
+  }
+}
+
+module.exports = {
+  DEFAULT_MYTEAI_BASE,
+  buildSimpleAiPayload,
+  callMyteAiChat,
+  extractAssistantTextFromChatCompletion,
+  getMyteAiKey,
+  normalizeJsonAssistantText,
+  normalizeMyteAiBase,
+  stripJsonFences,
+};

package/package.json

@@ -1,12 +1,13 @@
 {
   "name": "@mytegroupinc/myte-core",
-  "version": "0.0.14",
-  "description": "Myte CLI core implementation (Project Assistant + deterministic diffs).",
+  "version": "0.0.15",
+  "description": "Myte CLI core implementation (Project Assistant + Myte AI gateway).",
   "type": "commonjs",
   "main": "cli.js",
   "files": [
     "README.md",
     "cli.js",
+    "lib",
     "package.json"
   ],
   "scripts": {