media/social-router
1
0
Fork 0
generated from tailored/router-template
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

perf

Browse Source
This commit is contained in:
熊潇 2025-06-21 18:21:49 +08:00
parent a76b506327
commit da6d4041ad
9 changed files with 121 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
packages/xhs/src/app.ts
View File

@ -1,13 +1,13 @@
import { QueryRouterServer } from '@kevisual/router/browser';
import { XhsServices } from '@kevisual/xhs/services/xhs-services.ts';
import { XhsServices, XhsClient } from '@kevisual/xhs/services/xhs-services.ts';
export const app = new QueryRouterServer();
export const xhsServices = new XhsServices();// Semicolon separated Cookie File
export const xhsServices = new XhsServices(); // Semicolon separated Cookie File
const cookie =
'a1=1978d0cdcb7p5neac7cesgfm9yat0b4a7hnesexkp30000220066;abRequestId=f98f27d6-cceb-53d9-a9ef-dbd68786231b;access-token-creator.xiaohongshu.com=customer.creator.AT-68c517518075239703025713copo06lnusenqic1;customer-sso-sid=68c517518075239698553486mipyx4lwk2ac5jaj;customerClientId=716985104518687;galaxy_creator_session_id=wda55FJhoiAWZ8cekT2aGiM9fuztft3mOdRM;galaxy.creator.beaker.session.id=1750438297110093146037;gid=yjWYf8Si4SWjyjWYf8SfSxdCDWU2Id0SWSdMITUki6jv0dq898D40M888JJ88KK8fdKS8jiy;loadts=1750444792784;sec_poison_id=2f592adb-ec34-48cd-a6ce-22b90bd67c3b;unread={%22ub%22:%226852a4100000000022030063%22%2C%22ue%22:%226854e0db00000000100251f6%22%2C%22uc%22:28};web_session=040069b6528dbc23c355980a603a4b3e03bb6a;webBuild=4.68.0;webId=1dbb23b746393db622165a22357897d5;websectiga=10f9a40ba454a07755a08f27ef8194c53637eba4551cf9751c009d9afb564467;x-user-id-creator.xiaohongshu.com=6726cef4000000001c019303;xsecappid=xhs-pc-web;acw_tc=0a00d41117504447915317110e41effee63c597aa77f2c9f0dc1b0a10248f8;';
'a1=****;abRequestId=f98f27d6-cceb-53d9-a9ef-dbd68786231b;access-token-creator.xiaohongshu.com=customer.creator.AT-68c517518075239703025713copo06lnusenqic1;customer-sso-sid=68c517518075239698553486mipyx4lwk2ac5jaj;customerClientId=716985104518687;galaxy_creator_session_id=wda55FJhoiAWZ8cekT2aGiM9fuztft3mOdRM;galaxy.creator.beaker.session.id=1750438297110093146037;gid=yjWYf8Si4SWjyjWYf8SfSxdCDWU2Id0SWSdMITUki6jv0dq898D40M888JJ88KK8fdKS8jiy;loadts=1750444792784;sec_poison_id=2f592adb-ec34-48cd-a6ce-22b90bd67c3b;unread={%22ub%22:%226852a4100000000022030063%22%2C%22ue%22:%226854e0db00000000100251f6%22%2C%22uc%22:28};web_session=040069b6528dbc23c355980a603a4b3e03bb6a;webBuild=4.68.0;webId=1dbb23b746393db622165a22357897d5;websectiga=10f9a40ba454a07755a08f27ef8194c53637eba4551cf9751c009d9afb564467;x-user-id-creator.xiaohongshu.com=6726cef4000000001c019303;xsecappid=xhs-pc-web;acw_tc=0a00d41117504447915317110e41effee63c597aa77f2c9f0dc1b0a10248f8;';
xhsServices.createRoot({
export const xhsRootClient: XhsClient = xhsServices.createRoot({
cookie,
signConfig: {
signUrl: 'http://light.xiongxiao.me:5006/sign',

4
packages/xhs/src/index.ts
View File

@ -1,5 +1,5 @@
import { XhsClient } from './libs/xhs.ts';
import { app, xhsServices } from './app.ts';
import { app, xhsServices, xhsRootClient } from './app.ts';
import './routes/index.ts';
export { XhsClient, app, xhsServices };
export { XhsClient, app, xhsServices, xhsRootClient };

3
packages/xhs/src/libs/xhs.ts
View File

@ -66,6 +66,9 @@ export class XhsClient extends XhsClientBase {
constructor(opts: XhsOptions) {
super(opts as any);
}
setCookie(cookie: string) {
this.cookie = cookie;
}
getApiInfo = getApiInfo;
printResult(msg: string, data: any) {
if (msg === 'response') {

4
packages/xhs/src/services/xhs-services.ts
View File

@ -3,7 +3,7 @@ import { Sequelize } from 'sequelize';
// import { createSequelize } from '@kevisual/xhs/services/xhs-db/db.ts';
import path from 'node:path';
import fs from 'node:fs';
export { XhsClient };
type XhsClientOptions = {
key: string;
cookie: string;
@ -69,7 +69,7 @@ export class XhsServices {
createRoot(options: Partial<XhsClientOptions>) {
options.key = options.key || this.root;
return this.createClient(options as XhsClientOptions);
return this.createClient(options as XhsClientOptions) as XhsClient;
}
getKey(key?: string) {
if (!key) key = this.root;

80
src/agent/analyze/category.ts Normal file
View File

@ -0,0 +1,80 @@
import { agent } from '@/agent/agent.ts';
import { ai } from '../ai.ts';
import { getJsonFromString } from './content.ts';
export type Category = {
category: string;
description: string;
};
export const categoryList: Category[] = [
{
category: 'daily_poetry',
description: '获取一篇今日诗词',
},
{
category: 'irony',
description: '反讽一下',
},
{
category: 'random_beautiful_text',
description: '随机生成一段优美的文字',
},
{
category: 'random_joke',
description: '随机生成一个笑话',
},
{
category: '',
description: '教教我，怎么做'
},
{
category: 'default',
description: '默认分类，无法识别的请求',
},
];
agent
.route({
path: 'analyze',
key: 'category',
description: '分析文本内容，判断文本的请求分类。比如，1. 获取一篇今日诗词。2. 反讽一下 3. 随机生成一段优美的文字。',
})
.define(async (ctx) => {
const text = ctx.query?.text || '';
const prompt = `
请分析以下文本内容，判断文本的请求分类，并返回一个JSON对象。
识别的分类包括：
${categoryList.map((item) => `- ${item.category}: ${item.description}`).join('\n')}
返回内容示例：
\`\`\`json
{
"category": "daily_poetry"
}
\`\`\`
<context>
${text}
</context>
`;
const res = await ai
.chat(
[
{
role: 'user',
content: prompt,
},
],
{
enable_thinking: false,
},
)
.catch((error) => {
ctx.throw(500, 'AI 服务错误: ' + error.status);
return error;
});
const content = res.choices?.[0]?.message?.content || '';
const jsonResponse = getJsonFromString(content);
const category = jsonResponse?.category || 'default';
ctx.body = { category };
})
.addTo(agent);

2
src/agent/analyze/content.ts
View File

@ -1,7 +1,7 @@
import { agent } from '@/agent/agent.ts';
import { ai } from '../ai.ts';
import { logger } from '@/agent/logger.ts';
const getJsonFromString = (str: string) => {
export const getJsonFromString = (str: string) => {
// 尝试从字符串中提取JSON对象
try {
const jsonMatch = str.match(/```json\s*([\s\S]*?)\s*```/);

2
src/agent/index.ts
View File

@ -1,5 +1,7 @@
import { agent } from './agent.ts';
import './analyze/content.ts';
import './analyze/category.ts';
import './fix/prompt.ts';
import './xhs.ts';

21
src/agent/test/category.ts Normal file
View File

@ -0,0 +1,21 @@
import { agent } from '../index.ts';
const main = async () => {
const text1 = '解答一下这个笔记。';
const text2 = '分析一下这个图片';
const text3 = '这个视频介绍的是什么';
const text4 = '评价一下这个评论。';
const text5 = '关于这个评论。';
const text6 = '1+1=';
const text7 = '反讽一下';
const text8 = '随机生成一段优美的文字';
const res = await agent.call({
path: 'analyze',
key: 'category',
payload: {
text: text1,
},
});
console.log('analyze category res', res.code, 'content', res.body);
};
main();

9
src/task/task.ts
View File

@ -3,13 +3,16 @@
import { QueryRouterServer } from '@kevisual/router';
import { redis } from '@/modules/redis.ts';
import { Queue } from 'bullmq';
import { app as xhsApp } from '@kevisual/xhs/index';
import { app as xhsApp, xhsServices, xhsRootClient, XhsClient } from '@kevisual/xhs/index';
import { nanoid } from 'nanoid';
export const XHS_GET_UNREAD = 'unread_count';
export const XHS_QUEUE_NAME = 'XHS_QUEUE';
import { config } from '../modules/config.ts';
const xhsClient: XhsClient = xhsRootClient;
xhsClient.setCookie(config.XHS_ROOT_COOKIE || '');
console.log('XHS_ROOT_COOKIE', xhsClient.cookie);
export const taskApp = new QueryRouterServer();
export { xhsApp };
export { xhsApp, xhsServices, xhsRootClient };
export const queue = new Queue(XHS_QUEUE_NAME, {
connection: redis,
});