flare-chat-core 0.2.1 → 0.2.2

package/src/chat-core/stream/sse-client.js

@@ -1,244 +1 @@
-const API_BASE_URL = import.meta.env.DEV ? '' : import.meta.env.VITE_API_URL || '';
-const DEFAULT_ENDPOINT = '/api/v1/chat/stream';
-
-function joinUrl(baseUrl, endpoint) {
-  if (!baseUrl) {
-    return endpoint;
-  }
-
-  if (!endpoint) {
-    return baseUrl;
-  }
-
-  return `${baseUrl.replace(/\/+$/, '')}/${endpoint.replace(/^\/+/, '')}`;
-}
-
-function buildClientRequestId() {
-  const random = Math.random().toString(36).slice(2, 10);
-  return `req_${Date.now()}_${random}`;
-}
-
-export class SSEClient {
-  constructor({ endpoint = DEFAULT_ENDPOINT, baseUrl = API_BASE_URL } = {}) {
-    this.abortController = null;
-    this.endpoint = endpoint;
-    this.baseUrl = baseUrl;
-    this.activeRequestId = '';
-    this.authReadyPromise = null;
-    this.authReadyAtMs = 0;
-  }
-
-  async sendMessage(sessionOrParams, contentArg, onEventArg, onCompleteArg, onErrorArg, optionsArg = {}) {
-    const isObjectCall = typeof sessionOrParams === 'object' && sessionOrParams !== null;
-    const params = isObjectCall
-      ? sessionOrParams
-      : {
-          sessionId: sessionOrParams,
-          content: contentArg,
-          enabledCapabilities: optionsArg.enabledCapabilities || [],
-        };
-
-    const sessionId = params.sessionId;
-    const content = params.content;
-    const enabledCapabilities = Array.isArray(params.enabledCapabilities) ? params.enabledCapabilities : [];
-    const modeKey = typeof params.modeKey === 'string' ? params.modeKey.trim() : '';
-    const manualModeKey = typeof params.manualModeKey === 'string' ? params.manualModeKey.trim() : '';
-    const payloadExtra = (
-      params.payloadExtra
-      && typeof params.payloadExtra === 'object'
-      && !Array.isArray(params.payloadExtra)
-    )
-      ? params.payloadExtra
-      : {};
-    const onEvent = isObjectCall ? contentArg : onEventArg;
-    const onComplete = isObjectCall ? onEventArg : onCompleteArg;
-    const onError = isObjectCall ? onCompleteArg : onErrorArg;
-    const url = params.url || joinUrl(params.baseUrl ?? this.baseUrl, params.endpoint || this.endpoint);
-    const command = typeof params.command === 'string' && params.command.trim()
-      ? params.command.trim()
-      : (typeof payloadExtra.command === 'string' && payloadExtra.command.trim() ? payloadExtra.command.trim() : 'send_message');
-    const clientRequestId = typeof params.clientRequestId === 'string' && params.clientRequestId.trim()
-      ? params.clientRequestId.trim()
-      : buildClientRequestId();
-    const contractVersion = typeof params.contractVersion === 'string' && params.contractVersion.trim()
-      ? params.contractVersion.trim()
-      : 'flare.v1';
-    const lastTurnId = typeof params.lastTurnId === 'string' ? params.lastTurnId.trim() : '';
-    const token = typeof params.token === 'string' ? params.token.trim() : '';
-    const onTiming = typeof params.onTiming === 'function' ? params.onTiming : null;
-    const authProvider = (
-      params.auth
-      && typeof params.auth === 'object'
-      && !Array.isArray(params.auth)
-    ) ? params.auth : null;
-
-    const nowMs = Date.now();
-    onTiming?.({
-      point: 't_submit',
-      at_ms: nowMs,
-      request_id: clientRequestId,
-      session_id: String(sessionId || '').trim(),
-    });
-
-    const waitAuthReady = async () => {
-      if (!authProvider || typeof authProvider.ensureReady !== 'function') {
-        return {
-          authorizationToken: token,
-          authReadyAtMs: nowMs,
-        };
-      }
-      if (!this.authReadyPromise) {
-        this.authReadyPromise = Promise.resolve()
-          .then(() => authProvider.ensureReady())
-          .then((resolved) => {
-            this.authReadyAtMs = Date.now();
-            return resolved || null;
-          })
-          .finally(() => {
-            this.authReadyPromise = null;
-          });
-      }
-      const authResult = await this.authReadyPromise;
-      const tokenFromAuthResult = typeof authResult?.token === 'string' ? authResult.token.trim() : '';
-      const tokenFromProvider = typeof authProvider.getToken === 'function'
-        ? String(authProvider.getToken() || '').trim()
-        : '';
-      return {
-        authorizationToken: tokenFromAuthResult || tokenFromProvider || token,
-        authReadyAtMs: this.authReadyAtMs || Date.now(),
-      };
-    };
-
-    const currentRequestId = clientRequestId;
-    if (this.abortController) {
-      this.abortController.abort();
-      this.abortController = null;
-    }
-    this.abortController = new AbortController();
-    this.activeRequestId = currentRequestId;
-
-    try {
-      const authResolved = await waitAuthReady();
-      onTiming?.({
-        point: 't_auth_ready',
-        at_ms: authResolved.authReadyAtMs || Date.now(),
-        request_id: currentRequestId,
-        session_id: String(sessionId || '').trim(),
-      });
-      if (this.activeRequestId !== currentRequestId) {
-        return;
-      }
-      const response = await fetch(url, {
-        method: 'POST',
-        headers: {
-          'Content-Type': 'application/json',
-          ...(authResolved.authorizationToken ? { Authorization: `Bearer ${authResolved.authorizationToken}` } : {}),
-        },
-        body: JSON.stringify({
-          contract_version: contractVersion,
-          client_request_id: clientRequestId,
-          command,
-          last_turn_id: lastTurnId || null,
-          message: content,
-          session_id: sessionId,
-          enabled_capabilities: enabledCapabilities,
-          ...(modeKey ? { mode: modeKey } : {}),
-          ...(manualModeKey ? { manual_mode: manualModeKey } : {}),
-          payload: {
-            message: content,
-            ...(modeKey ? { mode: modeKey } : {}),
-            ...(manualModeKey ? { manual_mode: manualModeKey } : {}),
-            ...payloadExtra,
-          },
-        }),
-        signal: this.abortController.signal,
-      });
-      onTiming?.({
-        point: 't_stream_connected',
-        at_ms: Date.now(),
-        request_id: currentRequestId,
-        session_id: String(sessionId || '').trim(),
-      });
-
-      if (!response.ok) {
-        throw new Error(`HTTP ${response.status}: ${response.statusText}`);
-      }
-
-      const reader = response.body.getReader();
-      const decoder = new TextDecoder();
-      let buffer = '';
-      let currentEvent = '';
-
-      while (true) {
-        const { done, value } = await reader.read();
-        if (done) break;
-
-        buffer += decoder.decode(value, { stream: true });
-        const lines = buffer.split('\n');
-        buffer = lines.pop() || '';
-
-        for (const line of lines) {
-          if (line.startsWith('event:')) {
-            currentEvent = line.substring(6).trim();
-            continue;
-          }
-
-          if (!line.startsWith('data:')) {
-            continue;
-          }
-
-          const data = line.substring(5).trim();
-          if (!data || !currentEvent) {
-            continue;
-          }
-
-          try {
-            const eventData = JSON.parse(data);
-            if (this.activeRequestId !== currentRequestId) {
-              return;
-            }
-            onEvent?.({
-              type: currentEvent,
-              data: eventData,
-            });
-
-            if (currentEvent === 'complete' || currentEvent === 'done') {
-              await onComplete?.();
-              if (this.activeRequestId === currentRequestId) {
-                this.abortController = null;
-                this.activeRequestId = '';
-              }
-              return;
-            }
-          } catch (error) {
-            console.error('SSE 数据解析失败:', error, 'data:', data);
-          }
-        }
-      }
-
-      await onComplete?.();
-      if (this.activeRequestId === currentRequestId) {
-        this.abortController = null;
-        this.activeRequestId = '';
-      }
-    } catch (error) {
-      if (error?.name === 'AbortError') {
-        return;
-      }
-
-      onError?.(error instanceof Error ? error.message : '发送消息失败');
-      if (this.activeRequestId === currentRequestId) {
-        this.abortController = null;
-        this.activeRequestId = '';
-      }
-    }
-  }
-
-  abort() {
-    if (this.abortController) {
-      this.abortController.abort();
-      this.abortController = null;
-    }
-    this.activeRequestId = '';
-  }
-}
+export { SSEClient } from '../../adapters/sse-client.js';

package/src/chat-core/stream/sse-event-dispatcher.js

@@ -0,0 +1 @@
+export * from '../../adapters/sse-event-dispatcher.js';