llmasaservice-client 0.2.2 → 0.2.4

package/CHANGELOG.md

@@ -1,5 +1,17 @@
 # llmasaservice-client
 
+## 0.2.4
+
+### Patch Changes
+
+- new callback signature
+
+## 0.2.3
+
+### Patch Changes
+
+- Fixed callback
+
 ## 0.2.2
 
 ### Patch Changes

package/dist/index.js

@@ -161,9 +161,6 @@ var useLLM = (options) => {
           }
           if (done) {
             setIdle(true);
-            if (onCompleteCallback) {
-              onCompleteCallback(result);
-            }
             break;
           }
           result += decoder.decode(value);
@@ -185,6 +182,9 @@ var useLLM = (options) => {
         reader.cancel();
         setIdle(true);
       }
+      if (onCompleteCallback) {
+        onCompleteCallback();
+      }
       return result;
     });
   }

package/dist/index.mjs

@@ -124,9 +124,6 @@ var useLLM = (options) => {
           }
           if (done) {
             setIdle(true);
-            if (onCompleteCallback) {
-              onCompleteCallback(result);
-            }
             break;
           }
           result += decoder.decode(value);
@@ -148,6 +145,9 @@ var useLLM = (options) => {
         reader.cancel();
         setIdle(true);
       }
+      if (onCompleteCallback) {
+        onCompleteCallback();
+      }
       return result;
     });
   }

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "llmasaservice-client",
   "license": "MIT",
-  "version": "0.2.2",
+  "version": "0.2.4",
   "main": "dist/index.js",
   "module": "dist/index.mjs",
   "types": "dist/index.d.ts",

package/src/useLLM.ts

@@ -46,6 +46,7 @@ export const useLLM = (options?: LLMServiceType): UseLLMReturnType => {
    * @param stream  Determines whether to stream results back in the response property as they return from the service or batch them up and return them all at once in the response property as a string.
    * @param abortController The AbortController used to abort this request once its started. This allows you to add a stop button to your UI.
    * @param service The service to use for the request. If null, load balancing will be applied. This is typically only used for testing.
+   * @param onCompleteCallback The callback function to be called once the stream completes, with the final result string.
    * @returns a StreamReader object if stream is true, otherwise a string of the response. Typically this isn't used when streaming, the stream is exposed in the response property.
    */
   async function send(
@@ -54,7 +55,7 @@ export const useLLM = (options?: LLMServiceType): UseLLMReturnType => {
     stream: boolean = true,
     abortController: AbortController = new AbortController(),
     service: string | null = null, // null means use the default service and apply services load balancing
-    onCompleteCallback?: Function
+    onCompleteCallback?: (result: string) => void
   ): Promise<ReadableStreamDefaultReader<any> | string | undefined> {
     setResponse("");
     setIdle(false);
@@ -144,12 +145,6 @@ export const useLLM = (options?: LLMServiceType): UseLLMReturnType => {
         // If the stream has been read to the end, exit the loop
         if (done) {
           setIdle(true);
-
-          if (onCompleteCallback) {
-            onCompleteCallback(result);
-            
-          }
-
           break;
         }
 
@@ -176,6 +171,10 @@ export const useLLM = (options?: LLMServiceType): UseLLMReturnType => {
       setIdle(true);
     }
 
+    if (onCompleteCallback) {
+      onCompleteCallback();
+    }
+
     return result;
   }