@alpic80/rivet-ai-sdk-provider 2.0.4 → 2.0.6

package/dist/index.d.mts

@@ -3,6 +3,9 @@ import { FetchFunction } from '@ai-sdk/provider-utils';
 import { z } from 'zod/v4';
 
 type RivetChatModelId = string;
+type JSONValue = null | string | number | boolean | {
+    [key: string]: JSONValue;
+} | JSONValue[];
 declare const rivetLanguageModelOptions: z.ZodObject<{
     runParams: z.ZodObject<{
         openaiApiKey: z.ZodOptional<z.ZodString>;
@@ -53,6 +56,10 @@ declare const rivetLanguageModelOptions: z.ZodObject<{
             pcm16: "pcm16";
         }>>;
     }, z.core.$strip>;
+    rivetInputs: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodObject<{
+        type: z.ZodString;
+        value: z.ZodType<JSONValue, unknown, z.core.$ZodTypeInternals<JSONValue, unknown>>;
+    }, z.core.$strip>>>;
 }, z.core.$strip>;
 type RivetLanguageModelOptions = z.infer<typeof rivetLanguageModelOptions>;
 

package/dist/index.d.ts

@@ -3,6 +3,9 @@ import { FetchFunction } from '@ai-sdk/provider-utils';
 import { z } from 'zod/v4';
 
 type RivetChatModelId = string;
+type JSONValue = null | string | number | boolean | {
+    [key: string]: JSONValue;
+} | JSONValue[];
 declare const rivetLanguageModelOptions: z.ZodObject<{
     runParams: z.ZodObject<{
         openaiApiKey: z.ZodOptional<z.ZodString>;
@@ -53,6 +56,10 @@ declare const rivetLanguageModelOptions: z.ZodObject<{
             pcm16: "pcm16";
         }>>;
     }, z.core.$strip>;
+    rivetInputs: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodObject<{
+        type: z.ZodString;
+        value: z.ZodType<JSONValue, unknown, z.core.$ZodTypeInternals<JSONValue, unknown>>;
+    }, z.core.$strip>>>;
 }, z.core.$strip>;
 type RivetLanguageModelOptions = z.infer<typeof rivetLanguageModelOptions>;
 

package/dist/index.js

