@@ -33,7 +33,7 @@ export default class AiChat extends Service {
3333 * @return
3434 */
3535
36- async getAnswerFromAi ( messages : ChatCompletionMessageParam [ ] , chatConfig : any , res : any = null ) {
36+ async getAnswerFromAi ( messages : ChatCompletionMessageParam [ ] , chatConfig : any ) {
3737 let result : any = null ;
3838
3939 try {
@@ -48,26 +48,10 @@ export default class AiChat extends Service {
4848 stream : chatConfig . streamStatus
4949 } ) ;
5050
51- // 逐块发送数据到前端
52- if ( chatConfig . streamStatus ) {
53- for await ( const chunk of result ) {
54- const content = chunk . choices [ 0 ] ?. delta ?. content || '' ;
55- res . write ( `data: ${ JSON . stringify ( { content } ) } \n\n` ) ; // SSE 格式
56- }
57- } else {
58- return result ;
59- }
51+ return result ;
6052 } catch ( e : any ) {
6153 this . ctx . logger . debug ( `调用AI大模型接口失败: ${ ( e as Error ) . message } ` ) ;
6254 return this . ctx . helper . getResponseData ( `调用AI大模型接口失败: ${ ( e as Error ) . message } ` ) ;
63- } finally {
64- if ( res ) {
65- res . end ( ) ; // 关闭连接
66- }
67- }
68-
69- if ( ! res ) {
70- return this . ctx . helper . getResponseData ( `调用AI大模型接口未返回正确数据.` ) ;
7155 }
7256 }
7357
@@ -143,9 +127,12 @@ export default class AiChat extends Service {
143127 let params = this . createApiInfo ( process . env . ALIBABA_CLOUD_WORKSPACE_ID ) ;
144128 // query params
145129 let queries : { [ key : string ] : any } = { } ;
146- queries [ 'Query' ] = content ;
147- queries [ 'EnableRewrite' ] = true ;
148- queries [ 'IndexId' ] = process . env . ALIBABA_CLOUD_INDEX_ID ;
130+ const QUERY = 'Query' ;
131+ const ENABLE_REWRITE = 'EnableRewrite' ;
132+ const INDEX_ID = 'IndexId' ;
133+ queries [ QUERY ] = content ;
134+ queries [ ENABLE_REWRITE ] = true ;
135+ queries [ INDEX_ID ] = process . env . ALIBABA_CLOUD_INDEX_ID ;
149136 // runtime options
150137 let runtime = new $Util . RuntimeOptions ( { } ) ;
151138 let request = new $OpenApi . OpenApiRequest ( {
0 commit comments