admin-valentin/awesome-llm-apps
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

final changes1

Browse source
This commit is contained in:
Madhu 2025-01-14 22:37:06 +05:30
parent da9fd37d35
commit a6c13cc852
1 changed files with 13 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
ai_agent_tutorials/ai_chess_game/ai_chess_agents.py
View file

@ -52,25 +52,28 @@ def execute_move(move: str) -> str:
if chess_move not in st.session_state.board.legal_moves: if chess_move not in st.session_state.board.legal_moves:
return f"Invalid move: {move}. Please call available_moves() to see valid moves." return f"Invalid move: {move}. Please call available_moves() to see valid moves."
# Update board state
st.session_state.board.push(chess_move) st.session_state.board.push(chess_move)
st.session_state.made_move = True st.session_state.made_move = True
board_svg = chess.svg.board( # Generate and store board visualization
st.session_state.board, board_svg = chess.svg.board(st.session_state.board,
arrows=[(chess_move.from_square, chess_move.to_square)], arrows=[(chess_move.from_square, chess_move.to_square)],
fill={chess_move.from_square: "gray"}, fill={chess_move.from_square: "gray"},
size=400 size=400)
)
st.session_state.board_svg = board_svg st.session_state.board_svg = board_svg
st.session_state.move_history.append(board_svg) st.session_state.move_history.append(board_svg)
# Get piece information
moved_piece = st.session_state.board.piece_at(chess_move.to_square) moved_piece = st.session_state.board.piece_at(chess_move.to_square)
piece_unicode = moved_piece.unicode_symbol() piece_unicode = moved_piece.unicode_symbol()
piece_type_name = chess.piece_name(moved_piece.piece_type) piece_type_name = chess.piece_name(moved_piece.piece_type)
piece_name = piece_type_name.capitalize() if piece_unicode.isupper() else piece_type_name piece_name = piece_type_name.capitalize() if piece_unicode.isupper() else piece_type_name
move_desc = f"Moved {piece_name} ({piece_unicode}) from {chess.SQUARE_NAMES[chess_move.from_square]} to {chess.SQUARE_NAMES[chess_move.to_square]}." # Generate move description
from_square = chess.SQUARE_NAMES[chess_move.from_square]
to_square = chess.SQUARE_NAMES[chess_move.to_square]
move_desc = f"Moved {piece_name} ({piece_unicode}) from {from_square} to {to_square}."
if st.session_state.board.is_checkmate(): if st.session_state.board.is_checkmate():
winner = 'White' if st.session_state.board.turn == chess.BLACK else 'Black' winner = 'White' if st.session_state.board.turn == chess.BLACK else 'Black'
move_desc += f"\nCheckmate! {winner} wins!" move_desc += f"\nCheckmate! {winner} wins!"
@ -111,7 +114,7 @@ if st.session_state.openai_api_key:
agent_white = ConversableAgent( agent_white = ConversableAgent(
name="Agent_White", name="Agent_White",
system_message="You are a professional chess player and you play as white. " system_message="You are a professional chess player and you play as white. "
"First call available_moves() first, to get list of legal moves. " "First call available_moves() first, to get list of legal available moves. "
"Then call execute_move(move) to make a move.", "Then call execute_move(move) to make a move.",
llm_config={"config_list": agent_white_config_list, "cache_seed": None}, llm_config={"config_list": agent_white_config_list, "cache_seed": None},
) )
@ -119,7 +122,7 @@ if st.session_state.openai_api_key:
agent_black = ConversableAgent( agent_black = ConversableAgent(
name="Agent_Black", name="Agent_Black",
system_message="You are a professional chess player and you play as black. " system_message="You are a professional chess player and you play as black. "
"First call available_moves() first, to get list of legal moves. " "First call available_moves() first, to get list of legal available moves. "
"Then call execute_move(move) to make a move.", "Then call execute_move(move) to make a move.",
llm_config={"config_list": agent_black_config_list, "cache_seed": None}, llm_config={"config_list": agent_black_config_list, "cache_seed": None},
) )