graphlit-client 1.0.20250613008 → 1.0.20250613009

package/dist/client.js

@@ -1499,13 +1499,17 @@ class Graphlit {
             // 3. Tool calling loop
             const allToolCalls = [];
             let rounds = 0;
-            let totalTokens = 0;
+            let totalTokens = currentMessage?.tokens || 0;
+            const toolStartTime = Date.now();
+            let toolTime = 0;
             while (currentMessage.toolCalls?.length &&
                 rounds < maxRounds &&
                 !abortController.signal.aborted) {
                 rounds++;
                 // Execute tools
+                const toolExecStart = Date.now();
                 const toolResults = await this.executeToolsForPromptAgent(currentMessage.toolCalls.filter((tc) => tc !== null), toolHandlers || {}, allToolCalls, abortController.signal);
+                toolTime += Date.now() - toolExecStart;
                 if (abortController.signal.aborted) {
                     throw new Error("Operation timed out");
                 }
@@ -1519,9 +1523,31 @@ class Graphlit {
                     totalTokens += continueResponse.continueConversation.message.tokens;
                 }
             }
+            // Calculate metrics
+            const totalTime = Date.now() - startTime;
+            const llmTime = totalTime - toolTime;
+            const metrics = {
+                totalTime,
+                llmTime,
+                toolTime,
+                toolExecutions: allToolCalls.length,
+                rounds,
+            };
+            // Build usage info if we have token data
+            const usage = totalTokens > 0 ? {
+                promptTokens: 0, // We don't have this breakdown from the API
+                completionTokens: totalTokens,
+                totalTokens: totalTokens,
+                model: currentMessage?.model || undefined,
+            } : undefined;
             return {
                 message: currentMessage?.message || "",
                 conversationId: actualConversationId,
+                conversationMessage: currentMessage ? currentMessage : undefined,
+                toolCalls: currentMessage?.toolCalls?.filter((tc) => tc !== null),
+                toolResults: allToolCalls,
+                metrics,
+                usage,
             };
         }
         catch (error) {
@@ -1536,6 +1562,14 @@ class Graphlit {
                     recoverable: isTimeout || error.code === "RATE_LIMIT",
                     details: error.response?.data,
                 },
+                // Include partial metrics if available
+                metrics: {
+                    totalTime: Date.now() - startTime,
+                    llmTime: 0,
+                    toolTime: 0,
+                    toolExecutions: 0,
+                    rounds: 0,
+                },
             };
         }
         finally {

package/dist/types/agent.d.ts

@@ -1,11 +1,26 @@
+import { ConversationMessage, ConversationToolCall } from "../generated/graphql-types.js";
 export type ToolHandler = (args: any) => Promise<any>;
 export interface AgentOptions {
     maxToolRounds?: number;
     timeout?: number;
 }
+export interface AgentMetrics {
+    totalTime: number;
+    llmTime?: number;
+    toolTime?: number;
+    ttft?: number;
+    tokensPerSecond?: number;
+    toolExecutions?: number;
+    rounds?: number;
+}
 export interface AgentResult {
     message: string;
     conversationId: string;
+    conversationMessage?: ConversationMessage;
+    toolCalls?: ConversationToolCall[];
+    toolResults?: ToolCallResult[];
+    metrics?: AgentMetrics;
+    usage?: UsageInfo;
     error?: AgentError;
 }
 export interface StreamAgentOptions {

package/package.json

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