@baryonlabs/cli 0.2.1 → 0.2.2

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@baryonlabs/cli",
-  "version": "0.2.1",
+  "version": "0.2.2",
   "description": "Baryon CLI — AI 코딩·학습 에이전트. baryon.ai API에 기본 연결된 pi 코딩 에이전트 래퍼. 한 줄 설치, 상용·로컬 모델 전환.",
   "type": "module",
   "bin": {

package/src/config.js

@@ -11,6 +11,8 @@ import {
   PI_AGENT_DIR,
   PI_MODELS_JSON,
   PROVIDER,
+  SESSION_HEADER,
+  SESSION_ID_ENV,
 } from "./constants.js";
 
 function readJson(file, fallback) {
@@ -69,6 +71,9 @@ export function syncPiModels({ baseUrl, models }) {
     api: "openai-completions",
     apiKey: `$${API_KEY_ENV}`,
     authHeader: true,
+    // Per-launch session id (resolved by pi from the env at request time) so the
+    // gateway can group a run's turns into one session. Gateway requires it.
+    headers: { [SESSION_HEADER]: `$${SESSION_ID_ENV}` },
     models:
       Array.isArray(models) && models.length ? models : DEFAULT_MODELS,
   };
@@ -83,4 +88,19 @@ export function piProviderConfigured() {
   return Boolean(root?.providers?.[PROVIDER]);
 }
 
+/**
+ * Auto-heal older installs: ensure the baryon provider sends the session
+ * header. Safe to call on every launch — only writes when something changed.
+ * Returns true if the provider exists (after any patch).
+ */
+export function ensurePiSessionHeader() {
+  const root = readJson(PI_MODELS_JSON, {});
+  const p = root?.providers?.[PROVIDER];
+  if (!p) return false;
+  if (p.headers?.[SESSION_HEADER] === `$${SESSION_ID_ENV}`) return true;
+  p.headers = { ...(p.headers || {}), [SESSION_HEADER]: `$${SESSION_ID_ENV}` };
+  writeJson(PI_MODELS_JSON, root);
+  return true;
+}
+
 export { PI_MODELS_JSON, BARYON_CONFIG };

package/src/constants.js

@@ -11,6 +11,14 @@ export const DEFAULT_BASE_URL =
 /** Env var pi resolves at request time (apiKey: "$BARYON_API_KEY"). */
 export const API_KEY_ENV = "BARYON_API_KEY";
 
+/**
+ * Session id env var. The baryon provider sends it as `X-Baryon-Session` so the
+ * gateway can group every turn of one `baryon` run into a single session. A
+ * fresh id is minted per launch (see pi.js). The gateway requires it.
+ */
+export const SESSION_ID_ENV = "BARYON_SESSION_ID";
+export const SESSION_HEADER = "X-Baryon-Session";
+
 /** Underlying coding agent package + binary. */
 export const PI_PACKAGE = "@earendil-works/pi-coding-agent";
 export const PI_BIN = "pi";

package/src/pi.js

@@ -3,6 +3,7 @@
 import { spawn } from "node:child_process";
 import { createRequire } from "node:module";
 import { fileURLToPath } from "node:url";
+import { randomUUID } from "node:crypto";
 import fs from "node:fs";
 import path from "node:path";
 import {
@@ -10,7 +11,9 @@ import {
   PI_BIN,
   PI_PACKAGE,
   PROVIDER,
+  SESSION_ID_ENV,
 } from "./constants.js";
+import { ensurePiSessionHeader } from "./config.js";
 
 const require = createRequire(import.meta.url);
 
@@ -98,6 +101,11 @@ export function runPi(args, config, { injectTargeting = true } = {}) {
   if (config.apiKey) env[API_KEY_ENV] = config.apiKey;
   if (config.baseUrl) env.BARYON_BASE_URL = config.baseUrl;
 
+  // One session per launch: mint an id (unless the caller pinned one) and make
+  // sure the provider forwards it. The gateway requires a session id.
+  if (!env[SESSION_ID_ENV]) env[SESSION_ID_ENV] = `cli_${randomUUID()}`;
+  ensurePiSessionHeader();
+
   return new Promise((resolve, reject) => {
     const child = spawn(process.execPath, [entry, ...finalArgs], {
       stdio: "inherit",