@nyraxis/sdk 0.1.1

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Nyraxis
+
+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,61 @@
+# @nyraxis/sdk
+
+Universal AI agent observability for JavaScript/TypeScript.
+
+Automatically traces every LLM provider, framework, and vector DB.
+
+## Install
+
+```bash
+npm install @nyraxis/sdk
+```
+
+## Quick Start
+
+```typescript
+import { init, shutdown } from "@nyraxis/sdk";
+
+// Initialize once at app startup
+init({
+  apiKey: "nyx_...",
+  baseUrl: "https://your-nyraxis-backend.com",
+  agentName: "my-agent",
+});
+
+// All LLM calls are now auto-traced — no code changes needed.
+// Works with: OpenAI, Anthropic, Google, Cohere, Mistral, Bedrock, etc.
+
+// Before process exit:
+await shutdown();
+```
+
+## Trace Grouping
+
+```typescript
+import { withWorkflow, withTask } from "@nyraxis/sdk";
+
+// Group related LLM calls into a single trace
+const result = await withWorkflow({ name: "research-agent" }, async () => {
+  const summary = await withTask({ name: "summarize" }, async () => {
+    return await openai.chat.completions.create({ ... });
+  });
+  return summary;
+});
+```
+
+## Auto-Instrumented
+
+### LLM Providers
+OpenAI, Anthropic, Google (Gemini/PaLM/VertexAI), Cohere, Mistral, AWS Bedrock,
+Azure OpenAI, Ollama, Together AI, Groq, Replicate, HuggingFace, AI21, Aleph Alpha,
+DeepSeek, Fireworks, Perplexity, WatsonX, SageMaker
+
+### Frameworks
+LangChain.js, Vercel AI SDK
+
+### Vector Databases
+Pinecone, Chroma, Weaviate, Qdrant, Milvus, PGVector, Redis, LanceDB
+
+## License
+
+MIT

package/dist/index.d.ts

@@ -0,0 +1,43 @@
+/**
+ * @nyraxis/sdk — Universal AI agent observability for JavaScript/TypeScript.
+ *
+ * Auto-traces every LLM provider, framework, and vector DB.
+ *
+ * Usage:
+ *   import { init, shutdown } from "@nyraxis/sdk";
+ *   init({ apiKey: "nyx_...", agentName: "my-agent" });
+ *   // All LLM calls are now auto-traced.
+ *   await shutdown(); // before process exit
+ */
+/** Configuration for trace grouping decorators. */
+export interface DecoratorConfig {
+    name: string;
+    version?: number;
+    associationProperties?: Record<string, string>;
+    traceContent?: boolean;
+    inputParameters?: unknown[];
+    suppressTracing?: boolean;
+}
+export interface NyraxisConfig {
+    /** Your Nyraxis API key (nyx_...) */
+    apiKey: string;
+    /** Nyraxis backend URL (default: http://localhost:8000) */
+    baseUrl?: string;
+    /** Name for this agent in the dashboard */
+    agentName?: string;
+    /** Optional version string */
+    agentVersion?: string;
+    /** If true, send spans immediately (default: false) */
+    disableBatch?: boolean;
+}
+export declare function init(config: NyraxisConfig): void;
+/** Flush pending traces without full teardown. Safe to call multiple times. */
+export declare function flush(): Promise<void>;
+/** Flush and shut down. Call once before process exit. */
+export declare function shutdown(): Promise<void>;
+/** Check if the SDK has been initialized. */
+export declare function isInitialized(): boolean;
+export declare function withWorkflow<A extends unknown[], F extends (...args: A) => ReturnType<F>>(config: DecoratorConfig, fn: F, ...args: A): Promise<Awaited<ReturnType<F>>>;
+export declare function withTask<A extends unknown[], F extends (...args: A) => ReturnType<F>>(config: DecoratorConfig, fn: F, ...args: A): Promise<Awaited<ReturnType<F>>>;
+export declare function withAgent<A extends unknown[], F extends (...args: A) => ReturnType<F>>(config: DecoratorConfig, fn: F, ...args: A): Promise<Awaited<ReturnType<F>>>;
+export declare function withTool<A extends unknown[], F extends (...args: A) => ReturnType<F>>(config: DecoratorConfig, fn: F, ...args: A): Promise<Awaited<ReturnType<F>>>;

package/dist/index.js

