@agentuity/runtime 0.0.68 → 0.0.70

package/AGENTS.md

@@ -44,17 +44,59 @@ src/
 
 - **Agent context** - Every agent handler receives `AgentContext` as first parameter
 - **Schema validation** - Support StandardSchemaV1 (works with Zod, Valibot, etc.)
+- **Route validation** - Use `agent.validator()` for automatic input validation with full type safety
 - **Streaming support** - Agents can return ReadableStream for streaming responses
 - **WebSocket support** - Use `router.websocket()` for WebSocket routes
 - **SSE support** - Use `router.sse()` for Server-Sent Events
 - **Session tracking** - Each request gets unique sessionId
-- **Storage abstractions** - Provide kv, objectstore, stream, vector interfaces
+- **Storage abstractions** - Provide kv, stream, vector interfaces
+
+## Route Validation
+
+Routes can use `agent.validator()` to automatically validate request input using the agent's schema:
+
+```typescript
+import { createRouter } from '@agentuity/runtime';
+import myAgent from './my-agent';
+
+const router = createRouter();
+
+// Automatic validation using agent's input schema
+router.post('/', myAgent.validator(), async (c) => {
+	const data = c.req.valid('json'); // Fully typed from agent schema!
+	const output = await myAgent.run(data);
+	return c.json(output);
+});
+
+// Override with custom schema
+router.post(
+	'/custom',
+	agent.validator({
+		input: z.object({ custom: z.string() }),
+	}),
+	async (c) => {
+		const data = c.req.valid('json'); // Typed as { custom: string }
+		return c.json(data);
+	}
+);
+
+// GET routes don't need validation
+router.get('/', async (c) => {
+	return c.json({ hello: 'world' });
+});
+```
+
+The validator supports three overload signatures:
+
+- `agent.validator()` - Uses agent's input/output schemas
+- `agent.validator({ output: schema })` - Output-only validation (GET-compatible)
+- `agent.validator({ input: schema, output?: schema })` - Custom input/output schemas
 
 ## Agent Definition Pattern
 
 ```typescript
 import { createAgent } from '@agentuity/runtime';
-import { z } from 'zod';
+import { s } from '@agentuity/schema';
 
 export default createAgent({
 	metadata: {
@@ -66,10 +108,10 @@ export default createAgent({
 		version: 'hash-or-version',
 	},
 	schema: {
-		input: z.object({
+		input: s.object({
 			/* ... */
 		}),
-		output: z.object({
+		output: s.object({
 			/* ... */
 		}),
 	},
@@ -99,10 +141,8 @@ interface AgentContext {
 	tracer: Tracer; // OpenTelemetry tracer
 	sessionId: string; // Unique session ID
 	kv: KeyValueStorage; // Key-value storage
-	objectstore: ObjectStorage; // Object storage
 	stream: StreamStorage; // Stream storage
 	vector: VectorStorage; // Vector storage
-	agent: AgentRegistry; // Access other agents
 	waitUntil: (promise) => void; // Background tasks
 }
 ```
