diff --git a/package-lock.json b/package-lock.json
index a0f8489..99c9dd3 100644
--- a/package-lock.json
+++ b/package-lock.json
@@ -1,12 +1,12 @@
 {
   "name": "promptlayer",
-  "version": "0.0.19",
+  "version": "0.0.20",
   "lockfileVersion": 3,
   "requires": true,
   "packages": {
     "": {
       "name": "promptlayer",
-      "version": "0.0.19",
+      "version": "0.0.20",
       "license": "MIT",
       "devDependencies": {
         "@anthropic-ai/sdk": "^0.20.8",
diff --git a/package.json b/package.json
index 0f79236..3a9499d 100644
--- a/package.json
+++ b/package.json
@@ -1,7 +1,7 @@
 {
   "name": "promptlayer",
   "license": "MIT",
-  "version": "0.0.19",
+  "version": "0.0.20",
   "main": "dist/index.js",
   "module": "dist/index.esm.js",
   "types": "dist/index.d.ts",
diff --git a/src/utils.ts b/src/utils.ts
index f640d28..949396d 100644
--- a/src/utils.ts
+++ b/src/utils.ts
@@ -579,15 +579,22 @@ async function* proxyGenerator<Item>(
 ) {
   const results = [];
   for await (const value of generator) {
-    yield value;
+    yield body.return_pl_id ? [value, null] : value;
     results.push(value);
   }
   const request_response = cleaned_result(results, body.function_name);
-  await promptLayerApiRequest({
+  const response = await promptLayerApiRequest({
     ...body,
     request_response,
     request_end_time: new Date().toISOString(),
   });
+  if (response) {
+    if (body.return_pl_id) {
+      const request_id = (response as any)[1];
+      const lastResult = results.at(-1);
+      yield [lastResult, request_id];
+    }
+  }
 }
 
 const warnOnBadResponse = (request_response: any, main_message: string) => {