Skip to content

Commit f6c2de0

Browse files
committed
fix:fix codecheck
1 parent 387201f commit f6c2de0

File tree

2 files changed

+25
-26
lines changed

2 files changed

+25
-26
lines changed

app/controller/app-center/aiChat.ts

Lines changed: 17 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -30,11 +30,23 @@ export default class AiChatController extends Controller {
3030
'Cache-Control': 'no-cache',
3131
Connection: 'keep-alive'
3232
});
33-
await ctx.service.appCenter.aiChat.getAnswerFromAi(
34-
messages,
35-
{ apiKey, baseUrl, model, streamStatus },
36-
ctx.res
37-
);
33+
try {
34+
const result = await ctx.service.appCenter.aiChat.getAnswerFromAi(messages, {
35+
apiKey,
36+
baseUrl,
37+
model,
38+
streamStatus
39+
});
40+
for await (const chunk of result) {
41+
const content = chunk.choices[0]?.delta?.content || '';
42+
ctx.res.write(`data: ${JSON.stringify({ content })}\n\n`); // SSE 格式
43+
}
44+
} catch (e: any) {
45+
ctx.logger.debug(`调用AI大模型接口失败: ${(e as Error).message}`);
46+
} finally {
47+
ctx.res.end(); // 关闭连接
48+
}
49+
3850
return;
3951
}
4052

app/service/app-center/aiChat.ts

Lines changed: 8 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)