@yourgpt/llm-sdk 0.1.1 → 1.0.0

package/dist/index.mjs

@@ -3430,8 +3430,28 @@ var Runtime = class {
   }
   /**
    * Handle HTTP request (for use with any framework)
+   *
+   * @param request - The HTTP request
+   * @param options - Optional configuration including onFinish callback for persistence
+   *
+   * @example
+   * ```typescript
+   * // Basic usage
+   * return runtime.handleRequest(request);
+   *
+   * // With server-side persistence
+   * return runtime.handleRequest(request, {
+   *   onFinish: async ({ messages, threadId }) => {
+   *     await db.thread.upsert({
+   *       where: { id: threadId },
+   *       update: { messages, updatedAt: new Date() },
+   *       create: { id: threadId, messages },
+   *     });
+   *   },
+   * });
+   * ```
    */
-  async handleRequest(request) {
+  async handleRequest(request, options) {
     try {
       const body = await request.json();
       if (this.config.debug) {
@@ -3445,11 +3465,17 @@ var Runtime = class {
           body,
           signal,
           useAgentLoop || false,
-          request
+          request,
+          options
         );
       }
       const generator = useAgentLoop ? this.processChatWithLoop(body, signal, void 0, void 0, request) : this.processChat(body, signal);
-      return createSSEResponse(generator);
+      const wrappedGenerator = this.wrapGeneratorWithOnFinish(
+        generator,
+        body.threadId,
+        options
+      );
+      return createSSEResponse(wrappedGenerator);
     } catch (error) {
       console.error("[Copilot SDK] Error:", error);
       return new Response(
@@ -3463,10 +3489,34 @@ var Runtime = class {
       );
     }
   }
+  /**
+   * Wrap a generator to intercept the done event and call onFinish
+   */
+  async *wrapGeneratorWithOnFinish(generator, threadId, options) {
+    let doneMessages;
+    for await (const event of generator) {
+      if (event.type === "done" && event.messages) {
+        doneMessages = event.messages;
+      }
+      yield event;
+    }
+    if (options?.onFinish && doneMessages) {
+      try {
+        const result = {
+          messages: doneMessages,
+          threadId
+          // TODO: Add usage tracking when available from adapter
+        };
+        await options.onFinish(result);
+      } catch (error) {
+        console.error("[Copilot SDK] onFinish callback error:", error);
+      }
+    }
+  }
   /**
    * Handle non-streaming request - returns JSON instead of SSE
    */
-  async handleNonStreamingRequest(body, signal, useAgentLoop, httpRequest) {
+  async handleNonStreamingRequest(body, signal, useAgentLoop, httpRequest, options) {
     try {
       const generator = useAgentLoop ? this.processChatWithLoop(
         body,
@@ -3518,6 +3568,20 @@ var Runtime = class {
             break;
         }
       }
+      if (options?.onFinish && messages && !error) {
+        try {
+          const result = {
+            messages,
+            threadId: body.threadId
+          };
+          await options.onFinish(result);
+        } catch (callbackError) {
+          console.error(
+            "[Copilot SDK] onFinish callback error:",
+            callbackError
+          );
+        }
+      }
       const response = {
         success: !error,
         content,