modelfusion 0.113.0 → 0.114.1

package/model-provider/stability/StabilityImageGenerationModel.d.ts

@@ -1,5 +1,5 @@
 import { z } from "zod";
-import { FunctionOptions } from "../../core/FunctionOptions.js";
+import { FunctionCallOptions } from "../../core/FunctionOptions.js";
 import { ApiConfiguration } from "../../core/api/ApiConfiguration.js";
 import { AbstractModel } from "../../model-function/AbstractModel.js";
 import { PromptTemplate } from "../../model-function/PromptTemplate.js";
@@ -64,9 +64,9 @@ export declare class StabilityImageGenerationModel extends AbstractModel<Stabili
     constructor(settings: StabilityImageGenerationSettings);
     readonly provider: "stability";
     get modelName(): StabilityImageGenerationModelType;
-    callAPI(input: StabilityImageGenerationPrompt, options?: FunctionOptions): Promise<StabilityImageGenerationResponse>;
+    callAPI(input: StabilityImageGenerationPrompt, callOptions: FunctionCallOptions): Promise<StabilityImageGenerationResponse>;
     get settingsForEvent(): Partial<StabilityImageGenerationSettings>;
-    doGenerateImages(prompt: StabilityImageGenerationPrompt, options?: FunctionOptions): Promise<{
+    doGenerateImages(prompt: StabilityImageGenerationPrompt, callOptions: FunctionCallOptions): Promise<{
         response: {
             artifacts: {
                 base64: string;

package/model-provider/stability/StabilityImageGenerationModel.js

@@ -45,15 +45,20 @@ export class StabilityImageGenerationModel extends AbstractModel {
     get modelName() {
         return this.settings.model;
     }
-    async callAPI(input, options) {
+    async callAPI(input, callOptions) {
         const api = this.settings.api ?? new StabilityApiConfiguration();
-        const abortSignal = options?.run?.abortSignal;
+        const abortSignal = callOptions.run?.abortSignal;
         return callWithRetryAndThrottle({
             retry: this.settings.api?.retry,
             throttle: this.settings.api?.throttle,
             call: async () => postJsonToApi({
                 url: api.assembleUrl(`/generation/${this.settings.model}/text-to-image`),
-                headers: api.headers,
+                headers: api.headers({
+                    functionType: callOptions.functionType,
+                    functionId: callOptions.functionId,
+                    run: callOptions.run,
+                    callId: callOptions.callId,
+                }),
                 body: {
                     height: this.settings.height,
                     width: this.settings.width,
@@ -85,8 +90,8 @@ export class StabilityImageGenerationModel extends AbstractModel {
             stylePreset: this.settings.stylePreset,
         };
     }
-    async doGenerateImages(prompt, options) {
-        const response = await this.callAPI(prompt, options);
+    async doGenerateImages(prompt, callOptions) {
+        const response = await this.callAPI(prompt, callOptions);
         return {
             response,
             base64Images: response.artifacts.map((artifact) => artifact.base64),

package/model-provider/whispercpp/WhisperCppTranscriptionModel.cjs

@@ -26,19 +26,16 @@ class WhisperCppTranscriptionModel extends AbstractModel_js_1.AbstractModel {
         });
     }
     async doTranscribe(data, options) {
-        const response = await this.callAPI(data, {
-            functionId: options?.functionId,
-            run: options?.run,
-        });
+        const response = await this.callAPI(data, options);
         return {
             response,
             transcription: response.text,
         };
     }
-    async callAPI(data, options) {
+    async callAPI(data, callOptions) {
         const { temperature } = this.settings;
         const api = this.settings.api ?? new WhisperCppApiConfiguration_js_1.WhisperCppApiConfiguration();
-        const abortSignal = options.run?.abortSignal;
+        const abortSignal = callOptions.run?.abortSignal;
         return (0, callWithRetryAndThrottle_js_1.callWithRetryAndThrottle)({
             retry: api.retry,
             throttle: api.throttle,
@@ -51,7 +48,12 @@ class WhisperCppTranscriptionModel extends AbstractModel_js_1.AbstractModel {
                 }
                 return (0, postToApi_js_1.postToApi)({
                     url: api.assembleUrl("/inference"),
-                    headers: api.headers,
+                    headers: api.headers({
+                        functionType: callOptions.functionType,
+                        functionId: callOptions.functionId,
+                        run: callOptions.run,
+                        callId: callOptions.callId,
+                    }),
                     body: {
                         content: formData,
                         values: { temperature },

package/model-provider/whispercpp/WhisperCppTranscriptionModel.d.ts

@@ -1,5 +1,5 @@
 /// <reference types="node" />
-import { FunctionOptions } from "../../core/FunctionOptions.js";
+import { FunctionCallOptions } from "../../core/FunctionOptions.js";
 import { ApiConfiguration } from "../../core/api/ApiConfiguration.js";
 import { AbstractModel } from "../../model-function/AbstractModel.js";
 import { TranscriptionModel, TranscriptionModelSettings } from "../../model-function/generate-transcription/TranscriptionModel.js";
@@ -15,13 +15,13 @@ export declare class WhisperCppTranscriptionModel extends AbstractModel<WhisperC
     constructor(settings: WhisperCppTranscriptionModelSettings);
     readonly provider: "whispercpp";
     readonly modelName: null;
-    doTranscribe(data: WhisperCppTranscriptionInput, options?: FunctionOptions): Promise<{
+    doTranscribe(data: WhisperCppTranscriptionInput, options: FunctionCallOptions): Promise<{
         response: {
             text: string;
         };
         transcription: string;
     }>;
-    callAPI(data: WhisperCppTranscriptionInput, options: FunctionOptions): Promise<{
+    callAPI(data: WhisperCppTranscriptionInput, callOptions: FunctionCallOptions): Promise<{
         text: string;
     }>;
     get settingsForEvent(): Partial<WhisperCppTranscriptionModelSettings>;

package/model-provider/whispercpp/WhisperCppTranscriptionModel.js

@@ -23,19 +23,16 @@ export class WhisperCppTranscriptionModel extends AbstractModel {
         });
     }
     async doTranscribe(data, options) {
-        const response = await this.callAPI(data, {
-            functionId: options?.functionId,
-            run: options?.run,
-        });
+        const response = await this.callAPI(data, options);
         return {
             response,
             transcription: response.text,
         };
     }
-    async callAPI(data, options) {
+    async callAPI(data, callOptions) {
         const { temperature } = this.settings;
         const api = this.settings.api ?? new WhisperCppApiConfiguration();
-        const abortSignal = options.run?.abortSignal;
+        const abortSignal = callOptions.run?.abortSignal;
         return callWithRetryAndThrottle({
             retry: api.retry,
             throttle: api.throttle,
@@ -48,7 +45,12 @@ export class WhisperCppTranscriptionModel extends AbstractModel {
                 }
                 return postToApi({
                     url: api.assembleUrl("/inference"),
-                    headers: api.headers,
+                    headers: api.headers({
+                        functionType: callOptions.functionType,
+                        functionId: callOptions.functionId,
+                        run: callOptions.run,
+                        callId: callOptions.callId,
+                    }),
                     body: {
                         content: formData,
                         values: { temperature },

package/observability/helicone/HeliconeOpenAIApiConfiguration.cjs

@@ -4,7 +4,7 @@ exports.HeliconeOpenAIApiConfiguration = void 0;
 const BaseUrlApiConfiguration_js_1 = require("../../core/api/BaseUrlApiConfiguration.cjs");
 const loadApiKey_js_1 = require("../../core/api/loadApiKey.cjs");
 class HeliconeOpenAIApiConfiguration extends BaseUrlApiConfiguration_js_1.BaseUrlApiConfiguration {
-    constructor({ baseUrl = "https://oai.hconeai.com/v1", openAIApiKey, heliconeApiKey, retry, throttle, } = {}) {
+    constructor({ baseUrl = "https://oai.hconeai.com/v1", openAIApiKey, heliconeApiKey, retry, throttle, customCallHeaders, } = {}) {
         super({
             baseUrl,
             headers: {
@@ -23,6 +23,7 @@ class HeliconeOpenAIApiConfiguration extends BaseUrlApiConfiguration_js_1.BaseUr
             },
             retry,
             throttle,
+            customCallHeaders,
         });
     }
 }

package/observability/helicone/HeliconeOpenAIApiConfiguration.d.ts

@@ -1,12 +1,14 @@
 import { BaseUrlApiConfiguration } from "../../core/api/BaseUrlApiConfiguration.js";
+import { CustomHeaderProvider } from "../../core/api/CustomHeaderProvider.js";
 import { RetryFunction } from "../../core/api/RetryFunction.js";
 import { ThrottleFunction } from "../../core/api/ThrottleFunction.js";
 export declare class HeliconeOpenAIApiConfiguration extends BaseUrlApiConfiguration {
-    constructor({ baseUrl, openAIApiKey, heliconeApiKey, retry, throttle, }?: {
+    constructor({ baseUrl, openAIApiKey, heliconeApiKey, retry, throttle, customCallHeaders, }?: {
         baseUrl?: string;
         openAIApiKey?: string;
         heliconeApiKey?: string;
         retry?: RetryFunction;
         throttle?: ThrottleFunction;
+        customCallHeaders?: CustomHeaderProvider;
     });
 }

package/observability/helicone/HeliconeOpenAIApiConfiguration.js

@@ -1,7 +1,7 @@
 import { BaseUrlApiConfiguration } from "../../core/api/BaseUrlApiConfiguration.js";
 import { loadApiKey } from "../../core/api/loadApiKey.js";
 export class HeliconeOpenAIApiConfiguration extends BaseUrlApiConfiguration {
-    constructor({ baseUrl = "https://oai.hconeai.com/v1", openAIApiKey, heliconeApiKey, retry, throttle, } = {}) {
+    constructor({ baseUrl = "https://oai.hconeai.com/v1", openAIApiKey, heliconeApiKey, retry, throttle, customCallHeaders, } = {}) {
         super({
             baseUrl,
             headers: {
@@ -20,6 +20,7 @@ export class HeliconeOpenAIApiConfiguration extends BaseUrlApiConfiguration {
             },
             retry,
             throttle,
+            customCallHeaders,
         });
     }
 }

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "modelfusion",
   "description": "The TypeScript library for building AI applications.",
-  "version": "0.113.0",
+  "version": "0.114.1",
   "author": "Lars Grammel",
   "license": "MIT",
   "keywords": [