generated from tailored/router-template
54 lines
1.3 KiB
TypeScript
54 lines
1.3 KiB
TypeScript
import { nanoid } from 'nanoid';
|
|
import { agent } from './agent.ts';
|
|
import { ai } from './ai.ts';
|
|
/**
|
|
* 清除文本中的@信息
|
|
* @param text
|
|
*/
|
|
const clearAtInfo = (text: string = '') => {
|
|
const newText = text.replace(/@[\u4e00-\u9fa5\w]+/g, '').replace(/#.*?#/g, '');
|
|
return newText.trim();
|
|
};
|
|
agent
|
|
.route({
|
|
path: 'xhs',
|
|
})
|
|
.define(async (ctx) => {
|
|
const { text = '' } = ctx.query || {};
|
|
const id = nanoid();
|
|
const no_at_text = clearAtInfo(text);
|
|
const resFix = await agent.call({
|
|
path: 'fix',
|
|
key: 'xhs',
|
|
payload: {
|
|
text: no_at_text,
|
|
},
|
|
});
|
|
if (resFix.code !== 200) {
|
|
ctx.throw(500, 'AI 小红书prompt优化错误: ' + resFix.message);
|
|
return;
|
|
} else {
|
|
console.log('小红书优化的文本', resFix.body);
|
|
}
|
|
const prompt_text = resFix.body || '';
|
|
const res = await ai
|
|
.chat(
|
|
[
|
|
{
|
|
role: 'user',
|
|
content: prompt_text,
|
|
},
|
|
],
|
|
{
|
|
// @ts-ignore
|
|
enable_thinking: false,
|
|
},
|
|
)
|
|
.catch((error) => {
|
|
ctx.throw(500, 'AI 服务错误: ' + error.status);
|
|
return error;
|
|
});
|
|
ctx.body = res.choices?.[0]?.message?.content || '';
|
|
})
|
|
.addTo(agent);
|