Quality guardrail and strict validation

Author: 7vignesh

packages/agents/src/agents/base.agent.ts

@@ -10,6 +10,11 @@ export type AgentRunResult<T> = {
   cached: boolean;
   durationMs: number;
   tokensUsed: number;
+  estimatedInputTokens: number;
+  compressionPasses: number;
+  providerInputTokens: number;
+  providerOutputTokens: number;
+  model: string;
 };
 
 export async function runAgent<TInput, TOutput>(
@@ -28,6 +33,11 @@ export async function runAgent<TInput, TOutput>(
       cached: true,
       durationMs: 0,
       tokensUsed: 0,
+      estimatedInputTokens: 0,
+      compressionPasses: 0,
+      providerInputTokens: 0,
+      providerOutputTokens: 0,
+      model: "cache",
     };
   }
 
@@ -57,5 +67,10 @@ export async function runAgent<TInput, TOutput>(
     cached: false,
     durationMs,
     tokensUsed: response.tokensUsed,
+    estimatedInputTokens: optimized.estimatedInputTokens,
+    compressionPasses: optimized.compressionPasses,
+    providerInputTokens: response.inputTokens ?? optimized.estimatedInputTokens,
+    providerOutputTokens: response.outputTokens ?? optimized.maxOutputTokens,
+    model: response.model ?? optimized.model,
   };
 }

packages/agents/src/orchestrator/orchestrator.service.ts

@@ -36,13 +36,27 @@ function agentCompleted(
   durationMs: number,
   cached: boolean,
   tokensUsed: number,
+  estimatedInputTokens: number,
+  compressionPasses: number,
+  providerInputTokens: number,
+  providerOutputTokens: number,
+  model: string,
 ): SSEEvent {
   return {
     type: "agent_completed",
     jobId,
     agent,
     timestamp: now(),
-    payload: { durationMs, cached, tokensUsed },
+    payload: {
+      durationMs,
+      cached,
+      tokensUsed,
+      estimatedInputTokens,
+      compressionPasses,
+      providerInputTokens,
+      providerOutputTokens,
+      model,
+    },
   };
 }
 
@@ -54,12 +68,35 @@ async function runWithEmit<T>(
   jobId: string,
   agentName: AgentName,
   emit: (event: SSEEvent) => void,
-  fn: () => Promise<{ output: T; cached: boolean; durationMs: number; tokensUsed: number }>,
+  fn: () => Promise<{
+    output: T;
+    cached: boolean;
+    durationMs: number;
+    tokensUsed: number;
+    estimatedInputTokens: number;
+    compressionPasses: number;
+    providerInputTokens: number;
+    providerOutputTokens: number;
+    model: string;
+  }>,
 ): Promise<T> {
   emit(agentStarted(jobId, agentName));
   try {
     const result = await fn();
-    emit(agentCompleted(jobId, agentName, result.durationMs, result.cached, result.tokensUsed));
+    emit(
+      agentCompleted(
+        jobId,
+        agentName,
+        result.durationMs,
+        result.cached,
+        result.tokensUsed,
+        result.estimatedInputTokens,
+        result.compressionPasses,
+        result.providerInputTokens,
+        result.providerOutputTokens,
+        result.model,
+      ),
+    );
     return result.output;
   } catch (err) {
     const msg = err instanceof Error ? err.message : String(err);

packages/shared/src/schemas/sse.schema.ts

@@ -27,6 +27,11 @@ export const AgentCompletedEventSchema = SSEBaseSchema.extend({
     durationMs: z.number(),
     cached: z.boolean(),
     tokensUsed: z.number(),
+    estimatedInputTokens: z.number(),
+    compressionPasses: z.number(),
+    providerInputTokens: z.number(),
+    providerOutputTokens: z.number(),
+    model: z.string(),
   }),
 });