From af9bf0980ceaecae76014f80f110f8c839b5316e Mon Sep 17 00:00:00 2001
From: jelveh <nj@puter.com>
Date: Sat, 5 Jul 2025 11:36:29 -0700
Subject: [PATCH] Add puter.ai streaming test

---
 src/puter-js/src/modules/PuterDialog.js |  1 -
 src/puter-js/test/ai.test.js            | 64 +++++++++++++++++++++++++
 src/puter-js/test/run.html              |  2 +-
 3 files changed, 65 insertions(+), 2 deletions(-)

diff --git a/src/puter-js/src/modules/PuterDialog.js b/src/puter-js/src/modules/PuterDialog.js
index 5bc9d993..5203fccb 100644
--- a/src/puter-js/src/modules/PuterDialog.js
+++ b/src/puter-js/src/modules/PuterDialog.js
@@ -456,7 +456,6 @@ class PuterDialog extends HTMLElement {
     }
 
     open() {
-        console.log(this.hasUserActivation());
         if(this.hasUserActivation()){
             let w = 600;
             let h = 400;
diff --git a/src/puter-js/test/ai.test.js b/src/puter-js/test/ai.test.js
index 8b91d6d7..44440ff6 100644
--- a/src/puter-js/test/ai.test.js
+++ b/src/puter-js/test/ai.test.js
@@ -85,6 +85,57 @@ const testChatWithMessageArrayCore = async function(model) {
     assert(result.message.content.length > 0, "message content should not be empty");
 };
 
+const testChatStreamingCore = async function(model) {
+    // Test chat with streaming enabled
+    const result = await puter.ai.chat("Count from 1 to 5", { 
+        model: model,
+        stream: true,
+        max_tokens: 100
+    });
+    
+    // Check that result is an object and not null
+    assert(typeof result === 'object', "streaming chat should return an object");
+    assert(result !== null, "streaming chat should not return null");
+    
+    // For streaming, we need to check if it's an async iterator or has a different structure
+    // The exact structure depends on the implementation, but we should verify it's consumable
+    if (result[Symbol.asyncIterator]) {
+        // If it's an async iterator, test that we can consume it
+        let chunks = [];
+        let chunkCount = 0;
+        const maxChunks = 10; // Limit to prevent infinite loops in tests
+        
+        for await (const chunk of result) {
+            chunks.push(chunk);
+            chunkCount++;
+            
+            // Verify each chunk has expected structure
+            assert(typeof chunk === 'object', "each streaming chunk should be an object");
+            
+            // Break after reasonable number of chunks for testing
+            if (chunkCount >= maxChunks) break;
+        }
+        
+        assert(chunks.length > 0, "streaming should produce at least one chunk");
+        
+    } else {
+        // If not an async iterator, it might be a different streaming implementation
+        // Check for common streaming response patterns
+        
+        // Check basic result structure (similar to non-streaming but may have different properties)
+        assert(typeof result.message === 'object' || typeof result.content === 'string', 
+            "streaming result should have message object or content string");
+        
+        // Check that it has streaming-specific properties
+        assert(typeof result.stream === 'boolean' || result.stream === true, 
+            "streaming result should indicate it's a stream");
+        
+        // Check that toString() and valueOf() methods exist and work
+        assert(typeof result.toString === 'function', "streaming result should have toString method");
+        assert(typeof result.valueOf === 'function', "streaming result should have valueOf method");
+    }
+};
+
 // Function to generate test functions for a specific model
 const generateTestsForModel = function(model) {
     const modelName = model.replace(/[^a-zA-Z0-9]/g, '_'); // Sanitize model name for function names
@@ -128,6 +179,19 @@ const generateTestsForModel = function(model) {
                 }
             }
         },
+        
+        [`testChatStreaming_${modelName}`]: {
+            name: `testChatStreaming_${modelName}`,
+            description: `Test AI chat with streaming enabled using ${model} model`,
+            test: async function() {
+                try {
+                    await testChatStreamingCore(model);
+                    pass(`testChatStreaming_${modelName} passed`);
+                } catch (error) {
+                    fail(`testChatStreaming_${modelName} failed:`, error);
+                }
+            }
+        },
     };
 };
 
diff --git a/src/puter-js/test/run.html b/src/puter-js/test/run.html
index c485a31f..beca0a18 100644
--- a/src/puter-js/test/run.html
+++ b/src/puter-js/test/run.html
@@ -1,7 +1,7 @@
 <html>
 <head>
     <script src="https://ajax.googleapis.com/ajax/libs/jquery/3.5.1/jquery.min.js"></script>
-    <script src="http://puter.localhost:4100/sdk/puter.dev.js"></script>
+    <script src="https://js.puter.com/v2/"></script>
     <script src="./kv.test.js"></script>
     <script src="./fs.test.js"></script>
     <script src="./ai.test.js"></script>