graphlit-client 1.0.20260405003 → 1.0.20260406002

package/dist/client.js

@@ -6711,6 +6711,31 @@ class Graphlit {
         let lastContextWindow;
         let conversationId = options?.conversationId ?? "";
         let resolvedAgentId = agentId ?? "";
+        // Accumulate token usage across all turns from conversation_completed events
+        let accumulatedUsage = {
+            promptTokens: 0,
+            completionTokens: 0,
+            totalTokens: 0,
+        };
+        let lastTurnUsage;
+        // Wrap onEvent to capture per-turn usage from conversation_completed events
+        const wrappedOnEvent = (event) => {
+            if (event.type === 'conversation_completed' && event.usage) {
+                const u = event.usage;
+                lastTurnUsage = {
+                    promptTokens: u.promptTokens || 0,
+                    completionTokens: u.completionTokens || 0,
+                    totalTokens: u.totalTokens || 0,
+                    model: u.model,
+                };
+                accumulatedUsage.promptTokens += lastTurnUsage.promptTokens;
+                accumulatedUsage.completionTokens += lastTurnUsage.completionTokens;
+                accumulatedUsage.totalTokens += lastTurnUsage.totalTokens;
+                if (lastTurnUsage.model)
+                    accumulatedUsage.model = lastTurnUsage.model;
+            }
+            onEvent(event);
+        };
         try {
             // ── Phase 1: Setup ─────────────────────────────────────────────────
             // Check abort early
@@ -6723,6 +6748,7 @@ class Graphlit {
                     turnResults,
                     totalToolCalls,
                     wallClockMs: Date.now() - runStart,
+                    usage: accumulatedUsage.totalTokens > 0 ? accumulatedUsage : undefined,
                 });
             }
             // Resolve or create Agent entity
@@ -6987,7 +7013,7 @@ class Graphlit {
                     // Emit context window event
                     if (formatDetails?.tokenLimit && formatDetails?.messages) {
                         const usedTokens = formatDetails.messages.reduce((sum, msg) => sum + (msg?.tokens || 0), 0);
-                        onEvent({
+                        wrappedOnEvent({
                             type: "context_window",
                             usage: {
                                 usedTokens,
@@ -7004,7 +7030,7 @@ class Graphlit {
                 const modelEnum = fullSpec ? getModelEnum(fullSpec) : undefined;
                 const serviceType = fullSpec ? getServiceType(fullSpec) : undefined;
                 const uiAdapter = new UIEventAdapter((event) => {
-                    onEvent(event);
+                    wrappedOnEvent(event);
                     options?.onStreamEvent?.(event);
                 }, conversationId, {
                     smoothingEnabled: options?.smoothingEnabled ?? true,
@@ -7109,7 +7135,10 @@ class Graphlit {
                         ? loopResult.contextActions
                         : undefined,
                     errors: loopResult.errors.length > 0 ? loopResult.errors : undefined,
+                    usage: lastTurnUsage,
                 };
+                // Reset per-turn usage for next turn
+                lastTurnUsage = undefined;
                 turnResults.push(turnResult);
                 totalToolCalls += loopResult.toolCallCount;
                 finalMessage = loopResult.finalAssistantMessage || loopResult.fullMessage;
@@ -7185,6 +7214,7 @@ class Graphlit {
                 error: errorMessage,
                 stuckPattern,
                 qualityAssessment,
+                usage: accumulatedUsage.totalTokens > 0 ? accumulatedUsage : undefined,
             });
         }
         catch (error) {
@@ -7210,6 +7240,7 @@ class Graphlit {
                 contextWindowAtEnd: lastContextWindow,
                 error: errorMessage,
                 stuckPattern,
+                usage: accumulatedUsage.totalTokens > 0 ? accumulatedUsage : undefined,
             });
         }
     }
@@ -7418,6 +7449,7 @@ class Graphlit {
             error: params.error,
             stuckPattern: params.stuckPattern,
             qualityAssessment: params.qualityAssessment,
+            usage: params.usage,
         };
     }
     /**

package/dist/generated/graphql-documents.js

@@ -81,6 +81,10 @@ export const GetAgent = gql `
       id
       name
     }
+    persona {
+      id
+      name
+    }
     trigger {
       types
       fileTypes
@@ -327,6 +331,10 @@ export const QueryAgents = gql `
         id
         name
       }
+      persona {
+        id
+        name
+      }
       trigger {
         types
         fileTypes

package/dist/generated/graphql-types.d.ts

@@ -139,6 +139,8 @@ export type Agent = {
     name: Scalars['String']['output'];
     /** The owner of the agent. */
     owner: Owner;
+    /** The persona associated with this agent. */
+    persona?: Maybe<Persona>;
     /** The execution prompt run on each triggered activation (schedule, webhook, event, manual). */
     prompt?: Maybe<Scalars['String']['output']>;
     /** The relevance score of the agent. */
@@ -295,6 +297,8 @@ export type AgentInput = {
     filter?: InputMaybe<ContentCriteriaInput>;
     /** The name of the agent. */
     name: Scalars['String']['input'];
+    /** The persona associated with this agent. */
+    persona?: InputMaybe<EntityReferenceInput>;
     /** The execution prompt run on each triggered activation. */
     prompt?: InputMaybe<Scalars['String']['input']>;
     /** The agent research depth. Defaults to STANDARD. */
@@ -396,6 +400,8 @@ export type AgentUpdateInput = {
     id: Scalars['ID']['input'];
     /** The name of the agent. */
     name?: InputMaybe<Scalars['String']['input']>;
+    /** The persona associated with this agent. */
+    persona?: InputMaybe<EntityReferenceInput>;
     /** The execution prompt run on each triggered activation. */
     prompt?: InputMaybe<Scalars['String']['input']>;
     /** The agent research depth. Defaults to STANDARD. */
@@ -25353,6 +25359,11 @@ export type GetAgentQuery = {
             id: string;
             name: string;
         } | null;
+        persona?: {
+            __typename?: 'Persona';
+            id: string;
+            name: string;
+        } | null;
         trigger?: {
             __typename?: 'AgentTriggerFilter';
             types?: Array<ContentTypes> | null;
@@ -25664,6 +25675,11 @@ export type QueryAgentsQuery = {
                 id: string;
                 name: string;
             } | null;
+            persona?: {
+                __typename?: 'Persona';
+                id: string;
+                name: string;
+            } | null;
             trigger?: {
                 __typename?: 'AgentTriggerFilter';
                 types?: Array<ContentTypes> | null;

package/dist/types/agent.d.ts

@@ -114,6 +114,8 @@ export interface TurnResult {
     contextWindowUsage?: ContextWindowUsage;
     contextActions?: ContextManagementAction[];
     errors?: string[];
+    /** Token usage for this turn (from native LLM provider). */
+    usage?: UsageInfo;
 }
 /** Budget snapshot passed to the onBudgetWarning callback. */
 export interface BudgetSnapshot {
@@ -181,6 +183,8 @@ export interface RunAgentResult {
     error?: string;
     stuckPattern?: string;
     qualityAssessment?: QualityAssessment;
+    /** Accumulated token usage across all turns (from native LLM provider). */
+    usage?: UsageInfo;
 }
 import type { ReasoningMetadata } from "./ui-events.js";
 import type { TokenBudgetTracker } from "../helpers/context-management.js";

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "graphlit-client",
-  "version": "1.0.20260405003",
+  "version": "1.0.20260406002",
   "description": "Graphlit API Client for TypeScript",
   "type": "module",
   "main": "./dist/client.js",