@core-ai/axiom 0.11.1

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 Omnifact (https://omnifact.ai)
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/README.md

@@ -0,0 +1,57 @@
+# @core-ai/axiom
+
+[![npm](https://img.shields.io/npm/v/@core-ai/axiom.svg)](https://www.npmjs.com/package/@core-ai/axiom)
+
+Axiom GenAI telemetry middleware for `@core-ai/core-ai`.
+
+## Installation
+
+```bash
+npm install @core-ai/axiom @opentelemetry/api @opentelemetry/sdk-node @opentelemetry/exporter-trace-otlp-http
+```
+
+`@opentelemetry/api` is a peer dependency. Configure an OpenTelemetry SDK in your application to export spans to Axiom.
+
+## Usage
+
+```ts
+import { OTLPTraceExporter } from '@opentelemetry/exporter-trace-otlp-http';
+import { NodeSDK } from '@opentelemetry/sdk-node';
+import { generate, wrapChatModel } from '@core-ai/core-ai';
+import { createAxiomExporterOptions, createAxiomMiddleware } from '@core-ai/axiom';
+import { createOpenAI } from '@core-ai/openai';
+
+function getRequiredEnv(
+    name: 'AXIOM_DATASET' | 'AXIOM_TOKEN' | 'OPENAI_API_KEY'
+): string {
+    const value = process.env[name];
+    if (!value) {
+        throw new Error(`Missing required environment variable: ${name}`);
+    }
+    return value;
+}
+
+const sdk = new NodeSDK({
+    traceExporter: new OTLPTraceExporter(
+        createAxiomExporterOptions({
+            token: getRequiredEnv('AXIOM_TOKEN'),
+            dataset: getRequiredEnv('AXIOM_DATASET'),
+        })
+    ),
+});
+
+sdk.start();
+
+const openai = createOpenAI({ apiKey: getRequiredEnv('OPENAI_API_KEY') });
+const model = wrapChatModel({
+    model: openai.chatModel('gpt-5-mini'),
+    middleware: [createAxiomMiddleware()],
+});
+
+const result = await generate({
+    model,
+    messages: [{ role: 'user', content: 'Hello!' }],
+});
+```
+
+Axiom provisions a GenAI dashboard for datasets that receive spans with `gen_ai.*` attributes. The Axiom middleware is an Axiom-branded wrapper around `@core-ai/opentelemetry`, so it emits OpenTelemetry GenAI semantic convention attributes for chat, embeddings, and image generation.

package/dist/index.d.ts

@@ -0,0 +1,20 @@
+import { OtelMiddlewareOptions } from '@core-ai/opentelemetry';
+export { createOtelEmbeddingMiddleware as createAxiomEmbeddingMiddleware, createOtelImageMiddleware as createAxiomImageMiddleware, createOtelMiddleware as createAxiomMiddleware } from '@core-ai/opentelemetry';
+
+type AxiomMiddlewareOptions = OtelMiddlewareOptions;
+declare const AXIOM_OTLP_TRACES_ENDPOINT: "https://api.axiom.co/v1/traces";
+type AxiomExporterConfig = {
+    token: string;
+    dataset: string;
+    endpoint?: string;
+};
+type AxiomExporterOptions = {
+    url: string;
+    headers: {
+        Authorization: string;
+        'X-Axiom-Dataset': string;
+    };
+};
+declare function createAxiomExporterOptions(config: AxiomExporterConfig): AxiomExporterOptions;
+
+export { AXIOM_OTLP_TRACES_ENDPOINT, type AxiomExporterConfig, type AxiomExporterOptions, type AxiomMiddlewareOptions, createAxiomExporterOptions };

package/dist/index.js

@@ -0,0 +1,23 @@
+// src/index.ts
+import {
+  createOtelEmbeddingMiddleware,
+  createOtelImageMiddleware,
+  createOtelMiddleware
+} from "@core-ai/opentelemetry";
+var AXIOM_OTLP_TRACES_ENDPOINT = "https://api.axiom.co/v1/traces";
+function createAxiomExporterOptions(config) {
+  return {
+    url: config.endpoint ?? AXIOM_OTLP_TRACES_ENDPOINT,
+    headers: {
+      Authorization: `Bearer ${config.token}`,
+      "X-Axiom-Dataset": config.dataset
+    }
+  };
+}
+export {
+  AXIOM_OTLP_TRACES_ENDPOINT,
+  createOtelEmbeddingMiddleware as createAxiomEmbeddingMiddleware,
+  createAxiomExporterOptions,
+  createOtelImageMiddleware as createAxiomImageMiddleware,
+  createOtelMiddleware as createAxiomMiddleware
+};

package/package.json

@@ -0,0 +1,59 @@
+{
+    "name": "@core-ai/axiom",
+    "version": "0.11.1",
+    "description": "Axiom GenAI telemetry middleware for @core-ai/core-ai",
+    "license": "MIT",
+    "author": "Omnifact (https://omnifact.ai)",
+    "repository": {
+        "type": "git",
+        "url": "git+https://github.com/agdevhq/core-ai.git",
+        "directory": "packages/axiom"
+    },
+    "keywords": [
+        "llm",
+        "ai",
+        "axiom",
+        "opentelemetry",
+        "observability"
+    ],
+    "type": "module",
+    "main": "./dist/index.js",
+    "types": "./dist/index.d.ts",
+    "exports": {
+        ".": {
+            "types": "./dist/index.d.ts",
+            "import": "./dist/index.js",
+            "require": "./dist/index.js",
+            "default": "./dist/index.js"
+        }
+    },
+    "files": [
+        "dist",
+        "README.md",
+        "LICENSE"
+    ],
+    "publishConfig": {
+        "access": "public",
+        "provenance": true
+    },
+    "scripts": {
+        "build": "tsup",
+        "lint": "eslint src/ --max-warnings 0",
+        "check-types": "tsc --noEmit",
+        "test": "vitest run",
+        "test:watch": "vitest"
+    },
+    "dependencies": {
+        "@core-ai/opentelemetry": "^0.11.1"
+    },
+    "peerDependencies": {
+        "@opentelemetry/api": "^1.9.1"
+    },
+    "devDependencies": {
+        "@core-ai/eslint-config": "*",
+        "@core-ai/typescript-config": "*",
+        "@opentelemetry/api": "^1.9.1",
+        "typescript": "^5.7.3",
+        "vitest": "^3.2.4"
+    }
+}