morpheus-cli 0.8.0 → 0.8.3

package/dist/runtime/tools/trinity-tool.js

@@ -7,6 +7,7 @@ import { DisplayManager } from "../display.js";
 import { ConfigManager } from "../../config/manager.js";
 import { Trinity } from "../trinity.js";
 import { ChannelRegistry } from "../../channels/registry.js";
+import { AuditRepository } from "../audit/repository.js";
 const TRINITY_BASE_DESCRIPTION = `Delegate a database task to Trinity, the specialized database subagent, asynchronously.
 
 This tool enqueues a background task and returns an acknowledgement with task id.
@@ -71,7 +72,21 @@ export const TrinityDelegateTool = tool(async ({ task, context }) => {
                     source: 'TrinityDelegateTool',
                     level: 'info',
                 });
-                return result;
+                if (result.usage) {
+                    AuditRepository.getInstance().insert({
+                        session_id: sessionId,
+                        event_type: 'llm_call',
+                        agent: 'trinity',
+                        provider: result.usage.provider,
+                        model: result.usage.model,
+                        input_tokens: result.usage.inputTokens,
+                        output_tokens: result.usage.outputTokens,
+                        duration_ms: result.usage.durationMs,
+                        status: 'success',
+                        metadata: { step_count: result.usage.stepCount, mode: 'sync' },
+                    });
+                }
+                return result.output;
             }
             catch (syncErr) {
                 // Still count as sync delegation so Oracle passes through the error message

package/dist/runtime/trinity.js

@@ -210,31 +210,46 @@ ${context ? `CONTEXT FROM ORACLE:\n${context}` : ''}
         const userMessage = new HumanMessage(task);
         const messages = [systemMessage, userMessage];
         try {
+            const startMs = Date.now();
             const response = await this.agent.invoke({ messages });
+            const durationMs = Date.now() - startMs;
             const lastMessage = response.messages[response.messages.length - 1];
             const content = typeof lastMessage.content === 'string'
                 ? lastMessage.content
                 : JSON.stringify(lastMessage.content);
+            const rawUsage = lastMessage.usage_metadata
+                ?? lastMessage.response_metadata?.usage
+                ?? lastMessage.response_metadata?.tokenUsage
+                ?? lastMessage.usage;
+            const inputTokens = rawUsage?.input_tokens ?? 0;
+            const outputTokens = rawUsage?.output_tokens ?? 0;
+            const stepCount = response.messages.filter((m) => m instanceof AIMessage).length;
             const targetSession = sessionId ?? Trinity.currentSessionId ?? 'trinity';
             const history = new SQLiteChatMessageHistory({ sessionId: targetSession });
             try {
                 const persisted = new AIMessage(content);
-                persisted.usage_metadata =
-                    lastMessage.usage_metadata ??
-                        lastMessage.response_metadata?.usage ??
-                        lastMessage.response_metadata?.tokenUsage ??
-                        lastMessage.usage;
-                persisted.provider_metadata = {
-                    provider: trinityConfig.provider,
-                    model: trinityConfig.model,
-                };
+                if (rawUsage)
+                    persisted.usage_metadata = rawUsage;
+                persisted.provider_metadata = { provider: trinityConfig.provider, model: trinityConfig.model };
+                persisted.agent_metadata = { agent: 'trinity' };
+                persisted.duration_ms = durationMs;
                 await history.addMessage(persisted);
             }
             finally {
                 history.close();
             }
             this.display.log('Trinity task completed.', { source: 'Trinity' });
-            return content;
+            return {
+                output: content,
+                usage: {
+                    provider: trinityConfig.provider,
+                    model: trinityConfig.model,
+                    inputTokens,
+                    outputTokens,
+                    durationMs,
+                    stepCount,
+                },
+            };
         }
         catch (err) {
             throw new ProviderError(trinityConfig.provider, err, 'Trinity task execution failed');