@intelli-1113/stratos-sdk 1.2.2 → 1.2.3

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@intelli-1113/stratos-sdk",
-  "version": "1.2.2",
+  "version": "1.2.3",
   "description": "One install for Stratos AI: OpenLLMetry agent telemetry (--import) AND the stratos-mcp-proxy for MCP servers.",
   "type": "module",
   "main": "./src/index.js",

package/src/index.js

@@ -48,7 +48,8 @@ function detectFramework() {
 // Best-effort model from common env vars (or opts.model). Spans still refine it.
 function detectModel(opts) {
   return opts.model || process.env.STRATOS_MODEL || process.env.NVIDIA_MODEL || process.env.OPENAI_MODEL
-    || process.env.ANTHROPIC_MODEL || process.env.LLM_MODEL || process.env.MODEL || null;
+    || process.env.ANTHROPIC_MODEL || process.env.GEMINI_MODEL || process.env.GOOGLE_MODEL
+    || process.env.LLM_MODEL || process.env.MODEL || null;
 }
 
 let started = false;
@@ -91,6 +92,22 @@ export function start(opts = {}) {
   ping();
   if (heartbeatMs > 0) setInterval(ping, heartbeatMs).unref();
 
+  // disableBatch exports each span the moment it ends via an async HTTP POST —
+  // fine for long-running servers, but a one-shot script (no open server/handle)
+  // drains the event loop and exits the instant it's done, killing that request
+  // mid-flight before it reaches Stratos. 'beforeExit' fires exactly at that point
+  // (and, unlike 'exit', still lets us start async work), so flushing there makes
+  // every consumer — script or server — deliver its spans with zero code changes.
+  process.on("beforeExit", () => {
+    traceloop.forceFlush().catch(() => {});
+  });
+  // Also flush on Ctrl+C / kill so a killed server doesn't lose its last spans.
+  for (const sig of ["SIGINT", "SIGTERM"]) {
+    process.on(sig, () => {
+      traceloop.forceFlush().finally(() => process.exit(0));
+    });
+  }
+
   console.log(`[stratos] telemetry → ${ingest} (heartbeat ${heartbeatMs}ms, framework=${meta.framework || "?"})`);
 }