tor-browser

The Tor Browser
git clone https://git.dasho.dev/tor-browser.git
Log | Files | Refs | README | LICENSE

language-model-prompt-streaming.tentative.https.window.js (805B)


      1 // META: title=Language Model Prompt Streaming
      2 // META: script=/resources/testdriver.js
      3 // META: script=../resources/util.js
      4 // META: timeout=long
      5 
      6 'use strict';
      7 
      8 promise_test(async t => {
      9  await ensureLanguageModel();
     10 
     11  // Start a new session.
     12  const session = await createLanguageModel();
     13  // Test the streaming prompt API.
     14  const streamingResponse =
     15    session.promptStreaming(kTestPrompt);
     16  assert_equals(
     17    Object.prototype.toString.call(streamingResponse),
     18    "[object ReadableStream]"
     19  );
     20  const reader = streamingResponse.getReader();
     21  let result = "";
     22  while (true) {
     23    const { value, done } = await reader.read();
     24    if (done) {
     25      break;
     26    }
     27    if (value) {
     28      result += value;
     29    }
     30  }
     31  assert_greater_than(result.length, 0, "The result should not be empty.");
     32 });