n8n-nodes-rooyai-chat 0.3.0 → 0.4.1

package/dist/nodes/Rooyai/N8nLlmTracing.d.ts

@@ -1,11 +1,21 @@
 import { BaseCallbackHandler } from '@langchain/core/callbacks/base';
 import type { Serialized } from '@langchain/core/load/serializable';
-import { ISupplyDataFunctions } from 'n8n-workflow';
+import type { LLMResult } from '@langchain/core/outputs';
+import type { ISupplyDataFunctions } from 'n8n-workflow';
 export declare class N8nLlmTracing extends BaseCallbackHandler {
+    private executionFunctions;
     name: string;
-    supplyDataFunctions: ISupplyDataFunctions;
-    constructor(supplyDataFunctions: ISupplyDataFunctions);
-    handleLLMStart(llm: Serialized, prompts: string[], runId: string, parentRunId?: string, extraParams?: Record<string, unknown>, tags?: string[], metadata?: Record<string, unknown>): Promise<void>;
-    handleLLMEnd(output: any, runId: string, parentRunId?: string, tags?: string[]): Promise<void>;
-    handleLLMError(error: Error, runId: string, parentRunId?: string, tags?: string[]): Promise<void>;
+    awaitHandlers: boolean;
+    connectionType: "ai_languageModel";
+    promptTokensEstimate: number;
+    completionTokensEstimate: number;
+    runsMap: Record<string, {
+        index: number;
+        messages: string[];
+        options: any;
+    }>;
+    constructor(executionFunctions: ISupplyDataFunctions);
+    handleLLMStart(llm: Serialized, prompts: string[], runId: string): Promise<void>;
+    handleLLMEnd(output: LLMResult, runId: string): Promise<void>;
+    handleLLMError(error: Error, runId: string): Promise<void>;
 }

package/dist/nodes/Rooyai/N8nLlmTracing.js

@@ -2,37 +2,82 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.N8nLlmTracing = void 0;
 const base_1 = require("@langchain/core/callbacks/base");
+const n8n_workflow_1 = require("n8n-workflow");
 class N8nLlmTracing extends base_1.BaseCallbackHandler {
-    constructor(supplyDataFunctions) {
+    constructor(executionFunctions) {
         super();
-        this.name = 'n8n_llm_tracing';
-        this.supplyDataFunctions = supplyDataFunctions;
+        this.executionFunctions = executionFunctions;
+        this.name = 'N8nLlmTracing';
+        this.awaitHandlers = true;
+        this.connectionType = n8n_workflow_1.NodeConnectionTypes.AiLanguageModel;
+        this.promptTokensEstimate = 0;
+        this.completionTokensEstimate = 0;
+        this.runsMap = {};
     }
-    async handleLLMStart(llm, prompts, runId, parentRunId, extraParams, tags, metadata) {
-        if (this.supplyDataFunctions.logger) {
-            this.supplyDataFunctions.logger.info('🚀 Rooyai LLM started', {
+    async handleLLMStart(llm, prompts, runId) {
+        const estimatedTokens = prompts.reduce((sum, p) => sum + Math.ceil(p.length / 4), 0);
+        this.promptTokensEstimate = estimatedTokens;
+        const options = llm.type === 'constructor' ? llm.kwargs : llm;
+        const { index } = this.executionFunctions.addInputData(this.connectionType, [
+            [
+                {
+                    json: {
+                        messages: prompts,
+                        estimatedTokens,
+                        options,
+                    },
+                },
+            ],
+        ]);
+        this.runsMap[runId] = {
+            index,
+            options,
+            messages: prompts,
+        };
+        if (this.executionFunctions.logger) {
+            this.executionFunctions.logger.info('🚀 Rooyai LLM started', {
                 runId,
                 promptCount: prompts.length,
-                model: extraParams?.invocation_params?.model || 'unknown',
+                estimatedTokens,
             });
         }
     }
-    async handleLLMEnd(output, runId, parentRunId, tags) {
-        if (this.supplyDataFunctions.logger) {
-            this.supplyDataFunctions.logger.info('✅ Rooyai LLM finished successfully', {
+    async handleLLMEnd(output, runId) {
+        const runDetails = this.runsMap[runId] ?? { index: Object.keys(this.runsMap).length, messages: [], options: {} };
+        const tokenUsage = {
+            completionTokens: output?.llmOutput?.tokenUsage?.completionTokens ?? 0,
+            promptTokens: output?.llmOutput?.tokenUsage?.promptTokens ?? 0,
+            totalTokens: output?.llmOutput?.tokenUsage?.totalTokens ?? 0,
+        };
+        const response = {
+            response: { generations: output.generations },
+        };
+        if (tokenUsage.totalTokens > 0) {
+            response.tokenUsage = tokenUsage;
+        }
+        if (output?.llmOutput?.costUsd !== undefined) {
+            response.costUsd = output.llmOutput.costUsd;
+        }
+        this.executionFunctions.addOutputData(this.connectionType, runDetails.index, [[{ json: { ...response } }]]);
+        if (this.executionFunctions.logger) {
+            this.executionFunctions.logger.info('✅ Rooyai LLM finished successfully', {
                 runId,
-                generationCount: output?.generations?.length || 0,
+                generationCount: output.generations?.length || 0,
+                cost: output?.llmOutput?.costUsd,
             });
         }
+        delete this.runsMap[runId];
     }
-    async handleLLMError(error, runId, parentRunId, tags) {
-        if (this.supplyDataFunctions.logger) {
-            this.supplyDataFunctions.logger.error('❌ Rooyai LLM error', {
+    async handleLLMError(error, runId) {
+        const runDetails = this.runsMap[runId] ?? { index: Object.keys(this.runsMap).length, messages: [], options: {} };
+        if (this.executionFunctions.logger) {
+            this.executionFunctions.logger.error('❌ Rooyai LLM error', {
                 runId,
                 error: error.message,
                 stack: error.stack,
             });
         }
+        delete this.runsMap[runId];
     }
 }
 exports.N8nLlmTracing = N8nLlmTracing;

package/dist/nodes/Rooyai/Rooyai.node.js

@@ -38,14 +38,14 @@ class Rooyai {
                     name: 'model',
                     type: 'options',
                     options: [
-                        {
-                            name: 'Gemini 2.0 Flash',
-                            value: 'gemini-2.0-flash',
-                        },
                         {
                             name: 'Llama 3.3 70B',
                             value: 'llama-3.3-70b',
                         },
+                        {
+                            name: 'Gemini 2.0 Flash',
+                            value: 'gemini-2.0-flash',
+                        },
                         {
                             name: 'DeepSeek R1',
                             value: 'deepseek-r1',

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "n8n-nodes-rooyai-chat",
-	"version": "0.3.0",
+	"version": "0.4.1",
 	"description": "n8n supply node for Rooyai Chat API - Provides Rooyai language models for use with Basic LLM Chain, AI Agent, and other AI nodes.",
 	"keywords": [
 		"n8n-community-node-package",