BAML Integration

class HintResponse {
  word string @description("One-word hint (no spaces, not on the board)")
  count int @description("Number of words this hint relates to (1-9)")
  reasoning string @description("Brief explanation of strategy and word associations")
}

class GuessResponse {
  guesses string[] @description("List of words to guess, ordered by confidence")
  reasoning string @description("Brief explanation of why these words relate")
}

function GiveHint(
  team: string @description("Team color: 'blue' or 'red'"),
  my_words: string[] @description("Your team's unrevealed words"),
  opponent_words: string[] @description("Opponent's unrevealed words to avoid"),
  neutral_words: string[] @description("Neutral unrevealed words to avoid"),
  bomb_words: string[] @description("The bomb word(s) - NEVER hint at these!"),
  revealed_words: string[] @description("Already revealed words (for context)")
) -> HintResponse {
  client GPT4oMini  // Default client

  prompt #"
    You are playing Codenames as the {{ team | upper }} team's spymaster.

    YOUR GOAL: Give a one-word hint and a number to help your teammate.

    YOUR TEAM'S WORDS:
    {{ my_words | join(', ') }}

    OPPONENT'S WORDS (avoid these):
    {{ opponent_words | join(', ') }}

    NEUTRAL WORDS (avoid these):
    {{ neutral_words | join(', ') }}

    BOMB WORD(S) (NEVER hint at these):
    {{ bomb_words | join(', ') }}

    {% if revealed_words | length > 0 %}
    ALREADY REVEALED:
    {{ revealed_words | join(', ') }}
    {% endif %}

    RULES:
    1. Give a ONE-WORD hint (no spaces, no words from the board)
    2. Give a NUMBER indicating how many of your words relate to this hint
    3. Your hint should connect multiple of your words if possible
    4. Avoid hints that could lead to opponent words, neutral words, or BOMB(S)

    {{ ctx.output_format }}
  "#
}

function MakeGuesses(
  team: string @description("Team color: 'blue' or 'red'"),
  hint_word: string @description("The hint word given by your spymaster"),
  hint_count: int @description("Number of words the hint relates to"),
  board_words: string[] @description("All words currently on the board"),
  revealed_words: string[] @description("Already revealed words (don't guess these)")
) -> GuessResponse {
  client GPT4oMini

  prompt #"
    You are playing Codenames as the {{ team | upper }} team's field operative.

    YOUR HINT: "{{ hint_word }}" ({{ hint_count }})

    WORDS ON THE BOARD (unrevealed):
    {% for word in board_words if word not in revealed_words %}{{ word }}{% if not loop.last %}, {% endif %}{% endfor %}

    YOUR TASK:
    1. Identify which unrevealed words relate to "{{ hint_word }}"
    2. Return up to {{ hint_count }} words (you can guess fewer if unsure)
    3. Order them by confidence (most confident first)

    IMPORTANT:
    - Only guess words from the unrevealed list above
    - If you guess wrong, your turn ends immediately
    - Be thoughtful - quality over quantity

    {{ ctx.output_format }}
  "#
}

client GPT4oMini {
  provider openai
  options {
    model "gpt-4o-mini"
    api_key env.OPENAI_API_KEY
    temperature 0.7
  }
}

client ClaudeSonnet45 {
  provider anthropic
  options {
    model "claude-sonnet-4-5-20250929"
    api_key env.ANTHROPIC_API_KEY
    temperature 0.7
  }
}

client Gemini25Flash {
  provider google-ai
  options {
    model "gemini-2.5-flash"
    api_key env.GOOGLE_API_KEY
    temperature 0.7
  }
}

from baml_client.baml_client.sync_client import b
from baml_py import ClientRegistry

# Create and configure registry
registry = ClientRegistry()
registry.set_primary("ClaudeSonnet45")  # Override default

# Pass registry to BAML function
response = b.GiveHint(
    team="blue",
    my_words=["dog", "cat"],
    opponent_words=["house"],
    neutral_words=["tree"],
    bomb_words=["bomb"],
    revealed_words=[],
    baml_options={"client_registry": registry}
)

class BAMLHintGiver(HintGiver):
    def __init__(self, team: Team, model: BAMLModel = BAMLModel.GPT4O_MINI):
        super().__init__(team)
        self.model = model
        self._registry = ClientRegistry()
        self._registry.set_primary(model.value)  # e.g., "GPT4oMini"
    
    def give_hint(self, ...) -> HintResponse:
        baml_response = b.GiveHint(
            ...,
            baml_options={"client_registry": self._registry}
        )
        return HintResponse(word=baml_response.word, count=baml_response.count)