@automagik/omni 2.260529.2 → 2.260530.1

package/dist/commands/listen.d.ts

@@ -2,6 +2,7 @@
  * Listen Command — Speech-to-text verb
  *
  * omni listen audio.ogg                       — print transcription to stdout
+ * omni listen audio.ogg --provider openai     — force OpenAI STT (quality default)
  * omni listen audio.ogg --provider gemini     — force Gemini STT (handles >19.5MB)
  * omni listen audio.ogg --provider groq       — force Groq Whisper (fast, 19.5MB cap)
  * omni listen audio.ogg --language pt         — language hint

package/dist/commands/listen.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"listen.d.ts","sourceRoot":"","sources":["../../src/commands/listen.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAIH,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AA2DpC,wBAAgB,mBAAmB,IAAI,OAAO,CAwH7C"}
+{"version":3,"file":"listen.d.ts","sourceRoot":"","sources":["../../src/commands/listen.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;GAeG;AAIH,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AA8DpC,wBAAgB,mBAAmB,IAAI,OAAO,CAqI7C"}

package/dist/index.js

@@ -35857,6 +35857,82 @@ function buildProviderRequestContext(context) {
 }
 var OMNI_EXECUTION_CONTEXT_EXTENSION_URI = "https://omni.dev/extensions/execution-context/v1";
 
+// ../core/src/providers/trace-context.ts
+import { createHash as createHash3 } from "crypto";
+function isNonZeroHex(value) {
+  return value.length > 0 && !/^0+$/.test(value);
+}
+function hashHex(value, length) {
+  return createHash3("sha256").update(value).digest("hex").slice(0, length);
+}
+function normalizeTraceId(value) {
+  const lower = value.toLowerCase();
+  if (W3C_TRACE_ID.test(lower) && isNonZeroHex(lower)) {
+    return lower;
+  }
+  return hashHex(lower, 32);
+}
+function normalizeSpanId(value) {
+  const lower = value.toLowerCase();
+  if (W3C_SPAN_ID.test(lower) && isNonZeroHex(lower)) {
+    return lower;
+  }
+  return hashHex(lower, 16);
+}
+function createTraceContextFromTraceId(traceId, spanSeed = "provider") {
+  if (!traceId)
+    return;
+  return {
+    traceId: normalizeTraceId(traceId),
+    spanId: normalizeSpanId(`${traceId}:${spanSeed}`),
+    traceFlags: 1
+  };
+}
+function formatTraceparent(ctx) {
+  const traceFlags = (ctx.traceFlags ?? 1) & 255;
+  const paddedFlagsHex = traceFlags.toString(16).padStart(2, "0");
+  return `00-${normalizeTraceId(ctx.traceId)}-${normalizeSpanId(ctx.spanId)}-${paddedFlagsHex}`;
+}
+function buildTraceHeaders2(ctx, khal) {
+  const traceHeaders = {};
+  if (ctx) {
+    traceHeaders.traceparent = formatTraceparent(ctx);
+    traceHeaders["x-trace-id"] = normalizeTraceId(ctx.traceId);
+    traceHeaders["x-span-id"] = normalizeSpanId(ctx.spanId);
+    const tracestate = ctx.tracestate ?? ctx.traceState;
+    if (tracestate) {
+      traceHeaders.tracestate = tracestate;
+    }
+    if (ctx.parentSpanId) {
+      traceHeaders["x-parent-span-id"] = normalizeSpanId(ctx.parentSpanId);
+    }
+  }
+  if (khal?.khalSessionId) {
+    traceHeaders["x-khal-session-id"] = khal.khalSessionId;
+  }
+  if (khal?.userId) {
+    traceHeaders["x-khal-user-id"] = khal.userId;
+  }
+  if (khal?.messageId) {
+    traceHeaders["x-khal-message-id"] = khal.messageId;
+  }
+  if (khal?.omni?.instanceId) {
+    traceHeaders["x-omni-instance-id"] = khal.omni.instanceId;
+  }
+  if (khal?.omni?.chatId) {
+    traceHeaders["x-omni-chat-id"] = khal.omni.chatId;
+  }
+  if (khal?.omni?.channel) {
+    traceHeaders["x-omni-channel"] = khal.omni.channel;
+  }
+  return traceHeaders;
+}
+var W3C_TRACE_ID, W3C_SPAN_ID;
+var init_trace_context = __esm(() => {
+  W3C_TRACE_ID = /^[0-9a-f]{32}$/i;
+  W3C_SPAN_ID = /^[0-9a-f]{16}$/i;
+});
+
 // ../core/src/providers/agno-client.ts
 class AgnoClient {
   baseUrl;
@@ -35985,12 +36061,14 @@ class AgnoClient {
     const formData = new FormData;
     formData.append("message", request.message);
     formData.append("stream", String(stream));
-    if (request.sessionId) {
-      formData.append("session_id", request.sessionId);
+    const sessionId = request.khalSessionId ?? request.sessionId;
+    if (sessionId) {
+      formData.append("session_id", sessionId);
     }
     if (request.userId) {
       formData.append("user_id", request.userId);
     }
+    this.appendStructuredMetadata(formData, request);
     if (request.files?.length) {
       for (const file of request.files) {
         try {
@@ -36008,6 +36086,28 @@ class AgnoClient {
     }
     return formData;
   }
+  appendStructuredMetadata(formData, request) {
+    const appendJson = (key, value) => {
+      if (value !== undefined) {
+        formData.append(key, JSON.stringify(value));
+      }
+    };
+    if (request.khalSessionId) {
+      formData.append("khal_session_id", request.khalSessionId);
+    }
+    if (request.messageId) {
+      formData.append("message_id", request.messageId);
+    }
+    if (request.replyToMessageId) {
+      formData.append("reply_to_message_id", request.replyToMessageId);
+    }
+    appendJson("platform", request.platform);
+    appendJson("sender", request.sender);
+    appendJson("chat", request.chat);
+    appendJson("mcp_url_params", request.mcpUrlParams);
+    appendJson("env", request.env);
+    appendJson("omni", request.omni);
+  }
   async runAgent(agentId, request) {
     return this.runEndpoint("agents", agentId, request);
   }
@@ -36023,7 +36123,15 @@ class AgnoClient {
     const timeoutMs = request.timeoutMs ?? this.defaultTimeoutMs;
     const response = await this.fetchWithTimeout(url, {
       method: "POST",
-      headers: this.getHeaders(),
+      headers: {
+        ...this.getHeaders(),
+        ...buildTraceHeaders2(request.traceContext, {
+          khalSessionId: request.khalSessionId ?? request.sessionId,
+          userId: request.userId,
+          messageId: request.omni?.messageId,
+          omni: request.omni
+        })
+      },
       body: formData
     }, timeoutMs);
     if (!response.ok) {
@@ -36057,7 +36165,15 @@ class AgnoClient {
     const timeoutMs = request.timeoutMs ?? this.defaultTimeoutMs;
     const response = await this.fetchWithTimeout(url, {
       method: "POST",
-      headers: this.getHeaders(),
+      headers: {
+        ...this.getHeaders(),
+        ...buildTraceHeaders2(request.traceContext, {
+          khalSessionId: request.khalSessionId ?? request.sessionId,
+          userId: request.userId,
+          messageId: request.omni?.messageId,
+          omni: request.omni
+        })
+      },
       body: formData
     }, timeoutMs);
     if (!response.ok) {
@@ -36197,6 +36313,7 @@ function createAgnoClient(config2) {
 }
 var DEFAULT_TIMEOUT_MS = 60000;
 var init_agno_client = __esm(() => {
+  init_trace_context();
   init_types6();
 });
 
@@ -60251,12 +60368,7 @@ class AgnoAgentProvider {
   }
   async trigger(context) {
     const startTime = Date.now();
-    let message2 = "";
-    if (context.content.text) {
-      message2 = context.content.text;
-    } else if (context.content.emoji) {
-      message2 = `[Reaction: ${context.content.emoji} on message ${context.content.referencedMessageId ?? context.source.messageId}]`;
-    }
+    const message2 = this.buildMessage(context);
     if (!message2) {
       log8.debug("No content to send to agent", { traceId: context.traceId });
       return {
@@ -60268,17 +60380,7 @@ class AgnoAgentProvider {
         }
       };
     }
-    if (this.config.prefixSenderName !== false && context.sender.displayName) {
-      message2 = `[${context.sender.displayName}]: ${message2}`;
-    }
-    const request = {
-      ...buildProviderRequestContext(context),
-      message: message2,
-      agentId: this.config.agentId,
-      agentType: this.config.agentType,
-      stream: false,
-      timeoutMs: this.config.timeoutMs ?? 60000
-    };
+    const request = this.buildRequest(context, message2, false);
     log8.info("Triggering Agno agent", {
       agentId: this.config.agentId,
       agentType: this.config.agentType,
@@ -60310,13 +60412,79 @@ class AgnoAgentProvider {
       }
     };
   }
+  async* triggerStream(context) {
+    const message2 = this.buildMessage(context);
+    if (!message2) {
+      log8.debug("No content to send to Agno agent (stream)", { traceId: context.traceId });
+      return;
+    }
+    const request = this.buildRequest(context, message2, true);
+    log8.info("Streaming Agno agent", {
+      agentId: this.config.agentId,
+      agentType: this.config.agentType,
+      triggerType: context.type,
+      traceId: context.traceId
+    });
+    try {
+      for await (const chunk of this.client.stream(request)) {
+        if (chunk.content) {
+          yield { phase: "content", content: chunk.content };
+        }
+        if (chunk.isComplete) {
+          yield { phase: "final", content: chunk.fullContent ?? chunk.content ?? "" };
+        }
+      }
+    } catch (error2) {
+      const message3 = error2 instanceof Error ? error2.message : String(error2);
+      log8.error("Error in Agno agent stream", {
+        agentId: this.config.agentId,
+        traceId: context.traceId,
+        error: message3
+      });
+      yield { phase: "error", error: message3 };
+    }
+  }
   async checkHealth() {
     return this.client.checkHealth();
   }
+  buildMessage(context) {
+    let message2 = "";
+    if (context.content.text) {
+      message2 = context.content.text;
+    } else if (context.content.emoji) {
+      message2 = `[Reaction: ${context.content.emoji} on message ${context.content.referencedMessageId ?? context.source.messageId}]`;
+    }
+    if (message2 && this.config.prefixSenderName !== false && context.sender.displayName) {
+      message2 = `[${context.sender.displayName}]: ${message2}`;
+    }
+    return message2;
+  }
+  buildRequest(context, message2, stream) {
+    const traceContext = context.traceContext ?? createTraceContextFromTraceId(context.traceId, `${context.source.instanceId}:${context.source.chatId}:${context.source.messageId}:agno`);
+    return {
+      ...buildProviderRequestContext(context),
+      message: message2,
+      agentId: this.config.agentId,
+      agentType: this.config.agentType,
+      stream,
+      timeoutMs: this.config.timeoutMs ?? 60000,
+      files: context.content.files,
+      traceContext,
+      khalSessionId: context.sessionId,
+      omni: {
+        instanceId: context.source.instanceId,
+        chatId: context.source.chatId,
+        messageId: context.source.messageId,
+        channel: context.source.channelType
+      },
+      ...context.source.chatId ? { mcpUrlParams: { chat_id: context.source.chatId } } : {}
+    };
+  }
 }
 var log8;
 var init_agno_provider = __esm(() => {
   init_logger();
+  init_trace_context();
   log8 = createLogger("provider:agno");
 });
 
@@ -62523,6 +62691,7 @@ var init_a2a_provider = __esm(() => {
 });
 
 // ../core/src/providers/nats-genie-provider.ts
+import { createHash as createHash5 } from "crypto";
 import { mkdir, writeFile } from "fs/promises";
 import { homedir as homedir10 } from "os";
 import { join as join16 } from "path";
@@ -62556,6 +62725,8 @@ class NatsGenieProvider {
       };
     }
     const topic = `omni.message.${this.config.instanceId}.${context.source.chatId}`;
+    const traceContext = this.resolveTraceContext(context);
+    const propagationHeaders = this.buildPropagationHeaders(context, traceContext);
     const payload = {
       content: message2,
       sender: context.sender.displayName ?? context.sender.platformUserId,
@@ -62565,22 +62736,28 @@ class NatsGenieProvider {
       timestamp: new Date().toISOString(),
       traceId: context.traceId,
       messageId: context.source.messageId,
+      contextMessages: context.contextMessages,
       files: context.content.files,
+      traceContext,
+      metadata: this.buildSafePayloadMetadata(propagationHeaders),
       env: buildOmniEnv(context),
       executionContext: buildOmniExecutionContext(context)
     };
     try {
       await this.ensureConnected();
-      this.nc?.publish(topic, this.sc.encode(JSON.stringify(payload)));
+      this.nc?.publish(topic, this.sc.encode(JSON.stringify(payload)), this.buildPublishOptions(propagationHeaders));
       log17.info("Published to NATS", {
-        topic,
+        subject: "omni.message.<instance>.<chat_hash>",
         agent: this.config.agentName,
-        chatId: context.source.chatId,
+        instanceId: this.config.instanceId,
+        chatHash: this.safeHash(context.source.chatId),
         traceId: context.traceId
       });
     } catch (error2) {
       log17.error("Failed to publish to NATS, writing to dead-letter queue", {
-        topic,
+        subject: "omni.message.<instance>.<chat_hash>",
+        instanceId: this.config.instanceId,
+        chatHash: this.safeHash(context.source.chatId),
         error: error2 instanceof Error ? error2.message : String(error2),
         traceId: context.traceId
       });
@@ -62623,7 +62800,8 @@ class NatsGenieProvider {
           }
         } catch (error2) {
           log17.error("Error processing reply", {
-            subject: msg.subject,
+            subject: "omni.reply.<instance>.<chat_hash>",
+            subjectHash: this.safeHash(msg.subject),
             error: error2 instanceof Error ? error2.message : String(error2)
           });
         }
@@ -62648,7 +62826,7 @@ class NatsGenieProvider {
     if (!chatId) {
       log17.warn("NATS session reset skipped: missing chatId", {
         providerId: this.id,
-        sessionKey
+        sessionKeyHash: this.safeHash(sessionKey)
       });
       throw new Error("chatId is required to reset NATS Genie session");
     }
@@ -62665,14 +62843,17 @@ class NatsGenieProvider {
       await this.ensureConnected();
       this.nc?.publish(topic, this.sc.encode(JSON.stringify(payload)));
       log17.info("Published session reset to NATS", {
-        topic,
+        subject: "omni.session.reset.<instance>.<chat_hash>",
         providerId: this.id,
-        sessionKey,
-        chatId
+        instanceId: resolvedInstanceId,
+        sessionKeyHash: this.safeHash(sessionKey),
+        chatHash: this.safeHash(chatId)
       });
     } catch (error2) {
       log17.error("Failed to publish NATS session reset", {
-        topic,
+        subject: "omni.session.reset.<instance>.<chat_hash>",
+        instanceId: resolvedInstanceId,
+        chatHash: this.safeHash(chatId),
         error: error2 instanceof Error ? error2.message : String(error2)
       });
       throw error2;
@@ -62702,6 +62883,48 @@ class NatsGenieProvider {
     });
     log17.info("Connected to NATS", { url: this.config.natsUrl });
   }
+  resolveTraceContext(context) {
+    return context.traceContext ?? createTraceContextFromTraceId(context.traceId, `${context.source.instanceId}:${context.source.chatId}:${context.source.messageId}:nats-genie`);
+  }
+  buildPropagationHeaders(context, traceContext) {
+    return buildTraceHeaders2(traceContext, {
+      khalSessionId: context.sessionId,
+      userId: context.sender.personId ?? context.sender.platformUserId,
+      messageId: context.source.messageId,
+      omni: {
+        instanceId: context.source.instanceId,
+        chatId: context.source.chatId,
+        channel: context.source.channelType
+      }
+    });
+  }
+  buildPublishOptions(traceHeaders) {
+    if (Object.keys(traceHeaders).length === 0)
+      return;
+    const natsHeaders2 = import_nats2.headers();
+    for (const [key, value] of Object.entries(traceHeaders)) {
+      natsHeaders2.set(key, value);
+    }
+    return { headers: natsHeaders2 };
+  }
+  buildSafePayloadMetadata(traceHeaders) {
+    const metadata = {};
+    for (const [key, value] of Object.entries(traceHeaders)) {
+      if (key === "x-khal-user-id") {
+        metadata["x-khal-user-hash"] = this.safeHash(value);
+        continue;
+      }
+      if (key === "x-omni-chat-id") {
+        metadata["x-omni-chat-hash"] = this.safeHash(value);
+        continue;
+      }
+      metadata[key] = value;
+    }
+    return metadata;
+  }
+  safeHash(value) {
+    return createHash5("sha256").update(value).digest("hex").slice(0, 12);
+  }
   buildMessage(context) {
     let message2 = "";
     if (context.content.text) {
@@ -62735,6 +62958,7 @@ class NatsGenieProvider {
 var import_nats2, log17;
 var init_nats_genie_provider = __esm(() => {
   init_logger();
+  init_trace_context();
   import_nats2 = __toESM(require_mod4(), 1);
   log17 = createLogger("provider:nats-genie");
 });
@@ -124293,7 +124517,10 @@ function createOmniClient(config) {
             provider: body.provider,
             language: body.language,
             timestamps: body.timestamps,
-            model: body.model
+            model: body.model,
+            prompt: body.prompt,
+            context: body.context,
+            glossary: body.glossary
           })
         });
         const json = await resp.json();
@@ -124723,7 +124950,7 @@ import { fileURLToPath } from "url";
 // package.json
 var package_default = {
   name: "@automagik/omni",
-  version: "2.260529.2",
+  version: "2.260530.1",
   description: "LLM-optimized CLI for Omni",
   type: "module",
   bin: {
@@ -133456,7 +133683,7 @@ function guessAudioMimeType(filePath) {
   }
 }
 function createListenCommand() {
-  return new Command("listen").description("Transcribe an audio file to text and print (or send as reply)").argument("<file>", "Path to the audio file to transcribe").option("--provider <name>", "STT provider (gemini, groq). Default: server config.").option("--language <code>", "BCP-47 language hint (e.g. en, pt-BR)").option("--timestamps", "Include per-segment timestamps").option("--format <fmt>", `Output format: ${ALLOWED_FORMATS.join(", ")} (default: text)`).option("--model <name>", "Model override (provider-specific)").option("--reply [message-id]", "Send transcript as a quote-reply to the trigger or a specific message").option("--instance <id>", "Override instance (default: from context, only with --reply)").option("--chat <id>", "Override chat (default: from context, only with --reply)").action(async (file, options) => {
+  return new Command("listen").description("Transcribe an audio file to text and print (or send as reply)").argument("<file>", "Path to the audio file to transcribe").option("--provider <name>", "STT provider (openai, gemini, groq). Default: server config.").option("--language <code>", "BCP-47 language hint (e.g. en, pt-BR)").option("--timestamps", "Include per-segment timestamps").option("--format <fmt>", `Output format: ${ALLOWED_FORMATS.join(", ")} (default: text)`).option("--model <name>", "Model override (provider-specific)").option("--prompt <text>", "Provider transcription prompt/instructions").option("--context <text>", "Domain context for acoustic disambiguation").option("--glossary <terms>", "Comma-separated likely names/acronyms/products").option("--reply [message-id]", "Send transcript as a quote-reply to the trigger or a specific message").option("--instance <id>", "Override instance (default: from context, only with --reply)").option("--chat <id>", "Override chat (default: from context, only with --reply)").action(async (file, options) => {
     const client = getClient();
     const format = parseFormat(options.format);
     const wantTimestamps = options.timestamps === true;
@@ -133471,6 +133698,7 @@ function createListenCommand() {
       return error(`File is empty: ${file}`);
     }
     const mimeType = guessAudioMimeType(file);
+    const glossary = options.glossary ? options.glossary.split(",").map((term) => term.trim()).filter(Boolean) : undefined;
     let result;
     try {
       result = await client.media.stt({
@@ -133479,7 +133707,10 @@ function createListenCommand() {
         provider: options.provider,
         language: options.language,
         timestamps: wantTimestamps,
-        model: options.model
+        model: options.model,
+        prompt: options.prompt,
+        context: options.context,
+        glossary
       });
     } catch (err2) {
       const message2 = err2 instanceof Error ? err2.message : "Unknown error";

package/dist/sdk/client.d.ts

@@ -1901,6 +1901,9 @@ export declare function createOmniClient(config: OmniClientConfig): {
             language?: string;
             timestamps?: boolean;
             model?: string;
+            prompt?: string;
+            context?: string;
+            glossary?: string[];
         }): Promise<{
             provider: string;
             text: string;