UNPKG

@langchain/community

Version:
35 lines (34 loc) 1.73 kB
import { BaseChatIflytekXinghuo } from "./common.js"; import { BaseWebSocketStream, } from "../../utils/iflytek_websocket_stream.js"; class WebSocketStream extends BaseWebSocketStream { openWebSocket(url, options) { return new WebSocket(url, options.protocols ?? []); } } /** * @example * ```typescript * const model = new ChatIflytekXinghuo(); * const response = await model.invoke([new HumanMessage("Nice to meet you!")]); * console.log(response); * ``` */ export class ChatIflytekXinghuo extends BaseChatIflytekXinghuo { async openWebSocketStream(options) { const host = "spark-api.xf-yun.com"; const date = new Date().toUTCString(); const url = `GET /${this.version}/chat HTTP/1.1`; const keyBuffer = new TextEncoder().encode(this.iflytekApiSecret); const dataBuffer = new TextEncoder().encode(`host: ${host}\ndate: ${date}\n${url}`); const cryptoKey = await crypto.subtle.importKey("raw", keyBuffer, { name: "HMAC", hash: "SHA-256" }, false, ["sign"]); const signature = await crypto.subtle.sign("HMAC", cryptoKey, dataBuffer); const hash = window.btoa(String.fromCharCode(...new Uint8Array(signature))); const authorization_origin = `api_key="${this.iflytekApiKey}", algorithm="hmac-sha256", headers="host date request-line", signature="${hash}"`; const authorization = window.btoa(authorization_origin); let authWebSocketUrl = this.apiUrl; authWebSocketUrl += `?authorization=${authorization}`; authWebSocketUrl += `&host=${encodeURIComponent(host)}`; authWebSocketUrl += `&date=${encodeURIComponent(date)}`; return new WebSocketStream(authWebSocketUrl, options); } }