return pl id when streaming (#65)

MagnivOrg · May 7, 2024 · e2fd939 · e2fd939
1 parent 77cd9a4
commit e2fd939
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 5 deletions.
diff --git a/package-lock.json b/package-lock.json
diff --git 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
@@ -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) => {