@@ -0,0 +1,79 @@
+"use strict";
+/**
+ * @nyraxis/sdk — Universal AI agent observability for JavaScript/TypeScript.
+ *
+ * Auto-traces every LLM provider, framework, and vector DB.
+ *
+ * Usage:
+ *   import { init, shutdown } from "@nyraxis/sdk";
+ *   init({ apiKey: "nyx_...", agentName: "my-agent" });
+ *   // All LLM calls are now auto-traced.
+ *   await shutdown(); // before process exit
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.init = init;
+exports.flush = flush;
+exports.shutdown = shutdown;
+exports.isInitialized = isInitialized;
+exports.withWorkflow = withWorkflow;
+exports.withTask = withTask;
+exports.withAgent = withAgent;
+exports.withTool = withTool;
+const node_server_sdk_1 = require("@traceloop/node-server-sdk");
+let _initialized = false;
+function init(config) {
+    if (_initialized)
+        return;
+    if (!config.apiKey) {
+        throw new Error("Nyraxis: apiKey is required. Get one from your dashboard.");
+    }
+    const baseUrl = (config.baseUrl ?? "http://localhost:8000").replace(/\/+$/, "");
+    const agentName = config.agentName ?? "nyraxis-agent";
+    // Set the API key header via the environment variable the OTLP exporter reads.
+    // This avoids needing a custom exporter — the SDK's built-in OTLP exporter
+    // picks up OTEL_EXPORTER_OTLP_HEADERS automatically.
+    process.env.OTEL_EXPORTER_OTLP_HEADERS = `X-API-Key=${config.apiKey}`;
+    (0, node_server_sdk_1.initialize)({
+        appName: agentName,
+        baseUrl: `${baseUrl}/api/v1/ingest/otel`,
+        disableBatch: config.disableBatch ?? false,
+        traceloopSyncEnabled: false,
+        silenceInitializationMessage: true,
+    });
+    _initialized = true;
+}
+/** Flush pending traces without full teardown. Safe to call multiple times. */
+async function flush() {
+    if (!_initialized)
+        return;
+    try {
+        await (0, node_server_sdk_1.forceFlush)();
+    }
+    catch {
+        // best-effort — don't crash the app
+    }
+}
+/** Flush and shut down. Call once before process exit. */
+async function shutdown() {
+    if (!_initialized)
+        return;
+    await flush();
+    _initialized = false;
+}
+/** Check if the SDK has been initialized. */
+function isInitialized() {
+    return _initialized;
+}
+// Re-export decorators for trace grouping
+async function withWorkflow(config, fn, ...args) {
+    return (0, node_server_sdk_1.withWorkflow)(config, fn, ...args);
+}
+async function withTask(config, fn, ...args) {
+    return (0, node_server_sdk_1.withTask)(config, fn, ...args);
+}
+async function withAgent(config, fn, ...args) {
+    return (0, node_server_sdk_1.withAgent)(config, fn, ...args);
+}
+async function withTool(config, fn, ...args) {
+    return (0, node_server_sdk_1.withTool)(config, fn, ...args);
+}

package/package.json

@@ -0,0 +1,46 @@
+{
+  "name": "@nyraxis/sdk",
+  "version": "0.1.1",
+  "description": "Universal AI agent observability for JavaScript/TypeScript — trace every LLM provider and framework",
+  "main": "dist/index.js",
+  "types": "dist/index.d.ts",
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "require": "./dist/index.js"
+    }
+  },
+  "scripts": {
+    "build": "tsc",
+    "test": "node --test src/__tests__/index.test.mjs",
+    "prepublishOnly": "npm run build && npm test"
+  },
+  "keywords": [
+    "observability", "llm", "tracing", "opentelemetry", "ai", "agents",
+    "openai", "anthropic", "langchain", "nyraxis"
+  ],
+  "author": "Nyraxis <support@nyraxis.io>",
+  "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/nyraxis/nyraxis"
+  },
+  "homepage": "https://docs.nyraxis.com",
+  "bugs": {
+    "url": "https://github.com/nyraxis/nyraxis/issues"
+  },
+  "engines": {
+    "node": ">=18"
+  },
+  "dependencies": {
+    "@traceloop/node-server-sdk": "^0.11.0"
+  },
+  "devDependencies": {
+    "typescript": "^5.3.0"
+  },
+  "files": [
+    "dist/",
+    "README.md",
+    "LICENSE"
+  ]
+}