n8n-nodes-rooyai-chat 0.2.0 → 0.4.0

package/README.md

@@ -4,7 +4,7 @@ Custom n8n node for Rooyai Chat API - A supply node that provides Rooyai languag
 
 ## Description
 
-This n8n custom node provides a **Rooyai Chat Model** supply node that can be connected to the **Basic LLM Chain** node, **AI Agent**, **Better AI Agent**, and other AI processing nodes in n8n.
+This n8n custom node provides a **Rooyai Chat Model** supply node that can be connected to the **Basic LLM Chain** node, **AI Agent**, and other AI processing nodes in n8n.
 
 ## Features
 
@@ -77,7 +77,7 @@ Then restart your n8n instance. The node will appear in the node palette under *
 Connect the **Model** output from **Rooyai Chat Model** to:
 - **Basic LLM Chain**
 - **AI Agent**
-- **Better AI Agent**
+
 - Any other n8n AI node that accepts a Language Model
 
 ## Available Models

package/dist/credentials/RooyaiApi.credentials.d.ts

@@ -1,8 +1,9 @@
-import { IAuthenticateGeneric, ICredentialType, INodeProperties } from 'n8n-workflow';
+import { IAuthenticateGeneric, ICredentialTestRequest, ICredentialType, INodeProperties } from 'n8n-workflow';
 export declare class RooyaiApi implements ICredentialType {
     name: string;
     displayName: string;
     documentationUrl: string;
     properties: INodeProperties[];
     authenticate: IAuthenticateGeneric;
+    test: ICredentialTestRequest;
 }

package/dist/credentials/RooyaiApi.credentials.js

@@ -35,6 +35,21 @@ class RooyaiApi {
                 },
             },
         };
+        this.test = {
+            request: {
+                baseURL: '={{$credentials.baseUrl}}',
+                url: '/chat',
+                method: 'POST',
+                headers: {
+                    'Content-Type': 'application/json',
+                },
+                body: {
+                    model: 'gemini-2.0-flash',
+                    messages: [{ role: 'user', content: 'test' }],
+                    max_tokens: 5,
+                },
+            },
+        };
     }
 }
 exports.RooyaiApi = RooyaiApi;

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.debug('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,
-                parentRunId,
-                prompts,
+                promptCount: prompts.length,
+                estimatedTokens,
             });
         }
     }
-    async handleLLMEnd(output, runId, parentRunId, tags) {
-        if (this.supplyDataFunctions.logger) {
-            this.supplyDataFunctions.logger.debug('Rooyai LLM finished', {
+    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,
-                parentRunId,
+                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,
-                parentRunId,
                 error: error.message,
+                stack: error.stack,
             });
         }
+        delete this.runsMap[runId];
     }
 }
 exports.N8nLlmTracing = N8nLlmTracing;

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

@@ -147,10 +147,11 @@ class Rooyai {
         const modelName = this.getNodeParameter('model', itemIndex);
         const options = this.getNodeParameter('options', itemIndex, {});
         if (this.logger) {
-            this.logger.info('Rooyai Chat Model initialized', {
+            this.logger.info('🎯 Initializing Rooyai Chat Model', {
                 model: modelName,
                 temperature: options.temperature ?? 0.7,
                 maxTokens: options.maxTokensToSample,
+                baseUrl,
             });
         }
         const model = new RooyaiLangChainWrapper_1.RooyaiLangChainWrapper({
@@ -161,7 +162,13 @@ class Rooyai {
             temperature: options.temperature ?? 0.7,
             supplyDataFunctions: this,
             callbacks: [new N8nLlmTracing_1.N8nLlmTracing(this)],
+            verbose: true,
         });
+        if (this.logger) {
+            this.logger.info('✅ Rooyai Chat Model ready', {
+                model: modelName,
+            });
+        }
         return {
             response: model,
         };

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

@@ -11,6 +11,7 @@ interface RooyaiLangChainWrapperParams {
     maxTokens?: number;
     supplyDataFunctions?: ISupplyDataFunctions;
     callbacks?: any[];
+    verbose?: boolean;
 }
 export declare class RooyaiLangChainWrapper extends BaseChatModel {
     lc_namespace: string[];
@@ -20,6 +21,7 @@ export declare class RooyaiLangChainWrapper extends BaseChatModel {
     private temperature;
     private maxTokens?;
     private supplyDataFunctions?;
+    verbose: boolean;
     constructor(params: RooyaiLangChainWrapperParams);
     _llmType(): string;
     _modelType(): string;

package/dist/nodes/Rooyai/RooyaiLangChainWrapper.js

@@ -13,6 +13,7 @@ class RooyaiLangChainWrapper extends chat_models_1.BaseChatModel {
         super({
             ...params,
             callbacks,
+            verbose: params.verbose ?? false,
         });
         this.lc_namespace = ['n8n', 'rooyai', 'chat'];
         this.apiKey = params.apiKey;
@@ -21,6 +22,7 @@ class RooyaiLangChainWrapper extends chat_models_1.BaseChatModel {
         this.temperature = params.temperature ?? 0.7;
         this.maxTokens = params.maxTokens;
         this.supplyDataFunctions = params.supplyDataFunctions;
+        this.verbose = params.verbose ?? false;
     }
     _llmType() {
         return 'rooyai';
@@ -34,6 +36,15 @@ class RooyaiLangChainWrapper extends chat_models_1.BaseChatModel {
             'Content-Type': 'application/json',
         };
         const url = `${this.baseUrl}/chat`;
+        if (this.verbose && this.supplyDataFunctions?.logger) {
+            this.supplyDataFunctions.logger.debug('📡 Calling Rooyai API', {
+                url,
+                model: body.model,
+                messageCount: body.messages.length,
+                temperature: body.temperature,
+                maxTokens: body.max_tokens,
+            });
+        }
         try {
             const response = await axios_1.default.post(url, body, {
                 headers,
@@ -45,9 +56,22 @@ class RooyaiLangChainWrapper extends chat_models_1.BaseChatModel {
                     : JSON.stringify(response.data);
                 throw new Error(`Rooyai API error (${response.status}): ${errorText}`);
             }
+            if (this.verbose && this.supplyDataFunctions?.logger) {
+                this.supplyDataFunctions.logger.debug('✅ Rooyai API response received', {
+                    status: response.status,
+                    hasChoices: !!response.data.choices,
+                    hasReply: !!response.data.reply,
+                    cost: response.data.usage?.cost_usd,
+                });
+            }
             return response.data;
         }
         catch (error) {
+            if (this.verbose && this.supplyDataFunctions?.logger) {
+                this.supplyDataFunctions.logger.error('❌ Rooyai API call failed', {
+                    error: error instanceof Error ? error.message : String(error),
+                });
+            }
             if (error instanceof Error) {
                 throw new Error(`Failed to call Rooyai API: ${error.message}`);
             }
@@ -104,6 +128,12 @@ class RooyaiLangChainWrapper extends chat_models_1.BaseChatModel {
         };
         if (response.usage?.cost_usd !== undefined) {
             llmOutput.costUsd = response.usage.cost_usd;
+            if (this.verbose && this.supplyDataFunctions?.logger) {
+                this.supplyDataFunctions.logger.info('💰 API Cost', {
+                    cost: response.usage.cost_usd,
+                    totalTokens: response.usage.total_tokens,
+                });
+            }
         }
         const generation = {
             message: aiMessage,

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "n8n-nodes-rooyai-chat",
-	"version": "0.2.0",
+	"version": "0.4.0",
 	"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",