NopePrime commited on
Commit
d6cc0ad
·
verified ·
1 Parent(s): f373131

Update local_tools.py

Browse files
Files changed (1) hide show
  1. local_tools.py +4 -8
local_tools.py CHANGED
@@ -6,14 +6,10 @@ mcp = FastMCP("Yue-Sing-Tools")
6
 
7
  @mcp.tool()
8
  def sing_song(song_name: str) -> str:
9
- """Hát một bài hát dựa trên tên file trong thư mục sing/tracks/"""
10
-
11
- # Kiểm tra nếu song_name chưa có đuôi .mp3 thì mới thêm vào
12
- if not song_name.lower().endswith(".mp3"):
13
- song_name = f"{song_name}.mp3"
14
-
15
- # Trả về đúng định dạng mà HTML đang chờ đợi
16
- return f"[[SING_COMMAND]]{song_name}"
17
 
18
  if __name__ == "__main__":
19
  # Chạy server MCP
 
6
 
7
  @mcp.tool()
8
  def sing_song(song_name: str) -> str:
9
+ # Nếu AI truyền "golden.mp3", ta giữ nguyên.
10
+ # Nếu AI truyền "golden", ta mới thêm .mp3.
11
+ clean_name = song_name if song_name.endswith(".mp3") else f"{song_name}.mp3"
12
+ return f"[[SING_COMMAND]]{clean_name}"
 
 
 
 
13
 
14
  if __name__ == "__main__":
15
  # Chạy server MCP