Spaces:
Sleeping
Sleeping
File size: 1,871 Bytes
70d877e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 |
import chess
from langchain_core.tools import BaseTool, tool
from langchain_mcp_adapters.client import MultiServerMCPClient
def create_base_tools(board: chess.Board) -> list[BaseTool]:
"""Create tools for interacting with a chess board.
Args:
board (chess.Board): The chess board to interact with.
Returns:
list[BaseTool]: A list of tools for interacting with the chess board.
"""
@tool
def get_fen() -> str:
"""Get the current FEN string of the chess board."""
return board.fen()
@tool
def set_fen(fen: str) -> str:
"""Set the chess board to a specific FEN string.
Don't use when you are playing a game, use the `make_move` tool instead.
Args:
fen (str): The FEN string to set the board to.
"""
try:
board.set_fen(fen)
return board.fen()
except ValueError as e:
return str(e)
@tool
def make_move(move: str) -> str:
"""Make a move on the chess board and return the new FEN string.
Args:
move (str): The move in UCI format (e.g., "e2e4").
"""
try:
chess_move = chess.Move.from_uci(move)
if chess_move in board.legal_moves:
board.push(chess_move)
return board.fen()
else:
return "Illegal move"
except Exception as e:
return str(e)
return [
get_fen,
set_fen,
make_move,
]
async def create_mcp_tools(
url: str = "http://localhost:7860/gradio_api/mcp/sse", transport: str = "sse"
) -> list[BaseTool]:
mcp_client = MultiServerMCPClient(
{
"chess": {
"url": url,
"transport": transport,
}
}
)
return await mcp_client.get_tools()
|