@agentmark-ai/shared-utils 0.3.1 → 0.3.2

package/dist/index.mjs

@@ -826,8 +826,7 @@ var KNOWN_METADATA_FIELDS = /* @__PURE__ */ new Set([
   "dataset_expected_output",
   "dataset_input",
   "prompt_name",
-  "props",
-  "commit_sha"
+  "props"
 ]);
 function parseMetadata(attributes, prefix = "agentmark.metadata.") {
   const result = {};
@@ -1612,7 +1611,7 @@ function parseAgentMarkAttributes(attributes, prefix = "agentmark.") {
   if (get("trace_name")) result.traceName = String(get("trace_name"));
   if (get("prompt_name")) result.promptName = String(get("prompt_name"));
   if (get("props")) result.props = String(get("props"));
-  if (get("span.kind")) result.kind = String(get("span.kind"));
+  if (get("span.kind")) result.semanticKind = String(get("span.kind"));
   if (get("dataset_run_id")) result.datasetRunId = String(get("dataset_run_id"));
   if (get("dataset_run_name")) result.datasetRunName = String(get("dataset_run_name"));
   if (get("dataset_item_name")) result.datasetItemName = String(get("dataset_item_name"));
@@ -1682,6 +1681,9 @@ var AgentMarkTransformer = class {
     if (operationName === OperationNames.CHAT || operationName === OperationNames.TEXT_COMPLETION) {
       return "GENERATION" /* GENERATION */;
     }
+    if (operationName === OperationNames.EMBEDDINGS) {
+      return "GENERATION" /* GENERATION */;
+    }
     if (span.name === SpanNames.CHAT || span.name.startsWith(SpanNames.CHAT + " ")) {
       return "GENERATION" /* GENERATION */;
     }
@@ -2104,6 +2106,83 @@ function extractResourceScopeSpan(resourceSpans) {
   return result;
 }
 
+// src/normalizer/resolvers/semantic-kind-resolver.ts
+var SEMANTIC_KINDS = ["function", "llm", "tool", "agent", "retrieval", "embedding", "guardrail"];
+var VALID_KINDS = new Set(SEMANTIC_KINDS);
+var OPENINFERENCE_MAP = {
+  "CHAIN": "function",
+  "LLM": "llm",
+  "TOOL": "tool",
+  "AGENT": "agent",
+  "RETRIEVER": "retrieval",
+  "EMBEDDING": "embedding",
+  "GUARDRAIL": "guardrail",
+  "RERANKER": "retrieval"
+};
+var FRAMEWORK_MAPPINGS = [
+  {
+    key: "ai.operationId",
+    // Vercel AI SDK
+    map: {
+      "embed": "embedding",
+      "ai.embed": "embedding",
+      "generateText": "llm",
+      "streamText": "llm",
+      "generateObject": "llm",
+      "streamObject": "llm"
+    }
+  },
+  {
+    key: "traceloop.span.kind",
+    // Traceloop / OpenLLMetry
+    map: { "LLM": "llm", "TOOL": "tool", "AGENT": "agent", "WORKFLOW": "function", "TASK": "function" }
+  },
+  {
+    key: "langchain.run_type",
+    // LangChain via OTLP
+    map: { "llm": "llm", "chat_model": "llm", "retriever": "retrieval", "tool": "tool", "chain": "function", "embedding": "embedding" }
+  },
+  {
+    key: "genkit:type",
+    // Firebase Genkit
+    map: { "model": "llm", "tool": "tool", "flow": "function", "retriever": "retrieval", "embedder": "embedding" }
+  }
+];
+function resolveSemanticKind(normalized, allAttributes) {
+  if (normalized.semanticKind && VALID_KINDS.has(normalized.semanticKind)) {
+    return normalized.semanticKind;
+  }
+  const oiKind = allAttributes["openinference.span.kind"];
+  if (oiKind) {
+    const mapped = OPENINFERENCE_MAP[String(oiKind).toUpperCase()];
+    if (mapped) return mapped;
+  }
+  for (const { key, map } of FRAMEWORK_MAPPINGS) {
+    const val = allAttributes[key];
+    if (val) {
+      const mapped = map[String(val)];
+      if (mapped) return mapped;
+    }
+  }
+  const opName = allAttributes["gen_ai.operation.name"];
+  if (opName) {
+    const op = String(opName).toLowerCase();
+    if (op === "chat" || op === "text_completion" || op === "generate_content") return "llm";
+    if (op === "embeddings") return "embedding";
+  }
+  if (normalized.type === "GENERATION" /* GENERATION */) {
+    return "llm";
+  }
+  if (normalized.toolCalls && normalized.toolCalls.length > 0) {
+    return "tool";
+  }
+  const name = (normalized.name || "").toLowerCase();
+  if (/retriev|search|rag/i.test(name)) return "retrieval";
+  if (/embed/i.test(name)) return "embedding";
+  if (/guard|safety/i.test(name)) return "guardrail";
+  return "function";
+}
+
 // src/normalizer/type-classifier.ts
 var TypeClassifier = class {
   classify(span, attributes) {
@@ -2182,6 +2261,7 @@ function normalizeSpan(resource, scope, span) {
   }
   const agentMarkAttributes = parseAgentMarkAttributes(allAttributes);
   Object.assign(normalized, agentMarkAttributes);
+  normalized.semanticKind = resolveSemanticKind(normalized, allAttributes);
   return normalized;
 }
 function normalizeOtlpSpans(resourceSpans) {
@@ -2201,6 +2281,7 @@ export {
   AgentMarkTransformer as ClaudeAgentTransformer,
   MastraTransformer,
   OtelGenAiTransformer,
+  SEMANTIC_KINDS,
   SpanType,
   TransformerRegistry,
   TypeClassifier,
@@ -2220,6 +2301,7 @@ export {
   parseMetadata,
   parseTokens,
   registry,
+  resolveSemanticKind,
   toFrontMatter,
   typeClassifier,
   verifySignature