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

package/dist/index.js

@@ -1019,6 +1019,7 @@ var rivetFailedResponseHandler = createJsonErrorResponseHandler({
 });
 
 // src/utils.ts
+var isDebug = process.env.RIVET_AI_PROVIDER_DEBUG === "true";
 function printObject(obj) {
   const seen = /* @__PURE__ */ new WeakSet();
   return JSON.stringify(
@@ -1039,6 +1040,11 @@ function printObject(obj) {
     // Indentation level for pretty-printing
   );
 }
+var debugLog = (...args) => {
+  if (isDebug) {
+    console.log(...args);
+  }
+};
 
 // src/rivet-prepare-tools.ts
 async function prepareTools({
@@ -1051,7 +1057,7 @@ async function prepareTools({
   if (tools == null) {
     return { tools: void 0, toolSchemas: void 0, toolChoice: void 0, toolWarnings };
   }
-  console.log(`
+  debugLog(`
 
 prepare tools input:${printObject(tools)}`);
   const rivetTools = [];
@@ -1081,7 +1087,7 @@ prepare tools input:${printObject(tools)}`);
       }
     }
   }
-  console.log(`rivetSchemas:${printObject(rivetSchemas)}`);
+  debugLog(`rivetSchemas:${printObject(rivetSchemas)}`);
   if (toolChoice == null) {
     return { tools: rivetTools, toolSchemas: rivetSchemas, toolChoice: void 0, toolWarnings };
   }
@@ -1109,7 +1115,7 @@ prepare tools input:${printObject(tools)}`);
 }
 
 // src/version.ts
-var VERSION = true ? "2.0.2" : "0.0.0-test";
+var VERSION = true ? "2.0.4" : "0.0.0-test";
 
 // src/post-to-api.ts
 var getOriginalFetch = () => globalThis.fetch;
@@ -1258,7 +1264,7 @@ function parseJsonEventStream({
   return stream.pipeThrough(new TransformStream({
     transform(chunk, controller) {
       const decoded = new TextDecoder().decode(chunk);
-      console.log(`Decoded:${printObject(decoded)}`);
+      debugLog(`Decoded:${printObject(decoded)}`);
       controller.enqueue(decoded);
     }
   })).pipeThrough(new EventSourceParserStream()).pipeThrough(
@@ -1267,7 +1273,7 @@ function parseJsonEventStream({
         if (data === "[DONE]") {
           return;
         }
-        console.log(`Data to schema:${printObject(data)}`);
+        debugLog(`Data to schema:${printObject(data)}`);
         controller.enqueue(await safeParseJSON({ text: data, schema }));
       }
     })
@@ -1323,13 +1329,13 @@ var RivetChatLanguageModel = class {
       runParams: {},
       chatConfig: {}
     };
-    console.log(`providerOptions:${printObject(providerOptions)}`);
+    debugLog(`providerOptions:${printObject(providerOptions)}`);
     const options = (_a15 = await parseProviderOptions({
       provider: "rivet",
       providerOptions,
       schema: rivetLanguageModelOptions
     })) != null ? _a15 : emptyOptions;
-    console.log(`options:${printObject(options)}`);
+    debugLog(`options:${printObject(options)}`);
     const {
       tools: rivetTools,
       toolSchemas: rivetSchemas,
@@ -1454,7 +1460,7 @@ var RivetChatLanguageModel = class {
     var _a15;
     const { args, warnings } = await this.getArgs(options);
     const body = args;
-    console.log(`body:${printObject(body)}`);
+    debugLog(`body:${printObject(body)}`);
     const headers = combineHeaders(this.config.headers(), options.headers);
     let responseId = (_a15 = headers["X-Completion-Id"]) != null ? _a15 : this.generateId();
     const model = this.modelId;
@@ -1756,12 +1762,12 @@ var rivetChatChunkSchema = import_v43.z.object({
 });
 function createRivetEventSourceResponseHandler(id, model) {
   return createEventSourceResponseHandler(import_v43.z.any().transform((data) => {
-    console.log(`SSE event received:${printObject(data)}`);
+    debugLog(`SSE event received:${printObject(data)}`);
     return mapRivetEventToOpenAIChunk(data, id, model);
   }));
 }
 function mapRivetEventToOpenAIChunk(eventData, id, model) {
-  var _a15, _b, _c, _d, _e, _f, _g, _h;
+  var _a15, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m, _n, _o, _p, _q;
   const eventType = eventData.type;
   switch (eventType) {
     case "partialOutput":
@@ -1783,6 +1789,8 @@ function mapRivetEventToOpenAIChunk(eventData, id, model) {
         usage: void 0
       };
     case "nodeFinish":
+      const rawContent = (_g = (_f = (_c = (_b = eventData.outputs) == null ? void 0 : _b.response) == null ? void 0 : _c.value) != null ? _f : (_e = (_d = eventData.outputs) == null ? void 0 : _d.valueOutput) == null ? void 0 : _e.value) != null ? _g : void 0;
+      const delta = typeof rawContent === "string" ? rawContent : rawContent !== void 0 ? JSON.stringify(rawContent) : void 0;
       return {
         id,
         created: Math.floor(Date.now() / 1e3),
@@ -1791,14 +1799,14 @@ function mapRivetEventToOpenAIChunk(eventData, id, model) {
           {
             delta: {
               role: "assistant",
-              content: (_b = eventData.delta) != null ? _b : "",
+              content: delta != null ? delta : "",
               tool_calls: void 0
             },
             finish_reason: null,
             index: 0
           }
         ],
-        usage: void 0
+        usage: toOpenAIUsage((_k = (_j = (_i = (_h = eventData == null ? void 0 : eventData.output) == null ? void 0 : _h.usages) == null ? void 0 : _i.value) == null ? void 0 : _j[0]) == null ? void 0 : _k.value)
       };
     case "done":
       return {
@@ -1809,16 +1817,17 @@ function mapRivetEventToOpenAIChunk(eventData, id, model) {
           {
             delta: {
               role: "assistant",
-              content: (_e = (_d = (_c = eventData.graphOutput) == null ? void 0 : _c.response) == null ? void 0 : _d.value) != null ? _e : "",
+              content: (_n = (_m = (_l = eventData.graphOutput) == null ? void 0 : _l.response) == null ? void 0 : _m.value) != null ? _n : "",
               tool_calls: void 0
             },
             finish_reason: "stop",
             index: 0
           }
         ],
-        usage: toOpenAIUsage((_h = (_g = (_f = eventData.graphOutput.usages) == null ? void 0 : _f.value) == null ? void 0 : _g[0]) == null ? void 0 : _h.value)
+        usage: toOpenAIUsage((_q = (_p = (_o = eventData.graphOutput.usages) == null ? void 0 : _o.value) == null ? void 0 : _p[0]) == null ? void 0 : _q.value)
       };
     default:
+      debugLog(`falling on default for type:${printObject(eventType)}`);
       return {
         id,
         created: Math.floor(Date.now() / 1e3),
@@ -1839,7 +1848,7 @@ function mapRivetEventToOpenAIChunk(eventData, id, model) {
   }
 }
 var toOpenAIUsage = (usage) => {
-  console.log(`Usage to convert:${printObject(usage)}`);
+  debugLog(`Usage to convert:${printObject(usage)}`);
   return usage ? {
     prompt_tokens: usage.prompt_tokens,
     completion_tokens: usage.completion_tokens,