@sentrial/sdk 0.3.1 → 0.3.2

package/dist/index.js

@@ -1108,17 +1108,6 @@ function wrapStreamText(originalFn, client, config) {
       const completionTokens = usage?.completionTokens || 0;
       const totalTokens = usage?.totalTokens || promptTokens + completionTokens;
       const cost = calculateCostForCall(provider, modelId, promptTokens, completionTokens);
-      await client.trackEvent({
-        sessionId: sid,
-        eventType: "llm_call",
-        eventData: {
-          model: modelId,
-          provider,
-          prompt_tokens: promptTokens,
-          completion_tokens: completionTokens,
-          total_tokens: totalTokens
-        }
-      });
       await client.completeSession({
         sessionId: sid,
         success: true,
@@ -1130,6 +1119,34 @@ function wrapStreamText(originalFn, client, config) {
         totalTokens
       });
     }
+    const originalTextStream = result.textStream;
+    let wrappedTextStream = null;
+    const proxied = new Proxy(result, {
+      get(target, prop, receiver) {
+        if (prop === "textStream") {
+          if (!wrappedTextStream) {
+            let fullText = "";
+            wrappedTextStream = (async function* () {
+              try {
+                for await (const chunk of originalTextStream) {
+                  fullText += chunk;
+                  yield chunk;
+                }
+                await trackCompletion(fullText);
+              } catch (error) {
+                await trackCompletion(
+                  fullText,
+                  error instanceof Error ? error : new Error(String(error))
+                );
+                throw error;
+              }
+            })();
+          }
+          return wrappedTextStream;
+        }
+        return Reflect.get(target, prop, receiver);
+      }
+    });
     const textProp = result.text;
     if (typeof textProp === "string") {
       trackCompletion(textProp).catch(() => {
@@ -1142,30 +1159,8 @@ function wrapStreamText(originalFn, client, config) {
         trackCompletion("", err instanceof Error ? err : new Error(String(err))).catch(() => {
         });
       });
-    } else {
-      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;
-          }
-        })();
-      }
     }
-    return result;
+    return proxied;
   };
 }
 function wrapGenerateObject(originalFn, client, config) {
@@ -1252,46 +1247,64 @@ function wrapStreamObject(originalFn, client, config) {
       }
     })();
     const result = originalFn(params);
-    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) return;
-        let usage;
-        try {
-          usage = result.usage ? await result.usage : void 0;
-        } catch {
+    const originalObjectPromise = result.object;
+    if (originalObjectPromise && typeof originalObjectPromise.then === "function") {
+      let tracked = false;
+      let wrappedObjectPromise = null;
+      return new Proxy(result, {
+        get(target, prop, receiver) {
+          if (prop === "object") {
+            if (!wrappedObjectPromise) {
+              wrappedObjectPromise = originalObjectPromise.then(async (obj) => {
+                if (tracked) return obj;
+                tracked = true;
+                const durationMs = Date.now() - startTime;
+                const sid = await sessionPromise;
+                if (!sid) return obj;
+                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
+                });
+                return obj;
+              }).catch(async (error) => {
+                if (tracked) throw error;
+                tracked = true;
+                const durationMs = Date.now() - startTime;
+                const sid = await sessionPromise;
+                if (!sid) throw error;
+                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;
+              });
+            }
+            return wrappedObjectPromise;
+          }
+          return Reflect.get(target, prop, receiver);
         }
-        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) 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;