Update agents/models/prompts.py

agents/models/prompts.py

@@ -3,14 +3,14 @@
 # https://claude.ai/public/artifacts/9d6efdeb-7c0d-4c8d-a71f-ba5c8f0da6e1
 
 PROMPT_IMG_TO_FEN = """
-You are an expert image to FEN generation assistant. Given a user question and an image of a chess position, your task is to generate the FEN of the image.
+You are an expert image to FEN generation assistant.
+Given a user question, image of a chess position, and active color, your task is to generate the FEN of the image.
 
 **Approach:**
 1. Following the rules below, generate the FEN of the image.
-2. IMPORTANT: If the user question below mentions NOT to move certain pieces, REMOVE them from the FEN.
-3. Once you have the FEN of the image, validate everything is correct.
+2. Once you have the FEN of the image, validate everything is correct.
 
-**Active Color**: See the user question below
+**Active Color**: {active_color}
 
 **Rules:**
 - Start from rank 8 to rank 1.
@@ -20,6 +20,7 @@ You are an expert image to FEN generation assistant. Given a user question and a
 - Use numbers (1-8) for consecutive empty squares.
 - Separate ranks with '/'.
 - Provide all portions of the FEN, including active color, castling rights, en passant, and move counters.
+- IMPORTANT: If the user question mentions not to move certain pieces, remove them.
 - Return ONLY the FEN, nothing else.
 
 **Example Output:** rnbqkbnr/pppppppp/8/8/8/8/PPPPPPPP/RNBQKBNR w KQkq - 0 1
@@ -30,25 +31,25 @@ You are an expert image to FEN generation assistant. Given a user question and a
 """
 
 PROMPT_ALGEBRAIC_CHESS_NOTATION = """
-You are an expert Universal Chess Interface (UCI) to algebraic chess notation (aka standard notation) generation assistant.
-Given a user question, image of a chess position, and chess position evaluation in UCI, your task is to answer the question in algebraic chess notation.
+You are an expert algebraic chess notation (aka standard notation) generation assistant.
+Given a user question, image of a chess position, and position evaluation in UCI, your task is to answer the question in algebraic chess notation.
 
 **Approach:**
 1. Following the rules below, generate the answer in algebraic chess notation.
-2. IMPORTANT: Add symbols for capture, check, checkmate, etc., IF APPLICABLE.
-3. Once you have the answer in algebraic chess notation, validate everything is correct.
+2. Once you have the answer in algebraic chess notation, validate everything is correct.
 
-**Active Color**: See the user question below
+**Position Evaluation in UCI**: {position_evaluation}
 
 **Rules:**
 - Each move lists the piece letter (K, Q, R, B, N). Pawns are written with no letter.
 - The destination square (file + rank) always appears (e.g., e4, c6).
-- Captures use an “x” before the destination (e.g., Bxe5, exd5).
-- Checks add “+”, checkmate adds “#”.
+- Captures use an "x" before the destination (e.g., Bxe5, exd5).
+- Checks add "+", checkmate adds "#".
 - Castling is written as O-O (king-side) or O-O-O (queen-side).
-- Pawn promotion adds “=Piece” after the move (e.g., e8=Q).
+- Pawn promotion adds "=Piece" after the move (e.g., e8=Q).
 - If two identical pieces can go to the same square, the move is disambiguated with file or rank (e.g., Nbd2 or R1e2).
 - En passant is recorded like a standard capture (e.g., exd6 e.p.).
+- IMPORTANT: If applicable based on the image, add symbols for capture, check, checkmate, etc.
 
 **Example Output:** 1...Qxb1 2...Ra8#