llmasaservice-client 0.1.2 → 0.2.0

package/CHANGELOG.md

@@ -1,5 +1,11 @@
 # llmasaservice-client
 
+## 0.2.0
+
+### Minor Changes
+
+- Added onCompleteCallback to send
+
 ## 0.1.2
 
 ### Patch Changes

package/package.json

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

package/src/useLLM.ts

@@ -53,7 +53,8 @@ export const useLLM = (options?: LLMServiceType): UseLLMReturnType => {
     messages = [],
     stream: boolean = true,
     abortController: AbortController = new AbortController(),
-    service: string | null = null // null means use the default service and apply services load balancing
+    service: string | null = null, // null means use the default service and apply services load balancing
+    onCompleteCallback?: Function
   ): Promise<ReadableStreamDefaultReader<any> | string | undefined> {
     setResponse("");
     setIdle(false);
@@ -98,7 +99,7 @@ export const useLLM = (options?: LLMServiceType): UseLLMReturnType => {
         } else {
           readStream(reader, decoder, stream, {
             signal: options.signal,
-          });
+          }, onCompleteCallback);
 
           return reader;
         }
@@ -117,7 +118,8 @@ export const useLLM = (options?: LLMServiceType): UseLLMReturnType => {
     reader: ReadableStreamDefaultReader,
     decoder: TextDecoder,
     stream: Boolean = true,
-    { signal: signal }: { signal: AbortSignal }
+    { signal: signal }: { signal: AbortSignal },
+    onCompleteCallback?: Function
   ): Promise<string> {
     let errorInRead = "";
     let result = "";
@@ -142,6 +144,11 @@ 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;
         }