@@ -344,8 +344,8 @@ fn rewrite_request(
344344 }
345345 let header = std:: str:: from_utf8 ( header_bytes) . map_err ( |_| 400_u16 ) ?;
346346 let public_authority = public_origin. strip_prefix ( "http://" ) . ok_or ( 500_u16 ) ?;
347- let pair_protocol = format ! ( "openwork -bearer.{}" , URL_SAFE_NO_PAD . encode( pair_token) ) ;
348- let runtime_protocol = format ! ( "openwork -bearer.{}" , URL_SAFE_NO_PAD . encode( runtime_token) ) ;
347+ let pair_protocol = format ! ( "qwen -bearer.{}" , URL_SAFE_NO_PAD . encode( pair_token) ) ;
348+ let runtime_protocol = format ! ( "qwen -bearer.{}" , URL_SAFE_NO_PAD . encode( runtime_token) ) ;
349349 let websocket = header. lines ( ) . any ( |line| {
350350 line. split_once ( ':' ) . is_some_and ( |( name, value) | {
351351 name. eq_ignore_ascii_case ( "upgrade" ) && value. trim ( ) . eq_ignore_ascii_case ( "websocket" )
@@ -387,7 +387,7 @@ fn rewrite_request(
387387 for protocol in value. split ( ',' ) . map ( str:: trim) {
388388 if protocol == pair_protocol {
389389 protocols. push ( runtime_protocol. as_str ( ) ) ;
390- } else if protocol. contains ( "openwork -bearer." ) {
390+ } else if protocol. contains ( "qwen -bearer." ) {
391391 return Err ( 403 ) ;
392392 } else {
393393 protocols. push ( protocol) ;
@@ -927,7 +927,7 @@ mod tests {
927927 let pair = URL_SAFE_NO_PAD . encode ( "pair-token" ) ;
928928 let runtime = URL_SAFE_NO_PAD . encode ( "runtime-token" ) ;
929929 let request = format ! (
930- "GET /acp HTTP/1.1\r \n Host: 192.168.1.10:49152\r \n Origin: http://192.168.1.10:49152\r \n Connection: Upgrade\r \n Upgrade: websocket\r \n Sec-WebSocket-Protocol: qwen-ws, openwork -bearer.{pair}\r \n \r \n "
930+ "GET /acp HTTP/1.1\r \n Host: 192.168.1.10:49152\r \n Origin: http://192.168.1.10:49152\r \n Connection: Upgrade\r \n Upgrade: websocket\r \n Sec-WebSocket-Protocol: qwen-ws, qwen -bearer.{pair}\r \n \r \n "
931931 ) ;
932932 let rewritten = rewrite_request (
933933 request. as_bytes ( ) ,
@@ -940,12 +940,12 @@ mod tests {
940940 )
941941 . expect ( "rewrite" ) ;
942942 let rewritten = String :: from_utf8 ( rewritten) . expect ( "utf8" ) ;
943- assert ! ( rewritten. contains( & format!( "openwork -bearer.{runtime}" ) ) ) ;
944- assert ! ( !rewritten. contains( & format!( "openwork -bearer.{pair}" ) ) ) ;
943+ assert ! ( rewritten. contains( & format!( "qwen -bearer.{runtime}" ) ) ) ;
944+ assert ! ( !rewritten. contains( & format!( "qwen -bearer.{pair}" ) ) ) ;
945945 assert ! ( rewritten. contains( "Connection: Upgrade\r \n " ) ) ;
946946
947947 let request = format ! (
948- "GET /acp HTTP/1.1\r \n Host: 192.168.1.10:49152\r \n Origin: http://192.168.1.10:49152\r \n Connection: Upgrade\r \n Upgrade: websocket\r \n Sec-WebSocket-Protocol: leak-openwork -bearer.{pair}, openwork -bearer.{pair}\r \n \r \n "
948+ "GET /acp HTTP/1.1\r \n Host: 192.168.1.10:49152\r \n Origin: http://192.168.1.10:49152\r \n Connection: Upgrade\r \n Upgrade: websocket\r \n Sec-WebSocket-Protocol: leak-qwen -bearer.{pair}, qwen -bearer.{pair}\r \n \r \n "
949949 ) ;
950950 assert_eq ! (
951951 rewrite_request(
0 commit comments