nas/MoviePilot
1
0
Fork 0
mirror of https://github.com/jxxghp/MoviePilot.git synced 2026-03-19 19:46:55 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix agent tools

Browse Source
This commit is contained in:
jxxghp
2025-11-17 12:34:20 +08:00
parent 076fae696c
commit c6806ee648
10 changed files with 20 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
app/agent/tools/impl/get_recommendations.py
View File

@@ -38,17 +38,17 @@ class GetRecommendationsTool(MoviePilotTool):
recommend_chain = RecommendChain()
results = []
if source == "tmdb_trending":
results = recommend_chain.tmdb_trending(limit=limit)
results = await recommend_chain.async_tmdb_trending(limit=limit)
elif source == "douban_hot":
if media_type == "movie":
results = recommend_chain.douban_movie_hot(limit=limit)
results = await recommend_chain.async_douban_movie_hot(limit=limit)
elif media_type == "tv":
results = recommend_chain.douban_tv_hot(limit=limit)
results = await recommend_chain.async_douban_tv_hot(limit=limit)
else: # all
results.extend(recommend_chain.douban_movie_hot(limit=limit))
results.extend(recommend_chain.douban_tv_hot(limit=limit))
results.extend(await recommend_chain.async_douban_movie_hot(limit=limit))
results.extend(await recommend_chain.async_douban_tv_hot(limit=limit))
elif source == "bangumi_calendar":
results = recommend_chain.bangumi_calendar(limit=limit)
results = await recommend_chain.async_bangumi_calendar(limit=limit)
if results:
# 限制最多20条结果