import { z } from 'zod';
import { defineToolConfig, handleSuccessResult, http } from '../utils';
const headers = {
'User-Agent':
'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/123.0.0.0 Safari/537.36',
Accept:
'text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,image/apng,*/*;q=0.8,application/signed-exchange;v=b3;q=0.7',
'Accept-Language': 'zh-CN,zh;q=0.9,en;q=0.8',
'Accept-Encoding': 'gzip, deflate, br',
'Cache-Control': 'no-cache',
Connection: 'keep-alive',
'Sec-Ch-Ua': '"Google Chrome";v="123", "Not:A-Brand";v="8", "Chromium";v="123"',
'Sec-Ch-Ua-Mobile': '?0',
'Sec-Ch-Ua-Platform': '"Windows"',
'Sec-Fetch-Dest': 'document',
'Sec-Fetch-Mode': 'navigate',
'Sec-Fetch-Site': 'same-origin',
'Sec-Fetch-User': '?1',
'Upgrade-Insecure-Requests': '1',
};
export default defineToolConfig(async () => {
const categoryResp = await http.get<{
err_no: number;
err_msg: string;
data: {
category_id: string;
category_name: string;
}[];
}>('https://api.juejin.cn/tag_api/v1/query_category_briefs', { headers });
if (categoryResp.data.err_no !== 0) {
throw new Error('获取掘金分类失败');
}
const juejinRequestSchema = z.object({
category_id: z
.union(
categoryResp.data.data.map((item) => z.literal(item.category_id).describe(item.category_name)) as [
z.ZodLiteral<string>,
z.ZodLiteral<string>,
],
)
.optional()
.default(categoryResp.data.data[0].category_id),
});
return {
name: 'get-juejin-article-rank',
description: '获取掘金文章榜，包含前端开发、后端技术、人工智能、移动开发及技术架构等领域的高质量中文技术文章和教程',
zodSchema: juejinRequestSchema,
func: async (args) => {
const { category_id } = juejinRequestSchema.parse(args);
const resp = await http.get<{
err_no: number;
err_msg: string;
data: any[];
}>('https://api.juejin.cn/content_api/v1/content/article_rank', {
params: {
category_id,
type: 'hot',
},
headers,
});
if (resp.data.err_no !== 0) {
throw new Error(resp.data.err_msg || '获取掘金文章榜失败');
}
return resp.data.data.map((item) => {
return {
title: item.content.title,
brief: item.content.brief || undefined,
author: item.author.name,
popularity: item.content_counter.hot_rank,
view_count: item.content_counter.view,
like_count: item.content_counter.like,
collect_count: item.content_counter.collect,
comment_count: item.content_counter.comment_count,
interact_count: item.content_counter.interact_count,
link: `https://juejin.cn/post/${item.content.content_id}`,
};
});
},
};
});