Function streamAsyncIterable

lib/client-api.ts:190–241

(stream: ReadableStream<Uint8Array>, abortSignal?: AbortSignal)

// 辅助函数：将ReadableStream转换为异步迭代器
async function* streamAsyncIterable(stream: ReadableStream<Uint8Array>, abortSignal?: AbortSignal) {
  const reader = stream.getReader();
  const decoder = new TextDecoder();
  
  try {
    while (true) {
      if (abortSignal?.aborted) {
        break;
      }
      
      const { done, value } = await reader.read();
      if (done) {
        return;
      }
      
      const chunk = decoder.decode(value, { stream: true });
      
      if (chunk) {
        const lines = chunk
          .split('\n')
          .filter(line => line.trim() !== '')
          .map(line => line.replace(/^data: /, '').trim());
        
        for (const line of lines) {
          if (line === '[DONE]') {
            return;
          }
          
          try {
            if (line) {
              if (line.startsWith('{')) {
                const parsed = JSON.parse(line);
                if (parsed.choices && parsed.choices[0] && parsed.choices[0].delta && parsed.choices[0].delta.content) {
                  yield parsed.choices[0].delta.content;
                }
              } else {
                yield line;
              }
            }
          } catch (e) {
            console.warn('解析流数据出错:', e, line);
            if (line) {
              yield line;
            }
          }
        }
      }
    }
  } finally {
    reader.releaseLock();
  }
}