mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-03-20 03:57:30 +08:00
86 lines
4.3 KiB
Python
86 lines
4.3 KiB
Python
"""搜索演员参演作品工具"""
|
|
|
|
import json
|
|
from typing import Optional, Type
|
|
|
|
from pydantic import BaseModel, Field
|
|
|
|
from app.agent.tools.base import MoviePilotTool
|
|
from app.chain.douban import DoubanChain
|
|
from app.chain.tmdb import TmdbChain
|
|
from app.chain.bangumi import BangumiChain
|
|
from app.log import logger
|
|
|
|
|
|
class SearchPersonCreditsInput(BaseModel):
|
|
"""搜索演员参演作品工具的输入参数模型"""
|
|
explanation: str = Field(..., description="Clear explanation of why this tool is being used in the current context")
|
|
person_id: int = Field(..., description="The ID of the person/actor to search for credits (e.g., 31 for Tom Hanks in TMDB)")
|
|
source: str = Field(..., description="The data source: 'tmdb' for TheMovieDB, 'douban' for Douban, 'bangumi' for Bangumi")
|
|
page: Optional[int] = Field(1, description="Page number for pagination (default: 1)")
|
|
|
|
|
|
class SearchPersonCreditsTool(MoviePilotTool):
|
|
name: str = "search_person_credits"
|
|
description: str = "Search for films and TV shows that a person/actor has appeared in (filmography). Supports searching by person ID from TMDB, Douban, or Bangumi database. Returns a list of media works the person has participated in."
|
|
args_schema: Type[BaseModel] = SearchPersonCreditsInput
|
|
|
|
def get_tool_message(self, **kwargs) -> Optional[str]:
|
|
"""根据搜索参数生成友好的提示消息"""
|
|
person_id = kwargs.get("person_id", "")
|
|
source = kwargs.get("source", "")
|
|
return f"正在搜索人物参演作品: {source} ID {person_id}"
|
|
|
|
async def run(self, person_id: int, source: str, page: Optional[int] = 1, **kwargs) -> str:
|
|
logger.info(f"执行工具: {self.name}, 参数: person_id={person_id}, source={source}, page={page}")
|
|
|
|
try:
|
|
# 根据source选择相应的chain
|
|
if source.lower() == "tmdb":
|
|
tmdb_chain = TmdbChain()
|
|
medias = await tmdb_chain.async_person_credits(person_id=person_id, page=page)
|
|
elif source.lower() == "douban":
|
|
douban_chain = DoubanChain()
|
|
medias = await douban_chain.async_person_credits(person_id=person_id, page=page)
|
|
elif source.lower() == "bangumi":
|
|
bangumi_chain = BangumiChain()
|
|
medias = await bangumi_chain.async_person_credits(person_id=person_id)
|
|
else:
|
|
return f"不支持的数据源: {source}。支持的数据源: tmdb, douban, bangumi"
|
|
|
|
if medias:
|
|
# 限制最多30条结果
|
|
total_count = len(medias)
|
|
limited_medias = medias[:30]
|
|
# 精简字段，只保留关键信息
|
|
simplified_results = []
|
|
for media in limited_medias:
|
|
simplified = {
|
|
"title": media.title,
|
|
"en_title": media.en_title,
|
|
"year": media.year,
|
|
"type": media.type.value if media.type else None,
|
|
"season": media.season,
|
|
"tmdb_id": media.tmdb_id,
|
|
"imdb_id": media.imdb_id,
|
|
"douban_id": media.douban_id,
|
|
"overview": media.overview[:200] + "..." if media.overview and len(media.overview) > 200 else media.overview,
|
|
"vote_average": media.vote_average,
|
|
"poster_path": media.poster_path,
|
|
"backdrop_path": media.backdrop_path,
|
|
"detail_link": media.detail_link
|
|
}
|
|
simplified_results.append(simplified)
|
|
|
|
result_json = json.dumps(simplified_results, ensure_ascii=False, indent=2)
|
|
# 如果结果被裁剪，添加提示信息
|
|
if total_count > 30:
|
|
return f"注意：搜索结果共找到 {total_count} 条，为节省上下文空间，仅显示前 30 条结果。\n\n{result_json}"
|
|
return result_json
|
|
else:
|
|
return f"未找到人物 ID {person_id} ({source}) 的参演作品"
|
|
except Exception as e:
|
|
error_message = f"搜索演员参演作品失败: {str(e)}"
|
|
logger.error(f"搜索演员参演作品失败: {e}", exc_info=True)
|
|
return error_message
|