generated from tailored/router-template
117 lines
2.6 KiB
TypeScript
117 lines
2.6 KiB
TypeScript
import { agent } from '@/agent/agent.ts';
|
||
import { ai } from '../ai.ts';
|
||
import { getJsonFromString } from './content.ts';
|
||
// import { getJsonFromString } from '@kevisual/ai/src/utils/json.ts';
|
||
import { logger } from '../logger.ts';
|
||
|
||
export const cmdList: {
|
||
category: string;
|
||
description?: string;
|
||
action?: {
|
||
path?: string;
|
||
key?: string;
|
||
};
|
||
}[] = [
|
||
{
|
||
category: '指令夸人',
|
||
description: `进行夸奖`,
|
||
action: {
|
||
path: 'tools',
|
||
key: 'good-job',
|
||
},
|
||
},
|
||
{
|
||
category: '指令来了',
|
||
description: `召唤小助手过来`,
|
||
action: {
|
||
path: 'tools',
|
||
key: 'call-xiaoxiao',
|
||
},
|
||
},
|
||
{
|
||
category: '指令总结',
|
||
description: `总结当前的笔记，缩写当前笔记的内容`,
|
||
action: {
|
||
path: 'tools',
|
||
key: 'summarize-note',
|
||
},
|
||
},
|
||
{
|
||
category: '指令安慰',
|
||
description: `安慰用户`,
|
||
action: {
|
||
path: 'tools',
|
||
key: 'comfort-user',
|
||
},
|
||
},
|
||
];
|
||
|
||
agent
|
||
.route({
|
||
path: 'analyze',
|
||
key: 'cmd',
|
||
description: '分析文本内容,意图分析，判断对应的指令内容',
|
||
})
|
||
.define(async (ctx) => {
|
||
let text = ctx.query?.text || '';
|
||
if (text.length > 40) {
|
||
text = text.slice(0, 40).trim();
|
||
}
|
||
let result = {
|
||
category: 'default',
|
||
};
|
||
|
||
const prompt = `
|
||
请分析<context>包函的内容，判断是否程序运行指令，返回一个JSON对象。
|
||
识别的分类包括：
|
||
${cmdList.map((item) => `- ${item.category}: ${item.description}`).join('\n')}
|
||
|
||
返回内容示例：
|
||
\`\`\`json
|
||
{
|
||
"category": "daily_poetry"
|
||
}
|
||
\`\`\`
|
||
|
||
分析的内容是
|
||
<context>
|
||
${text}
|
||
</context>
|
||
`;
|
||
logger.info('Command analysis prompt:', prompt);
|
||
|
||
const res = await ai
|
||
.chat(
|
||
[
|
||
{
|
||
role: 'user',
|
||
content: prompt,
|
||
},
|
||
],
|
||
{
|
||
// @ts-ignore
|
||
enable_thinking: false,
|
||
},
|
||
)
|
||
.catch((err) => {
|
||
console.log('AI service error:', err.status);
|
||
ctx.throw(500, 'AI service error: ' + err.status);
|
||
return err;
|
||
});
|
||
|
||
const ans = res.choices[0]?.message?.content || '';
|
||
console.log('Command analysis response:', ans);
|
||
const json = getJsonFromString(ans);
|
||
if (!json) {
|
||
logger.error('Invalid JSON format in response:', ans);
|
||
ctx.throw(400, 'Invalid JSON format in response');
|
||
}
|
||
result = {
|
||
category: json.category || 'default',
|
||
};
|
||
const cmd = cmdList.find((item) => item.category === result.category);
|
||
logger.info('Command analysis result:', cmd?.category, cmd?.action);
|
||
ctx.body = { cmd, text: text };
|
||
})
|
||
.addTo(agent);
|