@atomoz/workflows-nodes 0.1.23 → 0.1.25

package/dist/index.cjs

@@ -1088,7 +1088,7 @@ async function createLLMFromModel(modelConfig, authToken, streaming = false) {
     case "gemini":
       return new import_google_gauth.ChatGoogle({
         model: "gemini-flash-latest",
-        apiKey: "AIzaSyD9WiFuXp2fhVAokIMPp9YPKMlh7_Bvddc",
+        apiKey: "AIzaSyBzrL8Hx6dHhXgwc2HfLlQsf5Y-9pdtc9M",
         streaming
       });
     case "openai":
@@ -1342,8 +1342,27 @@ IMPORTANT: You must base your response on the last message in the conversation h
   };
   const finalModel = model?.integrationId || typeof model?.bindTools === "function" ? model : defaultModel;
   console.log(`\u{1F916} IaAgentNode "${name}": Using model:`, finalModel === defaultModel ? "DEFAULT (Gemini)" : finalModel?.model || "instance");
-  const callbacks = stream && emitter ? [
-    {
+  const tokenUsage = { input: 0, output: 0, total: 0 };
+  const callbacks = [
+    {
+      handleLLMEnd: (output2) => {
+        let usage = output2.llmOutput?.tokenUsage || output2.llmOutput?.estimatedTokenUsage;
+        if (!usage && output2.generations && Array.isArray(output2.generations) && output2.generations.length > 0) {
+          const firstGen = output2.generations[0][0];
+          if (firstGen?.generationInfo?.tokenUsage) {
+            usage = firstGen.generationInfo.tokenUsage;
+          }
+        }
+        if (usage) {
+          tokenUsage.input += usage.promptTokens || usage.input_tokens || 0;
+          tokenUsage.output += usage.completionTokens || usage.output_tokens || 0;
+          tokenUsage.total += usage.totalTokens || usage.total_tokens || 0;
+        }
+      }
+    }
+  ];
+  if (stream && emitter) {
+    callbacks.push({
       handleLLMNewToken: (token) => {
         if (emitter?.emitDelta) {
           emitter.emitDelta({
@@ -1354,8 +1373,8 @@ IMPORTANT: You must base your response on the last message in the conversation h
           });
         }
       }
-    }
-  ] : [];
+    });
+  }
   let llmInstance;
   if (finalModel?.integrationId) {
     if (!authToken) {
@@ -1367,7 +1386,7 @@ IMPORTANT: You must base your response on the last message in the conversation h
   } else {
     throw new Error("Invalid model: must have integrationId or be a valid LLM instance with bindTools method");
   }
-  if (stream && callbacks.length > 0) {
+  if (callbacks.length > 0) {
     if (llmInstance.callbacks) {
       if (Array.isArray(llmInstance.callbacks)) {
         llmInstance.callbacks.push(...callbacks);
@@ -1418,6 +1437,22 @@ IMPORTANT: You must base your response on the last message in the conversation h
         if (lastMessages.length > 0) {
           const lastMessage = lastMessages[lastMessages.length - 1];
           const content = lastMessage?.content;
+          const msg = lastMessage;
+          const usageMetadata = msg?.response_metadata?.tokenUsage || msg?.response_metadata?.usage || msg?.usage_metadata;
+          if (usageMetadata) {
+            const input = usageMetadata.promptTokens || usageMetadata.input_tokens || 0;
+            const output2 = usageMetadata.completionTokens || usageMetadata.output_tokens || 0;
+            const total = usageMetadata.totalTokens || usageMetadata.total_tokens || 0;
+            if (tokenUsage.total === 0) {
+              if (input + output2 > 0) {
+                tokenUsage.input = input;
+                tokenUsage.output = output2;
+                tokenUsage.total = input + output2;
+              } else if (total > 0) {
+                tokenUsage.total = total;
+              }
+            }
+          }
           if (typeof content === "string") {
             output = content;
           } else if (Array.isArray(content)) {
@@ -1436,6 +1471,22 @@ IMPORTANT: You must base your response on the last message in the conversation h
         if (result?.messages && result.messages.length > 0) {
           const lastMessage = result.messages[result.messages.length - 1];
           const content = lastMessage?.content;
+          const msg = lastMessage;
+          const usageMetadata = msg?.response_metadata?.tokenUsage || msg?.response_metadata?.usage || msg?.usage_metadata;
+          if (usageMetadata) {
+            const input = usageMetadata.promptTokens || usageMetadata.input_tokens || 0;
+            const output2 = usageMetadata.completionTokens || usageMetadata.output_tokens || 0;
+            const total = usageMetadata.totalTokens || usageMetadata.total_tokens || 0;
+            if (tokenUsage.total === 0) {
+              if (input + output2 > 0) {
+                tokenUsage.input = input;
+                tokenUsage.output = output2;
+                tokenUsage.total = input + output2;
+              } else if (total > 0) {
+                tokenUsage.total = total;
+              }
+            }
+          }
           if (typeof content === "string") {
             output = content;
           } else if (Array.isArray(content)) {
@@ -1494,6 +1545,12 @@ ${afterResults.message}`,
       output = `Error: ${error instanceof Error ? error.message : "Unknown error"}`;
     }
   }
+  if (tokenUsage.total === 0 && (tokenUsage.input > 0 || tokenUsage.output > 0)) {
+    tokenUsage.total = tokenUsage.input + tokenUsage.output;
+  } else if (tokenUsage.total !== tokenUsage.input + tokenUsage.output) {
+    tokenUsage.total = tokenUsage.input + tokenUsage.output;
+  }
+  console.log(`[TOKEN COUNT] Input: ${tokenUsage.input}, Output: ${tokenUsage.output}, Total: ${tokenUsage.total}`);
   return {
     agent,
     response: output || "",
@@ -1920,7 +1977,7 @@ var import_langgraph_checkpoint_postgres = require("@langchain/langgraph-checkpo
 var PostgresMemoryNodeFunction = async (inputs) => {
   const { $field: _$field, $req: _$req, $inputs: _$inputs, $vars: _$vars } = inputs;
   const fieldValues = inputs.fieldValues || {};
-  const connectionString = fieldValues.connectionString || inputs.connectionString || "postgresql://postgres:postgres@localhost:5432/workflows";
+  const connectionString = fieldValues.connectionString || inputs.connectionString || "postgresql://yugabyte:yugabyte@localhost:5433/workflows";
   try {
     const checkpointer = import_langgraph_checkpoint_postgres.PostgresSaver.fromConnString(connectionString);
     await checkpointer.setup();
@@ -2444,6 +2501,19 @@ var CustomCodeNode = {
         fieldType: "any",
         required: false
       }
+    },
+    {
+      id: "function",
+      label: "Function",
+      type: "function",
+      required: false,
+      typeable: false,
+      handle: {
+        type: "output",
+        label: "Function",
+        name: "function",
+        fieldType: "function"
+      }
     }
   ]
 };

package/dist/index.d.cts

@@ -34,6 +34,8 @@ interface NodeField {
     min?: number;
     max?: number;
     handle?: NodeHandle;
+    /** Whether this field is active/visible in the UI. Defaults to true if not specified. */
+    active?: boolean;
 }
 interface NodeInput {
     id: string;

package/dist/index.d.ts

@@ -34,6 +34,8 @@ interface NodeField {
     min?: number;
     max?: number;
     handle?: NodeHandle;
+    /** Whether this field is active/visible in the UI. Defaults to true if not specified. */
+    active?: boolean;
 }
 interface NodeInput {
     id: string;

package/dist/index.js

@@ -1008,7 +1008,7 @@ async function createLLMFromModel(modelConfig, authToken, streaming = false) {
     case "gemini":
       return new ChatGoogle({
         model: "gemini-flash-latest",
-        apiKey: "AIzaSyD9WiFuXp2fhVAokIMPp9YPKMlh7_Bvddc",
+        apiKey: "AIzaSyBzrL8Hx6dHhXgwc2HfLlQsf5Y-9pdtc9M",
         streaming
       });
     case "openai":
@@ -1262,8 +1262,27 @@ IMPORTANT: You must base your response on the last message in the conversation h
   };
   const finalModel = model?.integrationId || typeof model?.bindTools === "function" ? model : defaultModel;
   console.log(`\u{1F916} IaAgentNode "${name}": Using model:`, finalModel === defaultModel ? "DEFAULT (Gemini)" : finalModel?.model || "instance");
-  const callbacks = stream && emitter ? [
-    {
+  const tokenUsage = { input: 0, output: 0, total: 0 };
+  const callbacks = [
+    {
+      handleLLMEnd: (output2) => {
+        let usage = output2.llmOutput?.tokenUsage || output2.llmOutput?.estimatedTokenUsage;
+        if (!usage && output2.generations && Array.isArray(output2.generations) && output2.generations.length > 0) {
+          const firstGen = output2.generations[0][0];
+          if (firstGen?.generationInfo?.tokenUsage) {
+            usage = firstGen.generationInfo.tokenUsage;
+          }
+        }
+        if (usage) {
+          tokenUsage.input += usage.promptTokens || usage.input_tokens || 0;
+          tokenUsage.output += usage.completionTokens || usage.output_tokens || 0;
+          tokenUsage.total += usage.totalTokens || usage.total_tokens || 0;
+        }
+      }
+    }
+  ];
+  if (stream && emitter) {
+    callbacks.push({
       handleLLMNewToken: (token) => {
         if (emitter?.emitDelta) {
           emitter.emitDelta({
@@ -1274,8 +1293,8 @@ IMPORTANT: You must base your response on the last message in the conversation h
           });
         }
       }
-    }
-  ] : [];
+    });
+  }
   let llmInstance;
   if (finalModel?.integrationId) {
     if (!authToken) {
@@ -1287,7 +1306,7 @@ IMPORTANT: You must base your response on the last message in the conversation h
   } else {
     throw new Error("Invalid model: must have integrationId or be a valid LLM instance with bindTools method");
   }
-  if (stream && callbacks.length > 0) {
+  if (callbacks.length > 0) {
     if (llmInstance.callbacks) {
       if (Array.isArray(llmInstance.callbacks)) {
         llmInstance.callbacks.push(...callbacks);
@@ -1338,6 +1357,22 @@ IMPORTANT: You must base your response on the last message in the conversation h
         if (lastMessages.length > 0) {
           const lastMessage = lastMessages[lastMessages.length - 1];
           const content = lastMessage?.content;
+          const msg = lastMessage;
+          const usageMetadata = msg?.response_metadata?.tokenUsage || msg?.response_metadata?.usage || msg?.usage_metadata;
+          if (usageMetadata) {
+            const input = usageMetadata.promptTokens || usageMetadata.input_tokens || 0;
+            const output2 = usageMetadata.completionTokens || usageMetadata.output_tokens || 0;
+            const total = usageMetadata.totalTokens || usageMetadata.total_tokens || 0;
+            if (tokenUsage.total === 0) {
+              if (input + output2 > 0) {
+                tokenUsage.input = input;
+                tokenUsage.output = output2;
+                tokenUsage.total = input + output2;
+              } else if (total > 0) {
+                tokenUsage.total = total;
+              }
+            }
+          }
           if (typeof content === "string") {
             output = content;
           } else if (Array.isArray(content)) {
@@ -1356,6 +1391,22 @@ IMPORTANT: You must base your response on the last message in the conversation h
         if (result?.messages && result.messages.length > 0) {
           const lastMessage = result.messages[result.messages.length - 1];
           const content = lastMessage?.content;
+          const msg = lastMessage;
+          const usageMetadata = msg?.response_metadata?.tokenUsage || msg?.response_metadata?.usage || msg?.usage_metadata;
+          if (usageMetadata) {
+            const input = usageMetadata.promptTokens || usageMetadata.input_tokens || 0;
+            const output2 = usageMetadata.completionTokens || usageMetadata.output_tokens || 0;
+            const total = usageMetadata.totalTokens || usageMetadata.total_tokens || 0;
+            if (tokenUsage.total === 0) {
+              if (input + output2 > 0) {
+                tokenUsage.input = input;
+                tokenUsage.output = output2;
+                tokenUsage.total = input + output2;
+              } else if (total > 0) {
+                tokenUsage.total = total;
+              }
+            }
+          }
           if (typeof content === "string") {
             output = content;
           } else if (Array.isArray(content)) {
@@ -1414,6 +1465,12 @@ ${afterResults.message}`,
       output = `Error: ${error instanceof Error ? error.message : "Unknown error"}`;
     }
   }
+  if (tokenUsage.total === 0 && (tokenUsage.input > 0 || tokenUsage.output > 0)) {
+    tokenUsage.total = tokenUsage.input + tokenUsage.output;
+  } else if (tokenUsage.total !== tokenUsage.input + tokenUsage.output) {
+    tokenUsage.total = tokenUsage.input + tokenUsage.output;
+  }
+  console.log(`[TOKEN COUNT] Input: ${tokenUsage.input}, Output: ${tokenUsage.output}, Total: ${tokenUsage.total}`);
   return {
     agent,
     response: output || "",
@@ -1840,7 +1897,7 @@ import { PostgresSaver } from "@langchain/langgraph-checkpoint-postgres";
 var PostgresMemoryNodeFunction = async (inputs) => {
   const { $field: _$field, $req: _$req, $inputs: _$inputs, $vars: _$vars } = inputs;
   const fieldValues = inputs.fieldValues || {};
-  const connectionString = fieldValues.connectionString || inputs.connectionString || "postgresql://postgres:postgres@localhost:5432/workflows";
+  const connectionString = fieldValues.connectionString || inputs.connectionString || "postgresql://yugabyte:yugabyte@localhost:5433/workflows";
   try {
     const checkpointer = PostgresSaver.fromConnString(connectionString);
     await checkpointer.setup();
@@ -2364,6 +2421,19 @@ var CustomCodeNode = {
         fieldType: "any",
         required: false
       }
+    },
+    {
+      id: "function",
+      label: "Function",
+      type: "function",
+      required: false,
+      typeable: false,
+      handle: {
+        type: "output",
+        label: "Function",
+        name: "function",
+        fieldType: "function"
+      }
     }
   ]
 };

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@atomoz/workflows-nodes",
-    "version": "0.1.23",
+    "version": "0.1.25",
     "description": "Atomoz Workflows - Node Library",
     "type": "module",
     "main": "./dist/index.js",