@sentrial/sdk 0.3.0 → 0.3.1

package/dist/index.cjs

@@ -1205,34 +1205,35 @@ function wrapStreamText(originalFn, client, config) {
       trackCompletion(textProp).catch(() => {
       });
     } else if (textProp != null && typeof textProp.then === "function") {
-      const originalTextPromise = textProp;
-      result.text = originalTextPromise.then((text) => {
+      textProp.then((text) => {
         trackCompletion(text).catch(() => {
         });
-        return text;
       }).catch((err) => {
         trackCompletion("", err instanceof Error ? err : new Error(String(err))).catch(() => {
         });
-        throw err;
       });
     } else {
-      const originalTextStream = result.textStream;
-      let fullText = "";
-      result.textStream = (async function* () {
-        try {
-          for await (const chunk of originalTextStream) {
-            fullText += chunk;
-            yield chunk;
+      const desc = Object.getOwnPropertyDescriptor(result, "textStream") ?? Object.getOwnPropertyDescriptor(Object.getPrototypeOf(result), "textStream");
+      const isWritable = !desc || desc.writable || desc.set;
+      if (isWritable) {
+        const originalTextStream = result.textStream;
+        let fullText = "";
+        result.textStream = (async function* () {
+          try {
+            for await (const chunk of originalTextStream) {
+              fullText += chunk;
+              yield chunk;
+            }
+            await trackCompletion(fullText);
+          } catch (error) {
+            await trackCompletion(
+              "",
+              error instanceof Error ? error : new Error(String(error))
+            );
+            throw error;
           }
-          await trackCompletion(fullText);
-        } catch (error) {
-          await trackCompletion(
-            "",
-            error instanceof Error ? error : new Error(String(error))
-          );
-          throw error;
-        }
-      })();
+        })();
+      }
     }
     return result;
   };
@@ -1321,50 +1322,46 @@ function wrapStreamObject(originalFn, client, config) {
       }
     })();
     const result = originalFn(params);
-    if (result.object) {
-      const originalObjectPromise = result.object;
-      result.object = originalObjectPromise.then(async (obj) => {
+    const objectProp = result.object;
+    if (objectProp && typeof objectProp.then === "function") {
+      objectProp.then(async (obj) => {
         const durationMs = Date.now() - startTime;
         const sid = await sessionPromise;
-        if (sid) {
-          let usage;
-          try {
-            usage = result.usage ? await result.usage : void 0;
-          } catch {
-          }
-          const promptTokens = usage?.promptTokens || 0;
-          const completionTokens = usage?.completionTokens || 0;
-          const totalTokens = usage?.totalTokens || promptTokens + completionTokens;
-          const cost = calculateCostForCall(provider, modelId, promptTokens, completionTokens);
-          await client.completeSession({
-            sessionId: sid,
-            success: true,
-            output: JSON.stringify(obj),
-            durationMs,
-            estimatedCost: cost,
-            promptTokens,
-            completionTokens,
-            totalTokens
-          });
+        if (!sid) return;
+        let usage;
+        try {
+          usage = result.usage ? await result.usage : void 0;
+        } catch {
         }
-        return obj;
+        const promptTokens = usage?.promptTokens || 0;
+        const completionTokens = usage?.completionTokens || 0;
+        const totalTokens = usage?.totalTokens || promptTokens + completionTokens;
+        const cost = calculateCostForCall(provider, modelId, promptTokens, completionTokens);
+        await client.completeSession({
+          sessionId: sid,
+          success: true,
+          output: JSON.stringify(obj),
+          durationMs,
+          estimatedCost: cost,
+          promptTokens,
+          completionTokens,
+          totalTokens
+        });
       }).catch(async (error) => {
         const durationMs = Date.now() - startTime;
         const sid = await sessionPromise;
-        if (sid) {
-          await client.trackError({
-            sessionId: sid,
-            errorType: error instanceof Error ? error.name : "Error",
-            errorMessage: error instanceof Error ? error.message : "Unknown error"
-          });
-          await client.completeSession({
-            sessionId: sid,
-            success: false,
-            failureReason: error instanceof Error ? error.message : "Unknown error",
-            durationMs
-          });
-        }
-        throw error;
+        if (!sid) return;
+        await client.trackError({
+          sessionId: sid,
+          errorType: error instanceof Error ? error.name : "Error",
+          errorMessage: error instanceof Error ? error.message : "Unknown error"
+        });
+        await client.completeSession({
+          sessionId: sid,
+          success: false,
+          failureReason: error instanceof Error ? error.message : "Unknown error",
+          durationMs
+        });
       });
     }
     return result;