11import type { BubbleListProps } from '@ant-design/x' ;
22import { Bubble , Sender } from '@ant-design/x' ;
3- import { AbstractChatProvider , useXChat , XRequest } from '@ant-design/x-sdk' ;
3+ import { AbstractChatProvider , useXChat , XRequest , XRequestOptions } from '@ant-design/x-sdk' ;
44import { Button , Flex } from 'antd' ;
55import React from 'react' ;
66
@@ -9,7 +9,8 @@ import React from 'react';
99interface CustomInput {
1010 query : string ;
1111 role : 'user' ;
12- stream ?: boolean ;
12+ stream : boolean ;
13+ model : string ;
1314}
1415
1516interface CustomOutput {
@@ -30,14 +31,14 @@ class CustomProvider<
3031> extends AbstractChatProvider < ChatMessage , Input , Output > {
3132 // 转换请求参数:将用户输入转换为标准格式
3233 // Transform request parameters: convert user input to standard format
33- transformParams ( requestParams : Partial < Input > ) : Input {
34+
35+ transformParams ( requestParams : Partial < Input > , options : XRequestOptions < Input , Output > ) : Input {
3436 if ( typeof requestParams !== 'object' ) {
3537 throw new Error ( 'requestParams must be an object' ) ;
3638 }
3739 return {
38- query : requestParams . query || '' ,
39- role : 'user' ,
40- stream : requestParams . stream ?? false ,
40+ ...( options ?. params || { } ) ,
41+ ...( requestParams || { } ) ,
4142 } as Input ;
4243 }
4344
@@ -146,6 +147,10 @@ const App = () => {
146147 new CustomProvider < CustomMessage , CustomInput , CustomOutput > ( {
147148 request : XRequest ( 'https://api.x.ant.design/api/custom_chat_provider_stream' , {
148149 manual : true ,
150+ params : {
151+ stream : true ,
152+ model : 'qwen2.5-7b-instruct' ,
153+ } ,
149154 } ) ,
150155 } ) ,
151156 ) ;
0 commit comments