@@ -114,12 +154,92 @@ interface AgentContext {
 - **Metrics**: Access via `c.var.meter` in Hono context
 - **Environment**: Metrics/traces sent to OTLP endpoints
 
+## Type Safety
+
+**End-to-end type safety is a core feature of the runtime.** When you use `createAgent()` with schemas and `agent.validator()` in routes, TypeScript automatically infers correct types throughout your application.
+
+### What IS Type-Safe ✅
+
+1. **Route handler input types** - `c.req.valid('json')` is automatically typed from agent schema
+2. **Agent handler types** - Both `ctx` and `input` parameters are fully typed
+3. **Runtime validation** - Input/output validation happens automatically
+4. **Schema overrides** - Custom schemas in `agent.validator({ input, output })` maintain type safety
+5. **Multiple agents** - Each route maintains independent type safety
+
+### Type Inference Best Practices
+
+**CRITICAL:** Do NOT add type annotations to agent handler parameters - let TypeScript infer them:
+
+```typescript
+// ✅ CORRECT: Let TypeScript infer types from schema
+const agent = createAgent('user', {
+  schema: {
+    input: z.object({ name: z.string(), age: z.number() }),
+    output: z.object({ id: z.string() }),
+  },
+  handler: async (ctx, input) => {
+    // ctx is typed as AgentContext
+    // input is typed as { name: string, age: number }
+    return { id: `user-${input.name}` };
+  },
+});
+
+// ❌ WRONG: Explicit types defeat inference
+handler: async (ctx: AgentContext, input: any) => { ... }
+```
+
+### Hono Method Chaining
+
+For best type inference with Hono, use method chaining:
+
+```typescript
+// ✅ RECOMMENDED: Method chaining preserves types
+const app = new Hono()
+	.post('/users', userAgent.validator(), handler)
+	.get('/users/:id', userAgent.validator({ output: UserSchema }), handler);
+```
+
+### Testing Type Safety
+
+**IMPORTANT:** Due to Hono's `testClient()` type inference limitations with method-chained apps, use `app.request()` for testing:
+
+```typescript
+// ✅ CORRECT: Use app.request() for testing
+test('creates user', async () => {
+	const app = new Hono().post('/users', agent.validator(), async (c) => {
+		const data = c.req.valid('json'); // Fully typed!
+		return c.json({ id: `user-${data.name}` });
+	});
+
+	const res = await app.request('/users', {
+		method: 'POST',
+		headers: { 'Content-Type': 'application/json' },
+		body: JSON.stringify({ name: 'Alice', age: 30 }),
+	});
+
+	expect(res.status).toBe(200);
+	const result = await res.json();
+	expect(result.id).toBe('user-Alice');
+});
+
+// ❌ AVOID: testClient has type inference issues
+import { testClient } from 'hono/testing';
+const client = testClient(app); // Returns unknown type
+```
+
+See `test/agent-type-safety.test.ts` for comprehensive type-safe testing examples.
+
+For detailed type safety documentation, see `TYPE_SAFETY.md`.
+
 ## Testing
 
-- Use Bun's test runner: `bun test`
-- Mock storage interfaces (kv, objectstore, etc.)
-- Test agent handlers with mock context
-- Use Hono's testing utilities for routes
+- **Test runner**: Use Bun's test runner: `bun test`
+- **Test structure**: All tests in `test/` folder parallel to `src/`
+- **Test imports**: Import from `../src/` not `../`
+- **Mock contexts**: Use `TestAgentContext` from `test/helpers/test-context.ts`
+- **Mock services**: Use mock storage interfaces (kv, stream, vector, etc.)
+- **Route testing**: Use `app.request()` for testing routes (NOT `testClient()`)
+- **Type verification**: Let TypeScript infer agent handler types - do NOT add type annotations
 
 ## Publishing Checklist
 

package/README.md

@@ -30,15 +30,15 @@ logger.info('Server running on http://localhost:3500');
 
 ```typescript
 import { createAgent } from '@agentuity/runtime';
-import { z } from 'zod';
+import { s } from '@agentuity/schema';
 
 const agent = createAgent({
 	schema: {
-		input: z.object({
-			message: z.string(),
+		input: s.object({
+			message: s.string(),
 		}),
-		output: z.object({
-			response: z.string(),
+		output: s.object({
+			response: s.string(),
 		}),
 	},
 	handler: async (ctx, input) => {
@@ -136,7 +136,7 @@ Creates a type-safe agent with input/output validation.
 
 - `schema.input?` - Schema for input validation (Zod, Valibot, etc.)
 - `schema.output?` - Schema for output validation
-- `handler` - Agent handler function `(ctx: AgentContext, input) => output`
+- `handler` - Agent handler function `(ctx, input) => output`
 
 ### createRouter()
 
@@ -159,7 +159,6 @@ interface AgentContext {
 	tracer: Tracer; // OpenTelemetry tracer
 	sessionId: string; // Unique session ID
 	kv: KeyValueStorage; // Key-value storage
-	objectstore: ObjectStorage; // Object storage
 	stream: StreamStorage; // Stream storage
 	vector: VectorStorage; // Vector storage
 	agent: AgentRegistry; // Access to other agents
@@ -172,7 +171,6 @@ interface AgentContext {
 Agentuity provides built-in storage abstractions:
 
 - **KeyValueStorage** - Simple key-value storage
-- **ObjectStorage** - Object/blob storage
 - **StreamStorage** - Streaming data storage
 - **VectorStorage** - Vector embeddings storage
 

package/dist/_config.d.ts

@@ -65,4 +65,20 @@ export declare function getEnvironment(): string;
  * @returns boolean
  */
 export declare function isAuthenticated(): boolean;
+/**
+ * Symbol for accessing internal runtime state.
+ * Defined here to avoid circular dependency.
+ */
+export declare const AGENT_RUNTIME: unique symbol;
+/**
+ * Symbol for accessing internal agent from AgentRunner.
+ * @internal
+ */
+export declare const INTERNAL_AGENT: unique symbol;
+/**
+ * Symbol for tracking the current executing agent (for telemetry).
+ * Not exposed on public AgentContext interface.
+ * @internal
+ */
+export declare const CURRENT_AGENT: unique symbol;
 //# sourceMappingURL=_config.d.ts.map

package/dist/_config.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"_config.d.ts","sourceRoot":"","sources":["../src/_config.ts"],"names":[],"mappings":"AAMA,wBAAgB,IAAI,SAWnB;AAED;;;;GAIG;AACH,wBAAgB,aAAa,IAAI,MAAM,CAEtC;AAED;;;;GAIG;AACH,wBAAgB,UAAU,IAAI,MAAM,CAEnC;AAED;;;;GAIG;AACH,wBAAgB,aAAa,IAAI,MAAM,CAEtC;AAED;;;;GAIG;AACH,wBAAgB,iBAAiB,IAAI,MAAM,GAAG,SAAS,CAEtD;AAED;;;;GAIG;AACH,wBAAgB,YAAY,IAAI,MAAM,GAAG,SAAS,CAEjD;AAED;;;;GAIG;AACH,wBAAgB,eAAe,IAAI,MAAM,GAAG,SAAS,CAEpD;AAED;;;;GAIG;AACH,wBAAgB,SAAS,IAAI,OAAO,CAEnC;AAED;;;;GAIG;AACH,wBAAgB,YAAY,IAAI,OAAO,CAEtC;AAED;;;;GAIG;AACH,wBAAgB,aAAa,IAAI,MAAM,CAEtC;AAED;;;;GAIG;AACH,wBAAgB,cAAc,IAAI,MAAM,CAEvC;AAED;;;;GAIG;AACH,wBAAgB,eAAe,IAAI,OAAO,CAEzC"}
+{"version":3,"file":"_config.d.ts","sourceRoot":"","sources":["../src/_config.ts"],"names":[],"mappings":"AAMA,wBAAgB,IAAI,SAWnB;AAED;;;;GAIG;AACH,wBAAgB,aAAa,IAAI,MAAM,CAEtC;AAED;;;;GAIG;AACH,wBAAgB,UAAU,IAAI,MAAM,CAEnC;AAED;;;;GAIG;AACH,wBAAgB,aAAa,IAAI,MAAM,CAEtC;AAED;;;;GAIG;AACH,wBAAgB,iBAAiB,IAAI,MAAM,GAAG,SAAS,CAEtD;AAED;;;;GAIG;AACH,wBAAgB,YAAY,IAAI,MAAM,GAAG,SAAS,CAEjD;AAED;;;;GAIG;AACH,wBAAgB,eAAe,IAAI,MAAM,GAAG,SAAS,CAEpD;AAED;;;;GAIG;AACH,wBAAgB,SAAS,IAAI,OAAO,CAEnC;AAED;;;;GAIG;AACH,wBAAgB,YAAY,IAAI,OAAO,CAEtC;AAED;;;;GAIG;AACH,wBAAgB,aAAa,IAAI,MAAM,CAEtC;AAED;;;;GAIG;AACH,wBAAgB,cAAc,IAAI,MAAM,CAEvC;AAED;;;;GAIG;AACH,wBAAgB,eAAe,IAAI,OAAO,CAEzC;AAED;;;GAGG;AACH,eAAO,MAAM,aAAa,eAA0B,CAAC;AAErD;;;GAGG;AACH,eAAO,MAAM,cAAc,eAA2B,CAAC;AAEvD;;;;GAIG;AACH,eAAO,MAAM,aAAa,eAA0B,CAAC"}

package/dist/_config.js

@@ -103,4 +103,20 @@ export function getEnvironment() {
 export function isAuthenticated() {
     return !!process.env.AGENTUITY_SDK_KEY;
 }
+/**
+ * Symbol for accessing internal runtime state.
+ * Defined here to avoid circular dependency.
+ */
+export const AGENT_RUNTIME = Symbol('AGENT_RUNTIME');
+/**
+ * Symbol for accessing internal agent from AgentRunner.
+ * @internal
+ */
+export const INTERNAL_AGENT = Symbol('INTERNAL_AGENT');
+/**
+ * Symbol for tracking the current executing agent (for telemetry).
+ * Not exposed on public AgentContext interface.
+ * @internal
+ */
+export const CURRENT_AGENT = Symbol('CURRENT_AGENT');
 //# sourceMappingURL=_config.js.map

package/dist/_config.js.map

@@ -1 +1 @@
-{"version":3,"file":"_config.js","sourceRoot":"","sources":["../src/_config.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,UAAU,EAAE,MAAM,SAAS,CAAC;AACnD,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AAEjC,IAAI,OAA2B,CAAC;AAChC,IAAI,UAA8B,CAAC;AAEnC,MAAM,UAAU,IAAI;IACnB,MAAM,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,EAAE,kBAAkB,CAAC,CAAC;IACpD,IAAI,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC;QACnB,IAAI,CAAC;YACJ,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC,CAAC;YACjD,OAAO,GAAG,GAAG,CAAC,IAAI,CAAC;YACnB,UAAU,GAAG,GAAG,CAAC,OAAO,CAAC;QAC1B,CAAC;QAAC,MAAM,CAAC;YACR,wCAAwC;QACzC,CAAC;IACF,CAAC;AACF,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,aAAa;IAC5B,OAAO,OAAO,CAAC,GAAG,CAAC,2BAA2B,IAAI,SAAS,CAAC;AAC7D,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,UAAU;IACzB,OAAO,OAAO,IAAI,SAAS,CAAC;AAC7B,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,aAAa;IAC5B,OAAO,UAAU,IAAI,SAAS,CAAC;AAChC,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,iBAAiB;IAChC,OAAO,OAAO,CAAC,GAAG,CAAC,sBAAsB,CAAC;AAC3C,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,YAAY;IAC3B,OAAO,OAAO,CAAC,GAAG,CAAC,0BAA0B,CAAC;AAC/C,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,eAAe;IAC9B,OAAO,OAAO,CAAC,GAAG,CAAC,6BAA6B,CAAC;AAClD,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,SAAS;IACxB,OAAO,OAAO,CAAC,GAAG,CAAC,sBAAsB,KAAK,MAAM,CAAC;AACtD,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,YAAY;IAC3B,OAAO,OAAO,CAAC,GAAG,CAAC,QAAQ,KAAK,YAAY,IAAI,CAAC,SAAS,EAAE,CAAC;AAC9D,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,aAAa;IAC5B,OAAO,OAAO,CAAC,GAAG,CAAC,qBAAqB,IAAI,SAAS,CAAC;AACvD,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,cAAc;IAC7B,OAAO,OAAO,CAAC,GAAG,CAAC,qBAAqB,IAAI,OAAO,CAAC,GAAG,CAAC,QAAQ,IAAI,aAAa,CAAC;AACnF,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,eAAe;IAC9B,OAAO,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC;AACxC,CAAC"}
+{"version":3,"file":"_config.js","sourceRoot":"","sources":["../src/_config.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,UAAU,EAAE,MAAM,SAAS,CAAC;AACnD,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AAEjC,IAAI,OAA2B,CAAC;AAChC,IAAI,UAA8B,CAAC;AAEnC,MAAM,UAAU,IAAI;IACnB,MAAM,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,EAAE,kBAAkB,CAAC,CAAC;IACpD,IAAI,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC;QACnB,IAAI,CAAC;YACJ,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC,CAAC;YACjD,OAAO,GAAG,GAAG,CAAC,IAAI,CAAC;YACnB,UAAU,GAAG,GAAG,CAAC,OAAO,CAAC;QAC1B,CAAC;QAAC,MAAM,CAAC;YACR,wCAAwC;QACzC,CAAC;IACF,CAAC;AACF,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,aAAa;IAC5B,OAAO,OAAO,CAAC,GAAG,CAAC,2BAA2B,IAAI,SAAS,CAAC;AAC7D,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,UAAU;IACzB,OAAO,OAAO,IAAI,SAAS,CAAC;AAC7B,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,aAAa;IAC5B,OAAO,UAAU,IAAI,SAAS,CAAC;AAChC,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,iBAAiB;IAChC,OAAO,OAAO,CAAC,GAAG,CAAC,sBAAsB,CAAC;AAC3C,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,YAAY;IAC3B,OAAO,OAAO,CAAC,GAAG,CAAC,0BAA0B,CAAC;AAC/C,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,eAAe;IAC9B,OAAO,OAAO,CAAC,GAAG,CAAC,6BAA6B,CAAC;AAClD,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,SAAS;IACxB,OAAO,OAAO,CAAC,GAAG,CAAC,sBAAsB,KAAK,MAAM,CAAC;AACtD,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,YAAY;IAC3B,OAAO,OAAO,CAAC,GAAG,CAAC,QAAQ,KAAK,YAAY,IAAI,CAAC,SAAS,EAAE,CAAC;AAC9D,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,aAAa;IAC5B,OAAO,OAAO,CAAC,GAAG,CAAC,qBAAqB,IAAI,SAAS,CAAC;AACvD,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,cAAc;IAC7B,OAAO,OAAO,CAAC,GAAG,CAAC,qBAAqB,IAAI,OAAO,CAAC,GAAG,CAAC,QAAQ,IAAI,aAAa,CAAC;AACnF,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,eAAe;IAC9B,OAAO,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC;AACxC,CAAC;AAED;;;GAGG;AACH,MAAM,CAAC,MAAM,aAAa,GAAG,MAAM,CAAC,eAAe,CAAC,CAAC;AAErD;;;GAGG;AACH,MAAM,CAAC,MAAM,cAAc,GAAG,MAAM,CAAC,gBAAgB,CAAC,CAAC;AAEvD;;;;GAIG;AACH,MAAM,CAAC,MAAM,aAAa,GAAG,MAAM,CAAC,eAAe,CAAC,CAAC"}

package/dist/_context.d.ts

@@ -1,17 +1,15 @@
 import { AsyncLocalStorage } from 'node:async_hooks';
 import type { Context as HonoContext } from 'hono';
 import type { Tracer } from '@opentelemetry/api';
-import { type KeyValueStorage, type ObjectStorage, type StreamStorage, type VectorStorage } from '@agentuity/core';
-import type { AgentContext, AgentName, AgentRegistry, AgentRunner } from './agent';
+import { type KeyValueStorage, type StreamStorage, type VectorStorage } from '@agentuity/core';
+import type { AgentContext, AgentRegistry, AgentRunner, AgentRuntimeState } from './agent';
+import { AGENT_RUNTIME } from './_config';
 import type { Logger } from './logger';
 import type WaitUntilHandler from './_waituntil';
 import type { Thread, Session } from './session';
-export interface RequestAgentContextArgs<TAgentMap extends AgentRegistry = AgentRegistry, TCurrent extends AgentRunner<any, any, any> | undefined = AgentRunner<any, any, any> | undefined, TParent extends AgentRunner<any, any, any> | undefined = AgentRunner<any, any, any> | undefined, TConfig = unknown, TAppState = Record<string, never>> {
+export interface RequestAgentContextArgs<TAgentMap extends AgentRegistry = AgentRegistry, TConfig = unknown, TAppState = Record<string, never>> {
     sessionId: string;
     agent: TAgentMap;
-    current: TCurrent;
-    parent: TParent;
-    agentName: AgentName;
     logger: Logger;
     tracer: Tracer;
     session: Session;
@@ -19,17 +17,14 @@ export interface RequestAgentContextArgs<TAgentMap extends AgentRegistry = Agent
     handler: WaitUntilHandler;
     config: TConfig;
     app: TAppState;
+    runtime: AgentRuntimeState;
 }
-export declare class RequestAgentContext<TAgentMap extends AgentRegistry = AgentRegistry, TCurrent extends AgentRunner<any, any, any> | undefined = AgentRunner<any, any, any> | undefined, TParent extends AgentRunner<any, any, any> | undefined = AgentRunner<any, any, any> | undefined, TConfig = unknown, TAppState = Record<string, never>> implements AgentContext<TAgentMap, TCurrent, TParent, TConfig, TAppState> {
+export declare class RequestAgentContext<TAgentMap extends AgentRegistry = AgentRegistry, TConfig = unknown, TAppState = Record<string, never>> implements AgentContext<TAgentMap, TConfig, TAppState> {
     agent: TAgentMap;
-    current: TCurrent;
-    parent: TParent;
-    agentName: AgentName;
     logger: Logger;
     sessionId: string;
     tracer: Tracer;
     kv: KeyValueStorage;
-    objectstore: ObjectStorage;
     stream: StreamStorage;
     vector: VectorStorage;
     state: Map<string, unknown>;
@@ -37,16 +32,23 @@ export declare class RequestAgentContext<TAgentMap extends AgentRegistry = Agent
     thread: Thread;
     config: TConfig;
     app: TAppState;
+    [AGENT_RUNTIME]: AgentRuntimeState;
     private handler;
-    constructor(args: RequestAgentContextArgs<TAgentMap, TCurrent, TParent, TConfig, TAppState>);
+    constructor(args: RequestAgentContextArgs<TAgentMap, TConfig, TAppState>);
     waitUntil(callback: Promise<void> | (() => void | Promise<void>)): void;
 }
 export declare const inAgentContext: () => boolean;
 export declare const inHTTPContext: () => boolean;
-export declare const getAgentContext: () => AgentContext<any, any, any, any, any>;
+export declare const getAgentContext: () => AgentContext<any, any, any>;
 export declare const getHTTPContext: () => HonoContext;
-export declare const getAgentAsyncLocalStorage: () => AsyncLocalStorage<AgentContext<any, any, any, any, any>>;
+export declare const getAgentAsyncLocalStorage: () => AsyncLocalStorage<AgentContext<any, any, any>>;
 export declare const getHTTPAsyncLocalStorage: () => AsyncLocalStorage<HonoContext<any, any, {}>>;
-export declare const runInAgentContext: <TAgentMap extends AgentRegistry = AgentRegistry, TCurrent extends AgentRunner<any, any, any> | undefined = AgentRunner<any, any, any> | undefined, TParent extends AgentRunner<any, any, any> | undefined = AgentRunner<any, any, any> | undefined, TConfig = unknown, TAppState = Record<string, never>>(ctxObject: Record<string, unknown>, args: RequestAgentContextArgs<TAgentMap, TCurrent, TParent, TConfig, TAppState>, next: () => Promise<void>) => Promise<void>;
+/**
+ * Get the current executing agent's metadata (for internal telemetry use only).
+ * Returns undefined if not in an agent context or no agent is executing.
+ * @internal
+ */
+export declare const getCurrentAgentMetadata: () => AgentRunner["metadata"] | undefined;
+export declare const setupRequestAgentContext: <TAgentMap extends AgentRegistry = AgentRegistry, TConfig = unknown, TAppState = Record<string, never>>(ctxObject: Record<string, unknown>, args: RequestAgentContextArgs<TAgentMap, TConfig, TAppState>, next: () => Promise<void>) => Promise<void>;
 export declare const runInHTTPContext: <HonoContext>(ctx: HonoContext, next: () => Promise<void>) => Promise<void>;
 //# sourceMappingURL=_context.d.ts.map

package/dist/_context.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"_context.d.ts","sourceRoot":"","sources":["../src/_context.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AACrD,OAAO,KAAK,EAAE,OAAO,IAAI,WAAW,EAAE,MAAM,MAAM,CAAC;AACnD,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AACjD,OAAO,EAEN,KAAK,eAAe,EACpB,KAAK,aAAa,EAClB,KAAK,aAAa,EAClB,KAAK,aAAa,EAClB,MAAM,iBAAiB,CAAC;AACzB,OAAO,KAAK,EAAE,YAAY,EAAE,SAAS,EAAE,aAAa,EAAE,WAAW,EAAE,MAAM,SAAS,CAAC;AACnF,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,UAAU,CAAC;AACvC,OAAO,KAAK,gBAAgB,MAAM,cAAc,CAAC;AAEjD,OAAO,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAEjD,MAAM,WAAW,uBAAuB,CACvC,SAAS,SAAS,aAAa,GAAG,aAAa,EAE/C,QAAQ,SAAS,WAAW,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,GAAG,SAAS,GAAG,WAAW,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,GAAG,SAAS,EAEhG,OAAO,SAAS,WAAW,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,GAAG,SAAS,GAAG,WAAW,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,GAAG,SAAS,EAC/F,OAAO,GAAG,OAAO,EACjB,SAAS,GAAG,MAAM,CAAC,MAAM,EAAE,KAAK,CAAC;IAEjC,SAAS,EAAE,MAAM,CAAC;IAClB,KAAK,EAAE,SAAS,CAAC;IACjB,OAAO,EAAE,QAAQ,CAAC;IAClB,MAAM,EAAE,OAAO,CAAC;IAChB,SAAS,EAAE,SAAS,CAAC;IACrB,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,EAAE,OAAO,CAAC;IACjB,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,EAAE,gBAAgB,CAAC;IAC1B,MAAM,EAAE,OAAO,CAAC;IAChB,GAAG,EAAE,SAAS,CAAC;CACf;AAED,qBAAa,mBAAmB,CAC/B,SAAS,SAAS,aAAa,GAAG,aAAa,EAE/C,QAAQ,SAAS,WAAW,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,GAAG,SAAS,GAAG,WAAW,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,GAAG,SAAS,EAEhG,OAAO,SAAS,WAAW,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,GAAG,SAAS,GAAG,WAAW,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,GAAG,SAAS,EAC/F,OAAO,GAAG,OAAO,EACjB,SAAS,GAAG,MAAM,CAAC,MAAM,EAAE,KAAK,CAAC,CAChC,YAAW,YAAY,CAAC,SAAS,EAAE,QAAQ,EAAE,OAAO,EAAE,OAAO,EAAE,SAAS,CAAC;IAE1E,KAAK,EAAE,SAAS,CAAC;IACjB,OAAO,EAAE,QAAQ,CAAC;IAClB,MAAM,EAAE,OAAO,CAAC;IAChB,SAAS,EAAE,SAAS,CAAC;IACrB,MAAM,EAAE,MAAM,CAAC;IACf,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,EAAE,MAAM,CAAC;IACf,EAAE,EAAG,eAAe,CAAC;IACrB,WAAW,EAAG,aAAa,CAAC;IAC5B,MAAM,EAAG,aAAa,CAAC;IACvB,MAAM,EAAG,aAAa,CAAC;IACvB,KAAK,EAAE,GAAG,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAC5B,OAAO,EAAE,OAAO,CAAC;IACjB,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,EAAE,OAAO,CAAC;IAChB,GAAG,EAAE,SAAS,CAAC;IACf,OAAO,CAAC,OAAO,CAAmB;gBAEtB,IAAI,EAAE,uBAAuB,CAAC,SAAS,EAAE,QAAQ,EAAE,OAAO,EAAE,OAAO,EAAE,SAAS,CAAC;IAiB3F,SAAS,CAAC,QAAQ,EAAE,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,GAAG,IAAI;CAGvE;AAMD,eAAO,MAAM,cAAc,QAAO,OAGjC,CAAC;AAEF,eAAO,MAAM,aAAa,QAAO,OAGhC,CAAC;AAQF,eAAO,MAAM,eAAe,QAAO,YAAY,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,CAMtE,CAAC;AAOF,eAAO,MAAM,cAAc,QAAO,WAMjC,CAAC;AAEF,eAAO,MAAM,yBAAyB,gEAA+B,CAAC;AACtE,eAAO,MAAM,wBAAwB,oDAA8B,CAAC;AAEpE,eAAO,MAAM,iBAAiB,GAC7B,SAAS,SAAS,aAAa,GAAG,aAAa,EAE/C,QAAQ,SAAS,WAAW,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,GAAG,SAAS,GAAG,WAAW,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,GAAG,SAAS,EAEhG,OAAO,SAAS,WAAW,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,GAAG,SAAS,GAAG,WAAW,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,GAAG,SAAS,EAC/F,OAAO,GAAG,OAAO,EACjB,SAAS,GAAG,MAAM,CAAC,MAAM,EAAE,KAAK,CAAC,EAEjC,WAAW,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAClC,MAAM,uBAAuB,CAAC,SAAS,EAAE,QAAQ,EAAE,OAAO,EAAE,OAAO,EAAE,SAAS,CAAC,EAC/E,MAAM,MAAM,OAAO,CAAC,IAAI,CAAC,kBA4BzB,CAAC;AAEF,eAAO,MAAM,gBAAgB,GAAU,WAAW,EACjD,KAAK,WAAW,EAChB,MAAM,MAAM,OAAO,CAAC,IAAI,CAAC,kBAIzB,CAAC"}
+{"version":3,"file":"_context.d.ts","sourceRoot":"","sources":["../src/_context.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AACrD,OAAO,KAAK,EAAE,OAAO,IAAI,WAAW,EAAE,MAAM,MAAM,CAAC;AACnD,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AACjD,OAAO,EAEN,KAAK,eAAe,EACpB,KAAK,aAAa,EAClB,KAAK,aAAa,EAClB,MAAM,iBAAiB,CAAC;AACzB,OAAO,KAAK,EAAE,YAAY,EAAE,aAAa,EAAE,WAAW,EAAE,iBAAiB,EAAE,MAAM,SAAS,CAAC;AAC3F,OAAO,EAAE,aAAa,EAAiB,MAAM,WAAW,CAAC;AACzD,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,UAAU,CAAC;AACvC,OAAO,KAAK,gBAAgB,MAAM,cAAc,CAAC;AAEjD,OAAO,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAEjD,MAAM,WAAW,uBAAuB,CACvC,SAAS,SAAS,aAAa,GAAG,aAAa,EAC/C,OAAO,GAAG,OAAO,EACjB,SAAS,GAAG,MAAM,CAAC,MAAM,EAAE,KAAK,CAAC;IAEjC,SAAS,EAAE,MAAM,CAAC;IAClB,KAAK,EAAE,SAAS,CAAC;IACjB,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,EAAE,OAAO,CAAC;IACjB,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,EAAE,gBAAgB,CAAC;IAC1B,MAAM,EAAE,OAAO,CAAC;IAChB,GAAG,EAAE,SAAS,CAAC;IACf,OAAO,EAAE,iBAAiB,CAAC;CAC3B;AAED,qBAAa,mBAAmB,CAC/B,SAAS,SAAS,aAAa,GAAG,aAAa,EAC/C,OAAO,GAAG,OAAO,EACjB,SAAS,GAAG,MAAM,CAAC,MAAM,EAAE,KAAK,CAAC,CAChC,YAAW,YAAY,CAAC,SAAS,EAAE,OAAO,EAAE,SAAS,CAAC;IAEvD,KAAK,EAAE,SAAS,CAAC;IACjB,MAAM,EAAE,MAAM,CAAC;IACf,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,EAAE,MAAM,CAAC;IACf,EAAE,EAAG,eAAe,CAAC;IACrB,MAAM,EAAG,aAAa,CAAC;IACvB,MAAM,EAAG,aAAa,CAAC;IACvB,KAAK,EAAE,GAAG,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAC5B,OAAO,EAAE,OAAO,CAAC;IACjB,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,EAAE,OAAO,CAAC;IAChB,GAAG,EAAE,SAAS,CAAC;IACf,CAAC,aAAa,CAAC,EAAE,iBAAiB,CAAC;IACnC,OAAO,CAAC,OAAO,CAAmB;gBAEtB,IAAI,EAAE,uBAAuB,CAAC,SAAS,EAAE,OAAO,EAAE,SAAS,CAAC;IAexE,SAAS,CAAC,QAAQ,EAAE,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,GAAG,IAAI;CAGvE;AAMD,eAAO,MAAM,cAAc,QAAO,OAGjC,CAAC;AAEF,eAAO,MAAM,aAAa,QAAO,OAGhC,CAAC;AAQF,eAAO,MAAM,eAAe,QAAO,YAAY,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAM5D,CAAC;AAOF,eAAO,MAAM,cAAc,QAAO,WAMjC,CAAC;AAEF,eAAO,MAAM,yBAAyB,sDAA+B,CAAC;AACtE,eAAO,MAAM,wBAAwB,oDAA8B,CAAC;AAEpE;;;;GAIG;AACH,eAAO,MAAM,uBAAuB,QAAO,WAAW,CAAC,UAAU,CAAC,GAAG,SAMpE,CAAC;AAEF,eAAO,MAAM,wBAAwB,GACpC,SAAS,SAAS,aAAa,GAAG,aAAa,EAC/C,OAAO,GAAG,OAAO,EACjB,SAAS,GAAG,MAAM,CAAC,MAAM,EAAE,KAAK,CAAC,EAEjC,WAAW,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAClC,MAAM,uBAAuB,CAAC,SAAS,EAAE,OAAO,EAAE,SAAS,CAAC,EAC5D,MAAM,MAAM,OAAO,CAAC,IAAI,CAAC,kBA4BzB,CAAC;AAEF,eAAO,MAAM,gBAAgB,GAAU,WAAW,EACjD,KAAK,WAAW,EAChB,MAAM,MAAM,OAAO,CAAC,IAAI,CAAC,kBAIzB,CAAC"}

package/dist/_context.js

@@ -1,16 +1,13 @@
 import { AsyncLocalStorage } from 'node:async_hooks';
 import { StructuredError, } from '@agentuity/core';
+import { AGENT_RUNTIME, CURRENT_AGENT } from './_config';
 import { registerServices } from './_services';
 export class RequestAgentContext {
     agent;
-    current;
-    parent;
-    agentName;
     logger;
     sessionId;
     tracer;
     kv;
-    objectstore;
     stream;
     vector;
     state;
@@ -18,12 +15,10 @@ export class RequestAgentContext {
     thread;
     config;
     app;
+    [AGENT_RUNTIME];
     handler;
     constructor(args) {
         this.agent = args.agent;
-        this.current = args.current;
-        this.parent = args.parent;
-        this.agentName = args.agentName;
         this.logger = args.logger;
         this.sessionId = args.sessionId;
         this.tracer = args.tracer;
@@ -31,6 +26,7 @@ export class RequestAgentContext {
         this.session = args.session;
         this.config = args.config;
         this.app = args.app;
+        this[AGENT_RUNTIME] = args.runtime;
         this.state = new Map();
         this.handler = args.handler;
         registerServices(this, false); // agents already populated via args.agent
@@ -69,7 +65,20 @@ export const getHTTPContext = () => {
 };
 export const getAgentAsyncLocalStorage = () => agentAsyncLocalStorage;
 export const getHTTPAsyncLocalStorage = () => httpAsyncLocalStorage;
-export const runInAgentContext = (ctxObject, args, next) => {
+/**
+ * Get the current executing agent's metadata (for internal telemetry use only).
+ * Returns undefined if not in an agent context or no agent is executing.
+ * @internal
+ */
+export const getCurrentAgentMetadata = () => {
+    const context = agentAsyncLocalStorage.getStore();
+    if (!context)
+        return undefined;
+    // Access internal symbol property
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    return context[CURRENT_AGENT]?.metadata;
+};
+export const setupRequestAgentContext = (ctxObject, args, next) => {
     const ctx = new RequestAgentContext(args);
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
     const _ctx = ctx;

package/dist/_context.js.map

@@ -1 +1 @@
-{"version":3,"file":"_context.js","sourceRoot":"","sources":["../src/_context.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AAGrD,OAAO,EACN,eAAe,GAKf,MAAM,iBAAiB,CAAC;AAIzB,OAAO,EAAE,gBAAgB,EAAE,MAAM,aAAa,CAAC;AA0B/C,MAAM,OAAO,mBAAmB;IAU/B,KAAK,CAAY;IACjB,OAAO,CAAW;IAClB,MAAM,CAAU;IAChB,SAAS,CAAY;IACrB,MAAM,CAAS;IACf,SAAS,CAAS;IAClB,MAAM,CAAS;IACf,EAAE,CAAmB;IACrB,WAAW,CAAiB;IAC5B,MAAM,CAAiB;IACvB,MAAM,CAAiB;IACvB,KAAK,CAAuB;IAC5B,OAAO,CAAU;IACjB,MAAM,CAAS;IACf,MAAM,CAAU;IAChB,GAAG,CAAY;IACP,OAAO,CAAmB;IAElC,YAAY,IAA+E;QAC1F,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC;QACxB,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC;QAC5B,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;QAC1B,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC;QAChC,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;QAC1B,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC;QAChC,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;QAC1B,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;QAC1B,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC;QAC5B,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;QAC1B,IAAI,CAAC,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC;QACpB,IAAI,CAAC,KAAK,GAAG,IAAI,GAAG,EAAmB,CAAC;QACxC,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC;QAC5B,gBAAgB,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC,0CAA0C;IAC1E,CAAC;IAED,SAAS,CAAC,QAAsD;QAC/D,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;IAClC,CAAC;CACD;AAED,8DAA8D;AAC9D,MAAM,sBAAsB,GAAG,IAAI,iBAAiB,EAAyC,CAAC;AAC9F,MAAM,qBAAqB,GAAG,IAAI,iBAAiB,EAAe,CAAC;AAEnE,MAAM,CAAC,MAAM,cAAc,GAAG,GAAY,EAAE;IAC3C,MAAM,OAAO,GAAG,sBAAsB,CAAC,QAAQ,EAAE,CAAC;IAClD,OAAO,CAAC,CAAC,OAAO,CAAC;AAClB,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,aAAa,GAAG,GAAY,EAAE;IAC1C,MAAM,OAAO,GAAG,qBAAqB,CAAC,QAAQ,EAAE,CAAC;IACjD,OAAO,CAAC,CAAC,OAAO,CAAC;AAClB,CAAC,CAAC;AAEF,MAAM,6BAA6B,GAAG,eAAe,CACpD,+BAA+B,EAC/B,+BAA+B,CAC/B,CAAC;AAEF,8DAA8D;AAC9D,MAAM,CAAC,MAAM,eAAe,GAAG,GAA0C,EAAE;IAC1E,MAAM,OAAO,GAAG,sBAAsB,CAAC,QAAQ,EAAE,CAAC;IAClD,IAAI,CAAC,OAAO,EAAE,CAAC;QACd,MAAM,IAAI,6BAA6B,EAAE,CAAC;IAC3C,CAAC;IACD,OAAO,OAAO,CAAC;AAChB,CAAC,CAAC;AAEF,MAAM,4BAA4B,GAAG,eAAe,CACnD,8BAA8B,EAC9B,8BAA8B,CAC9B,CAAC;AAEF,MAAM,CAAC,MAAM,cAAc,GAAG,GAAgB,EAAE;IAC/C,MAAM,OAAO,GAAG,qBAAqB,CAAC,QAAQ,EAAE,CAAC;IACjD,IAAI,CAAC,OAAO,EAAE,CAAC;QACd,MAAM,IAAI,4BAA4B,EAAE,CAAC;IAC1C,CAAC;IACD,OAAO,OAAO,CAAC;AAChB,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,yBAAyB,GAAG,GAAG,EAAE,CAAC,sBAAsB,CAAC;AACtE,MAAM,CAAC,MAAM,wBAAwB,GAAG,GAAG,EAAE,CAAC,qBAAqB,CAAC;AAEpE,MAAM,CAAC,MAAM,iBAAiB,GAAG,CAShC,SAAkC,EAClC,IAA+E,EAC/E,IAAyB,EACxB,EAAE;IACH,MAAM,GAAG,GAAG,IAAI,mBAAmB,CAAmD,IAAI,CAAC,CAAC;IAC5F,8DAA8D;IAC9D,MAAM,IAAI,GAAG,GAAU,CAAC;IACxB,MAAM,CAAC,mBAAmB,CAAC,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE;QAC7C,SAAS,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;IACxB,CAAC,CAAC,CAAC;IACH,KAAK,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,EAAE,CAAC;QAC/B,SAAS,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;IACxB,CAAC;IACD,2DAA2D;IAC3D,MAAM,CAAC,cAAc,CAAC,SAAS,EAAE,cAAc,EAAE;QAChD,GAAG;YACF,OAAO;gBACN,SAAS,EAAE,CAAC,OAAyB,EAAE,EAAE;oBACxC,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,OAAwB,CAAC,CAAC;gBAClD,CAAC;gBACD,sBAAsB,EAAE,GAAG,EAAE,GAAE,CAAC;gBAChC,KAAK,EAAE,EAAE;aACT,CAAC;QACH,CAAC;QACD,YAAY,EAAE,IAAI;KAClB,CAAC,CAAC;IACH,OAAO,sBAAsB,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,IAAI,EAAE;QACjD,MAAM,MAAM,GAAG,MAAM,IAAI,EAAE,CAAC;QAC5B,OAAO,MAAM,CAAC;IACf,CAAC,CAAC,CAAC;AACJ,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,gBAAgB,GAAG,KAAK,EACpC,GAAgB,EAChB,IAAyB,EACxB,EAAE;IACH,8DAA8D;IAC9D,OAAO,qBAAqB,CAAC,GAAG,CAAC,GAAU,EAAE,IAAI,CAAC,CAAC;AACpD,CAAC,CAAC"}
+{"version":3,"file":"_context.js","sourceRoot":"","sources":["../src/_context.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AAGrD,OAAO,EACN,eAAe,GAIf,MAAM,iBAAiB,CAAC;AAEzB,OAAO,EAAE,aAAa,EAAE,aAAa,EAAE,MAAM,WAAW,CAAC;AAGzD,OAAO,EAAE,gBAAgB,EAAE,MAAM,aAAa,CAAC;AAoB/C,MAAM,OAAO,mBAAmB;IAM/B,KAAK,CAAY;IACjB,MAAM,CAAS;IACf,SAAS,CAAS;IAClB,MAAM,CAAS;IACf,EAAE,CAAmB;IACrB,MAAM,CAAiB;IACvB,MAAM,CAAiB;IACvB,KAAK,CAAuB;IAC5B,OAAO,CAAU;IACjB,MAAM,CAAS;IACf,MAAM,CAAU;IAChB,GAAG,CAAY;IACf,CAAC,aAAa,CAAC,CAAoB;IAC3B,OAAO,CAAmB;IAElC,YAAY,IAA4D;QACvE,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC;QACxB,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;QAC1B,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC;QAChC,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;QAC1B,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;QAC1B,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC;QAC5B,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;QAC1B,IAAI,CAAC,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC;QACpB,IAAI,CAAC,aAAa,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC;QACnC,IAAI,CAAC,KAAK,GAAG,IAAI,GAAG,EAAmB,CAAC;QACxC,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC;QAC5B,gBAAgB,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC,0CAA0C;IAC1E,CAAC;IAED,SAAS,CAAC,QAAsD;QAC/D,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;IAClC,CAAC;CACD;AAED,8DAA8D;AAC9D,MAAM,sBAAsB,GAAG,IAAI,iBAAiB,EAA+B,CAAC;AACpF,MAAM,qBAAqB,GAAG,IAAI,iBAAiB,EAAe,CAAC;AAEnE,MAAM,CAAC,MAAM,cAAc,GAAG,GAAY,EAAE;IAC3C,MAAM,OAAO,GAAG,sBAAsB,CAAC,QAAQ,EAAE,CAAC;IAClD,OAAO,CAAC,CAAC,OAAO,CAAC;AAClB,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,aAAa,GAAG,GAAY,EAAE;IAC1C,MAAM,OAAO,GAAG,qBAAqB,CAAC,QAAQ,EAAE,CAAC;IACjD,OAAO,CAAC,CAAC,OAAO,CAAC;AAClB,CAAC,CAAC;AAEF,MAAM,6BAA6B,GAAG,eAAe,CACpD,+BAA+B,EAC/B,+BAA+B,CAC/B,CAAC;AAEF,8DAA8D;AAC9D,MAAM,CAAC,MAAM,eAAe,GAAG,GAAgC,EAAE;IAChE,MAAM,OAAO,GAAG,sBAAsB,CAAC,QAAQ,EAAE,CAAC;IAClD,IAAI,CAAC,OAAO,EAAE,CAAC;QACd,MAAM,IAAI,6BAA6B,EAAE,CAAC;IAC3C,CAAC;IACD,OAAO,OAAO,CAAC;AAChB,CAAC,CAAC;AAEF,MAAM,4BAA4B,GAAG,eAAe,CACnD,8BAA8B,EAC9B,8BAA8B,CAC9B,CAAC;AAEF,MAAM,CAAC,MAAM,cAAc,GAAG,GAAgB,EAAE;IAC/C,MAAM,OAAO,GAAG,qBAAqB,CAAC,QAAQ,EAAE,CAAC;IACjD,IAAI,CAAC,OAAO,EAAE,CAAC;QACd,MAAM,IAAI,4BAA4B,EAAE,CAAC;IAC1C,CAAC;IACD,OAAO,OAAO,CAAC;AAChB,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,yBAAyB,GAAG,GAAG,EAAE,CAAC,sBAAsB,CAAC;AACtE,MAAM,CAAC,MAAM,wBAAwB,GAAG,GAAG,EAAE,CAAC,qBAAqB,CAAC;AAEpE;;;;GAIG;AACH,MAAM,CAAC,MAAM,uBAAuB,GAAG,GAAwC,EAAE;IAChF,MAAM,OAAO,GAAG,sBAAsB,CAAC,QAAQ,EAAE,CAAC;IAClD,IAAI,CAAC,OAAO;QAAE,OAAO,SAAS,CAAC;IAC/B,kCAAkC;IAClC,8DAA8D;IAC9D,OAAQ,OAAe,CAAC,aAAa,CAAC,EAAE,QAAQ,CAAC;AAClD,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,wBAAwB,GAAG,CAKvC,SAAkC,EAClC,IAA4D,EAC5D,IAAyB,EACxB,EAAE;IACH,MAAM,GAAG,GAAG,IAAI,mBAAmB,CAAgC,IAAI,CAAC,CAAC;IACzE,8DAA8D;IAC9D,MAAM,IAAI,GAAG,GAAU,CAAC;IACxB,MAAM,CAAC,mBAAmB,CAAC,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE;QAC7C,SAAS,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;IACxB,CAAC,CAAC,CAAC;IACH,KAAK,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,EAAE,CAAC;QAC/B,SAAS,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;IACxB,CAAC;IACD,2DAA2D;IAC3D,MAAM,CAAC,cAAc,CAAC,SAAS,EAAE,cAAc,EAAE;QAChD,GAAG;YACF,OAAO;gBACN,SAAS,EAAE,CAAC,OAAyB,EAAE,EAAE;oBACxC,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,OAAwB,CAAC,CAAC;gBAClD,CAAC;gBACD,sBAAsB,EAAE,GAAG,EAAE,GAAE,CAAC;gBAChC,KAAK,EAAE,EAAE;aACT,CAAC;QACH,CAAC;QACD,YAAY,EAAE,IAAI;KAClB,CAAC,CAAC;IACH,OAAO,sBAAsB,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,IAAI,EAAE;QACjD,MAAM,MAAM,GAAG,MAAM,IAAI,EAAE,CAAC;QAC5B,OAAO,MAAM,CAAC;IACf,CAAC,CAAC,CAAC;AACJ,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,gBAAgB,GAAG,KAAK,EACpC,GAAgB,EAChB,IAAyB,EACxB,EAAE;IACH,8DAA8D;IAC9D,OAAO,qBAAqB,CAAC,GAAG,CAAC,GAAU,EAAE,IAAI,CAAC,CAAC;AACpD,CAAC,CAAC"}

package/dist/_server.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"_server.d.ts","sourceRoot":"","sources":["../src/_server.ts"],"names":[],"mappings":"AAAA,OAAO,EAKN,KAAK,MAAM,EAIX,MAAM,oBAAoB,CAAC;AAG5B,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,+BAA+B,CAAC;AAInE,OAAO,EAAE,IAAI,EAAE,KAAK,OAAO,IAAI,WAAW,EAAE,MAAM,MAAM,CAAC;AAEzD,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,UAAU,CAAC;AAIjD,OAAO,KAAK,EAAE,SAAS,EAAE,GAAG,EAAE,gBAAgB,EAAE,MAAM,OAAO,CAAC;AAG9D,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,UAAU,CAAC;AAyBvC,wBAAgB,SAAS,wCAExB;AAED,wBAAgB,SAAS,mFAExB;AAID,wBAAgB,SAAS,kBAExB;AAED,wBAAgB,SAAS,kBAExB;AAED,wBAAgB,WAAW,QAE1B;AAcD;;;GAGG;AACH,wBAAgB,gBAAgB,CAAC,SAAS,EAAE,aAAa,QAExD;AA6CD,wBAAgB,cAAc,CAAC,CAAC,SAAS,GAAG,EAAE,CAAC,EAAE,WAAW,CAAC,CAAC,CAAC,GACvC,WAAW,CAAC;IAAE,SAAS,EAAE,gBAAgB,CAAA;CAAE,CAAC,CACnE;AAMD,eAAO,MAAM,WAAW,YAEvB,CAAC;AAEF,eAAO,MAAM,YAAY,GAAU,SAAS,EAC3C,QAAQ,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,EAC5B,qBAAqB,MAAM,OAAO,CAAC,SAAS,CAAC,EAC7C,SAAS,SAAS,CAAC,SAAS,CAAC,KAC3B,OAAO,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,gBAAgB,CAAC,EAAE,SAAS,CAAC,CAqRnD,CAAC"}
+{"version":3,"file":"_server.d.ts","sourceRoot":"","sources":["../src/_server.ts"],"names":[],"mappings":"AAAA,OAAO,EAKN,KAAK,MAAM,EAIX,MAAM,oBAAoB,CAAC;AAG5B,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,+BAA+B,CAAC;AAInE,OAAO,EAAE,IAAI,EAAE,KAAK,OAAO,IAAI,WAAW,EAAE,MAAM,MAAM,CAAC;AAEzD,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,UAAU,CAAC;AAIjD,OAAO,KAAK,EAAE,SAAS,EAAE,GAAG,EAAE,gBAAgB,EAAE,MAAM,OAAO,CAAC;AAG9D,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,UAAU,CAAC;AA0BvC,wBAAgB,SAAS,wCAExB;AAED,wBAAgB,SAAS,mFAExB;AAID,wBAAgB,SAAS,kBAExB;AAED,wBAAgB,SAAS,kBAExB;AAED,wBAAgB,WAAW,QAE1B;AAcD;;;GAGG;AACH,wBAAgB,gBAAgB,CAAC,SAAS,EAAE,aAAa,QAExD;AA+CD,wBAAgB,cAAc,CAAC,CAAC,SAAS,GAAG,EAAE,CAAC,EAAE,WAAW,CAAC,CAAC,CAAC,GACvC,WAAW,CAAC;IAAE,SAAS,EAAE,gBAAgB,CAAA;CAAE,CAAC,CACnE;AAMD,eAAO,MAAM,WAAW,YAEvB,CAAC;AAEF,eAAO,MAAM,YAAY,GAAU,SAAS,EAC3C,QAAQ,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,EAC5B,qBAAqB,MAAM,OAAO,CAAC,SAAS,CAAC,EAC7C,SAAS,SAAS,CAAC,SAAS,CAAC,KAC3B,OAAO,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,gBAAgB,CAAC,EAAE,SAAS,CAAC,CA8RnD,CAAC"}

package/dist/_server.js

@@ -12,7 +12,8 @@ import { extractTraceContextFromRequest } from './otel/http';
 import { register } from './otel/config';
 import { isIdle } from './_idle';
 import * as runtimeConfig from './_config';
-import { inAgentContext, getAgentContext, runInHTTPContext } from './_context';
+import { inAgentContext, runInHTTPContext } from './_context';
+import { runAgentShutdowns, createAgentMiddleware } from './agent';
 import { createServices, getThreadProvider, getSessionProvider, getSessionEventProvider, } from './_services';
 import { generateId } from './session';
 import WaitUntilHandler from './_waituntil';
@@ -72,12 +73,14 @@ function registerAgentuitySpanProcessor() {
                 '@agentuity/environment': environment,
             };
             if (inAgentContext()) {
-                const agentCtx = getAgentContext();
-                if (agentCtx.current?.metadata) {
-                    attrs['@agentuity/agentId'] = agentCtx.current.metadata.id;
-                    attrs['@agentuity/agentInstanceId'] = agentCtx.current.metadata.agentId;
-                    attrs['@agentuity/agentDescription'] = agentCtx.current.metadata.description;
-                    attrs['@agentuity/agentName'] = agentCtx.current.metadata.name;
+                // eslint-disable-next-line @typescript-eslint/no-require-imports
+                const { getCurrentAgentMetadata } = require('./_context');
+                const current = getCurrentAgentMetadata();
+                if (current) {
+                    attrs['@agentuity/agentId'] = current.id;
+                    attrs['@agentuity/agentInstanceId'] = current.agentId;
+                    attrs['@agentuity/agentDescription'] = current.description;
+                    attrs['@agentuity/agentName'] = current.name;
                 }
             }
             span.setAttributes(attrs);
@@ -214,9 +217,12 @@ export const createServer = async (router, appStateInitializer, config) => {
             }
             const file = Bun.file(routeMappingPath);
             if (!(await file.exists())) {
-                c.var.logger.fatal('error loading %s. this is a build issue!', routeMappingPath);
+                c.var.logger.warn('Route mapping file not found at %s. Route tracking will be disabled.', routeMappingPath);
+                routeMapping = {}; // Empty mapping, no route tracking
+            }
+            else {
+                routeMapping = (await file.json());
             }
-            routeMapping = (await file.json());
         }
         const matches = matchedRoutes(c).filter((m) => m.method !== 'ALL' && (m.path.startsWith('/api') || m.path.startsWith('/agent/')));
         const _c = privateContext(c);
@@ -233,11 +239,9 @@ export const createServer = async (router, appStateInitializer, config) => {
         _c.set('trigger', 'api'); // will get overwritten below if another trigger
         return next();
     });
-    router.use('/agent/*', routePathMapper);
     router.use('/api/*', routePathMapper);
     // Attach services and agent registry to context for API routes
     router.use('/api/*', async (c, next) => {
-        const { createAgentMiddleware } = await import('./agent');
         // Use a null agent name to just populate the agent registry without setting current agent
         return createAgentMiddleware('')(c, next);
     });
@@ -249,10 +253,16 @@ export const createServer = async (router, appStateInitializer, config) => {
             await next();
         });
         router.use(`/api/${trigger}/*`, middleware);
-        router.use(`/agent/${trigger}/*`, middleware);
     }
     router.use('/api/*', otelMiddleware);
-    router.use('/agent/*', otelMiddleware);
+    // Apply otelMiddleware to workbench routes for full telemetry and session tracking
+    if (config?.services?.workbench) {
+        router.use('/_agentuity/workbench/*', async (c, next) => {
+            // Use a null agent name to just populate the agent registry without setting current agent
+            return createAgentMiddleware('')(c, next);
+        });
+        router.use('/_agentuity/workbench/*', otelMiddleware);
+    }
     const shutdown = async () => {
         if (isShutdown) {
             return;
@@ -282,7 +292,6 @@ export const createServer = async (router, appStateInitializer, config) => {
             }
             otel.logger.debug('no more pending connections');
             // Run agent shutdowns first
-            const { runAgentShutdowns } = await import('./agent');
             await runAgentShutdowns(globalAppState);
             // Run app shutdown if provided
             if (config?.shutdown && globalAppState) {
@@ -432,6 +441,7 @@ const otelMiddleware = createMiddleware(async (c, next) => {
                             sessionEventProvider
                                 .complete({
                                 id: sessionId,
+                                threadId: thread.empty() ? null : thread.id,
                                 statusCode: c.res.status,
                                 agentIds: Array.from(agentIds),
                                 userData,
@@ -456,6 +466,7 @@ const otelMiddleware = createMiddleware(async (c, next) => {
                             sessionEventProvider
                                 .complete({
                                 id: sessionId,
+                                threadId: thread.empty() ? null : thread.id,
                                 statusCode: c.res.status,
                                 error: message,
                                 agentIds: Array.from(agentIds),
@@ -476,6 +487,7 @@ const otelMiddleware = createMiddleware(async (c, next) => {
                         sessionEventProvider
                             .complete({
                             id: sessionId,
+                            threadId: thread.empty() ? null : thread.id,
                             statusCode: c.res.status,
                             agentIds: Array.from(agentIds),
                             userData,
@@ -500,6 +512,7 @@ const otelMiddleware = createMiddleware(async (c, next) => {
                     sessionEventProvider
                         .complete({
                         id: sessionId,
+                        threadId: thread.empty() ? null : thread.id,
                         statusCode: c.res.status,
                         error: message,
                         agentIds: Array.from(agentIds),