@@ -931,6 +931,16 @@ function mapRivetFinishReason(finishReason) {
 
 // src/rivet-chat-options.ts
 var import_v4 = require("zod/v4");
+var jsonValue = import_v4.z.lazy(
+  () => import_v4.z.union([
+    import_v4.z.string(),
+    import_v4.z.number(),
+    import_v4.z.boolean(),
+    import_v4.z.null(),
+    import_v4.z.array(jsonValue),
+    import_v4.z.record(import_v4.z.string(), jsonValue)
+  ])
+);
 var rivetRunParamsSchema = import_v4.z.object({
   openaiApiKey: import_v4.z.string().optional(),
   openaiEndpoint: import_v4.z.string().optional(),
@@ -942,6 +952,14 @@ var rivetRunParamsSchema = import_v4.z.object({
   streamNode: import_v4.z.string().optional(),
   events: import_v4.z.string().optional()
 });
+var graphInputSchema = import_v4.z.record(
+  import_v4.z.string(),
+  // key type
+  import_v4.z.object({
+    type: import_v4.z.string(),
+    value: jsonValue
+  })
+).optional();
 var standardChatConfigSchema = import_v4.z.object({
   temperature: import_v4.z.number().optional(),
   top_p: import_v4.z.number().optional(),
@@ -989,7 +1007,8 @@ var rivetCustomChatConfigSchema = import_v4.z.object({
 });
 var rivetLanguageModelOptions = import_v4.z.object({
   runParams: rivetRunParamsSchema,
-  chatConfig: rivetCustomChatConfigSchema
+  chatConfig: rivetCustomChatConfigSchema,
+  rivetInputs: graphInputSchema
 });
 var rivetChatConfigSchema = standardChatConfigSchema.merge(rivetCustomChatConfigSchema);
 var rivetToolOptions = import_v4.z.record(
@@ -1115,7 +1134,7 @@ prepare tools input:${printObject(tools)}`);
 }
 
 // src/version.ts
-var VERSION = true ? "2.0.4" : "0.0.0-test";
+var VERSION = true ? "2.0.6" : "0.0.0-test";
 
 // src/post-to-api.ts
 var getOriginalFetch = () => globalThis.fetch;
@@ -1323,11 +1342,12 @@ var RivetChatLanguageModel = class {
     toolChoice
     //in Rivet tool section
   }) {
-    var _a15, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k;
+    var _a15, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m;
     const warnings = [];
     const emptyOptions = {
       runParams: {},
-      chatConfig: {}
+      chatConfig: {},
+      rivetInputs: {}
     };
     debugLog(`providerOptions:${printObject(providerOptions)}`);
     const options = (_a15 = await parseProviderOptions({
@@ -1344,15 +1364,16 @@ var RivetChatLanguageModel = class {
     } = await prepareTools({ tools, toolChoice });
     options.runParams = (_b = options.runParams) != null ? _b : {};
     options.chatConfig = (_c = options.chatConfig) != null ? _c : {};
+    options.rivetInputs = (_d = options.rivetInputs) != null ? _d : {};
     const chatConfig = __spreadValues({}, options.chatConfig);
-    chatConfig.maxTokens = (_d = chatConfig.maxTokens) != null ? _d : maxOutputTokens;
-    chatConfig.temperature = (_e = chatConfig.temperature) != null ? _e : temperature;
-    chatConfig.top_p = (_f = chatConfig.top_p) != null ? _f : topP;
-    chatConfig.top_k = (_g = chatConfig.top_k) != null ? _g : topK;
-    chatConfig.stop_sequences = (_h = chatConfig.stop_sequences) != null ? _h : stopSequences;
-    chatConfig.frequencyPenalty = (_i = chatConfig.frequencyPenalty) != null ? _i : frequencyPenalty;
-    chatConfig.presencePenalty = (_j = chatConfig.presencePenalty) != null ? _j : presencePenalty;
-    chatConfig.seed = (_k = chatConfig.seed) != null ? _k : seed;
+    chatConfig.maxTokens = (_e = chatConfig.maxTokens) != null ? _e : maxOutputTokens;
+    chatConfig.temperature = (_f = chatConfig.temperature) != null ? _f : temperature;
+    chatConfig.top_p = (_g = chatConfig.top_p) != null ? _g : topP;
+    chatConfig.top_k = (_h = chatConfig.top_k) != null ? _h : topK;
+    chatConfig.stop_sequences = (_i = chatConfig.stop_sequences) != null ? _i : stopSequences;
+    chatConfig.frequencyPenalty = (_j = chatConfig.frequencyPenalty) != null ? _j : frequencyPenalty;
+    chatConfig.presencePenalty = (_k = chatConfig.presencePenalty) != null ? _k : presencePenalty;
+    chatConfig.seed = (_l = chatConfig.seed) != null ? _l : seed;
     if (responseFormat) {
       const isJson = (responseFormat == null ? void 0 : responseFormat.type) === "json";
       const hasSchema = isJson && !!(responseFormat == null ? void 0 : responseFormat.schema);
@@ -1365,7 +1386,7 @@ var RivetChatLanguageModel = class {
     }
     const messages = convertToRivetChatMessages(prompt);
     return {
-      args: {
+      args: __spreadValues({
         chatConfig: {
           type: "object",
           value: chatConfig
@@ -1384,7 +1405,7 @@ var RivetChatLanguageModel = class {
           value: rivetSchemas
         },
         runParams: __spreadValues({}, options.runParams)
-      },
+      }, (_m = options.rivetInputs) != null ? _m : {}),
       warnings: [...warnings, ...toolWarnings]
     };
   }
@@ -1704,8 +1725,8 @@ var rivetChatResponseSchema = import_v43.z.object({
   usage: rivetUsageSchema
 });
 var rivetChatResponseHandler = (id, model) => async ({
-  url,
-  requestBodyValues,
+  url: _url,
+  requestBodyValues: _requestBodyValues,
   response
 }) => {
   const raw = await response.json();