chess_agent.py

from langchain_openai import ChatOpenAI
import chess
import chess.engine
from langchain_core.pydantic_v1 import BaseModel, Field
from dotenv import load_dotenv


load_dotenv()


class NextMove(BaseModel):
    move: str = Field(..., description="Best next move to win the chess game. It should be in Standard Algebraic Notation. Always plan 5 moves ahead before deciding the best next move.")
    reasoning: str = Field(..., description="Reasoning explaining why the move is the best")


model = ChatOpenAI(model="gpt-4")
chess_agent = model.with_structured_output(NextMove)