@alpic80/rivet-core 1.24.2-aidon.5 → 1.24.2-aidon.6

package/dist/esm/plugins/aidon/nodes/ChatAidonNode.js

@@ -53,7 +53,7 @@ class ChatAidonNodeImpl extends ChatNodeImpl {
         }
         return path;
     }
-    async callToolGet(parsedArgs, schemaDetail, path, data) {
+    async callToolGet(schemaDetail, path, parsedArgs, data) {
         const queryParams = new URLSearchParams(parsedArgs.parameters).toString();
         const fullUrl = schemaDetail.url + path + (queryParams ? "?" + queryParams : "");
         let headers = {};
@@ -112,8 +112,11 @@ class ChatAidonNodeImpl extends ChatNodeImpl {
     async process(inputs, context) {
         //make sure not to include functions if we have no way to run them after.
         inputs = this.removeInvalidInputs(inputs);
+        // const configData = this.data
+        // configData.frequencyPenalty = 2;
         // Call the parent class's process method to do its job
         let outputs = await super.process(inputs, context);
+        //Now check if the LLM wants us to do some tool calling
         const funcCallOutput = outputs['function-call'] ?? outputs['function-calls'];
         const funcCalls = funcCallOutput?.type === 'object[]'
             ? funcCallOutput.value
@@ -135,14 +138,13 @@ class ChatAidonNodeImpl extends ChatNodeImpl {
                 }
                 const schemaDetail = this.convertToolSchemaToSchemaDetail(toolSchema);
                 const path = this.extractPath(schemaDetail, functionCall.name, functionCall.arguments);
-                // Determine if the request should be in the body or as a query
                 let data = {};
-                if (schemaDetail.requestInBody) {
-                    // If the type is set to body
+                // Determine if the request should be in the body or as a query
+                if (schemaDetail.requestInBody) { // If the type is set to body
                     data = await this.callToolPost(schemaDetail, path, functionCall.arguments, data);
                 }
                 else { // If the type is set to query
-                    data = await this.callToolGet(functionCall.arguments, schemaDetail, path, data);
+                    data = await this.callToolGet(schemaDetail, path, functionCall.arguments, data);
                 }
                 messages['value'].push({
                     type: "function",

package/dist/esm/plugins/huggingface/nodes/ChatHuggingFace.js

@@ -1,6 +1,6 @@
 import { nanoid } from 'nanoid/non-secure';
 import {} from '../../../index.js';
-import { HfInference, HfInferenceEndpoint } from '@huggingface/inference';
+import { InferenceClient } from '@huggingface/inference';
 import { getInputOrData } from '../../../utils/inputs.js';
 import { coerceType } from '../../../utils/coerceType.js';
 import { dedent } from '../../../utils/misc.js';
@@ -206,7 +206,9 @@ export const ChatHuggingFaceNodeImpl = {
         const repetitionPenalty = getInputOrData(data, inputData, 'repetitionPenalty', 'number');
         const topP = getInputOrData(data, inputData, 'topP', 'number');
         const topK = getInputOrData(data, inputData, 'topK', 'number');
-        const hf = endpoint ? new HfInferenceEndpoint(endpoint, accessToken) : new HfInference(accessToken);
+        const hf = endpoint
+            ? new InferenceClient(accessToken, { endpointUrl: endpoint })
+            : new InferenceClient(accessToken);
         const generationStream = hf.textGenerationStream({
             inputs: prompt,
             model,

package/dist/esm/plugins/huggingface/nodes/TextToImageHuggingFace.js

@@ -1,6 +1,6 @@
 import { nanoid } from 'nanoid/non-secure';
 import {} from '../../../index.js';
-import { HfInference, HfInferenceEndpoint } from '@huggingface/inference';
+import { InferenceClient } from '@huggingface/inference';
 import { dedent } from 'ts-dedent';
 import { pluginNodeDefinition } from '../../../model/NodeDefinition.js';
 import { getInputOrData } from '../../../utils/inputs.js';
@@ -163,7 +163,9 @@ export const TextToImageHuggingFaceNodeImpl = {
         const negativePrompt = getInputOrData(data, inputData, 'negativePrompt') || undefined;
         const guidanceScale = getInputOrData(data, inputData, 'guidanceScale', 'number');
         const numInferenceSteps = getInputOrData(data, inputData, 'numInferenceSteps', 'number');
-        const hf = endpoint ? new HfInferenceEndpoint(endpoint, accessToken) : new HfInference(accessToken);
+        const hf = endpoint
+            ? new InferenceClient(accessToken, { endpointUrl: endpoint })
+            : new InferenceClient(accessToken);
         const image = await hf.textToImage({
             inputs: prompt,
             model,
@@ -174,7 +176,7 @@ export const TextToImageHuggingFaceNodeImpl = {
                 guidance_scale: guidanceScale,
                 num_inference_steps: numInferenceSteps,
             },
-        });
+        }, { outputType: "blob" });
         return {
             ['output']: {
                 type: 'image',

package/package.json

@@ -2,7 +2,7 @@
   "name": "@alpic80/rivet-core",
   "license": "MIT",
   "repository": "https://github.com/castortech/rivet",
-  "version": "1.24.2-aidon.5",
+  "version": "1.24.2-aidon.6",
   "packageManager": "yarn@3.5.0",
   "main": "dist/cjs/bundle.cjs",
   "module": "dist/esm/index.js",
@@ -47,8 +47,8 @@
     "@gentrace/core": "^2.2.5",
     "@google-cloud/vertexai": "^0.1.3",
     "@google/genai": "^0.12.0",
-    "@huggingface/inference": "^2.6.4",
-    "@ironclad/rivet-core": "npm:@alpic80/rivet-core@1.24.2-aidon.5",
+    "@huggingface/inference": "^4.13.0",
+    "@ironclad/rivet-core": "npm:@alpic80/rivet-core@1.24.2-aidon.6",
     "assemblyai": "^4.6.0",
     "autoevals": "^0.0.26",
     "cron-parser": "^4.9.0",
@@ -56,12 +56,12 @@
     "emittery": "^1.0.1",
     "emittery-0-13": "npm:emittery@^0.13.1",
     "gpt-tokenizer": "^2.1.2",
-    "jsonpath-plus": "^10.2.0",
+    "jsonpath-plus": "^10.3.0",
     "lodash-es": "^4.17.21",
     "mdast-util-gfm-table": "^2.0.0",
     "mdast-util-to-markdown": "^2.1.2",
     "minimatch": "^9.0.3",
-    "nanoid": "^3.3.6",
+    "nanoid": "^3.3.8",
     "openai": "^4.28.4",
     "p-queue": "^7.4.1",
     "p-queue-6": "npm:p-queue@^6.0.0",
@@ -83,7 +83,7 @@
     "@types/yaml": "^1.9.7",
     "@typescript-eslint/eslint-plugin": "^8.24.0",
     "@typescript-eslint/parser": "^8.24.0",
-    "esbuild": "^0.19.5",
+    "esbuild": "^0.25.12",
     "eslint": "^9.20.1",
     "eslint-import-resolver-typescript": "^3.6.1",
     "eslint-plugin-import": "^2.31.0",