@alpic80/rivet-ai-sdk-provider 2.0.1 → 2.0.3

package/dist/index.js

@@ -1109,7 +1109,7 @@ prepare tools input:${printObject(tools)}`);
 }
 
 // src/version.ts
-var VERSION = true ? "2.0.1" : "0.0.0-test";
+var VERSION = true ? "2.0.3" : "0.0.0-test";
 
 // src/post-to-api.ts
 var getOriginalFetch = () => globalThis.fetch;
@@ -1761,7 +1761,7 @@ function createRivetEventSourceResponseHandler(id, model) {
   }));
 }
 function mapRivetEventToOpenAIChunk(eventData, id, model) {
-  var _a15, _b, _c, _d, _e, _f, _g;
+  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":
@@ -1782,6 +1782,26 @@ 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),
+        model,
+        choices: [
+          {
+            delta: {
+              role: "assistant",
+              content: delta != null ? delta : "",
+              tool_calls: void 0
+            },
+            finish_reason: null,
+            index: 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 {
         id,
@@ -1791,16 +1811,17 @@ function mapRivetEventToOpenAIChunk(eventData, id, model) {
           {
             delta: {
               role: "assistant",
-              content: (_d = (_c = (_b = eventData.graphOutput) == null ? void 0 : _b.response) == null ? void 0 : _c.value) != null ? _d : "",
+              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((_g = (_f = (_e = eventData.graphOutput.usages) == null ? void 0 : _e.value) == null ? void 0 : _f[0]) == null ? void 0 : _g.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:
+      console.log(`failling on default for type:${printObject(eventType)}`);
       return {
         id,
         created: Math.floor(Date.now() / 1e3),