BAML Agents

from game import Team
from agents.llm.baml_agents import BAMLHintGiver, BAMLModel

# Using default model (GPT4oMini)
hint_giver = BAMLHintGiver(team=Team.BLUE)

# Using specific model
hint_giver = BAMLHintGiver(
    team=Team.RED,
    model=BAMLModel.CLAUDE_SONNET_45
)

model_name = hint_giver.get_model_name()
print(f"Using model: {model_name}")

response = hint_giver.give_hint(
    my_words=["whale", "dolphin", "shark"],
    opponent_words=["cat", "dog"],
    neutral_words=["table", "chair"],
    bomb_words=["explosion"],
    revealed_words=["car"],
    board_words=["whale", "dolphin", "shark", "cat", "dog", 
                 "table", "chair", "explosion", "car"]
)
print(f"Hint: {response.word} {response.count}")

from game import Team
from agents.llm.baml_agents import BAMLGuesser, BAMLModel

# Using default model (GPT4oMini)
guesser = BAMLGuesser(team=Team.BLUE)

# Using specific model
guesser = BAMLGuesser(
    team=Team.RED,
    model=BAMLModel.GEMINI_25_FLASH
)

model_name = guesser.get_model_name()
print(f"Using model: {model_name}")

guesses = guesser.make_guesses(
    hint_word="ocean",
    hint_count=3,
    board_words=["whale", "dolphin", "shark", "cat", "dog",
                 "table", "chair", "explosion"],
    revealed_words=["car"]
)
print(f"Guesses: {guesses}")

from game import CardColor

guesser.process_result(
    guessed_word="whale",
    was_correct=True,
    color=CardColor.BLUE
)

# Check history
print(guesser.guess_history)
# [{'word': 'whale', 'correct': True, 'color': 'blue'}]

# After a game completes
guesser.reset()
assert len(guesser.guess_history) == 0

from agents.llm.baml_agents import create_hint_giver
from game import Team

# With specific model
hint_giver = create_hint_giver("openai", "gpt-4o", Team.BLUE)

# Using provider default
hint_giver = create_hint_giver("anthropic", team=Team.RED)

# Google Gemini
hint_giver = create_hint_giver("google", "gemini-2.5-pro", Team.BLUE)

from agents.llm.baml_agents import create_guesser
from game import Team

# With specific model
guesser = create_guesser("openai", "gpt-4o", Team.BLUE)

# Using provider default
guesser = create_guesser("anthropic", team=Team.RED)

# Google Gemini
guesser = create_guesser("google", "gemini-2.5-flash", Team.BLUE)

from game import Game, Team
from agents.llm.baml_agents import (
    BAMLHintGiver,
    BAMLGuesser,
    BAMLModel,
    create_hint_giver,
    create_guesser
)

# Method 1: Direct instantiation with BAMLModel enum
blue_hint_giver = BAMLHintGiver(
    team=Team.BLUE,
    model=BAMLModel.GPT4O_MINI
)
blue_guesser = BAMLGuesser(
    team=Team.BLUE,
    model=BAMLModel.GPT4O_MINI
)

# Method 2: Factory functions with provider strings
red_hint_giver = create_hint_giver("anthropic", "claude-haiku-4-5-20251001", Team.RED)
red_guesser = create_guesser("anthropic", "claude-haiku-4-5-20251001", Team.RED)

# Create and run game
game = Game(
    blue_hint_giver=blue_hint_giver,
    blue_guesser=blue_guesser,
    red_hint_giver=red_hint_giver,
    red_guesser=red_guesser
)

result = game.play()
print(f"Winner: {result.winner.value}")
print(f"Turns: {result.num_turns}")

# Check guess history
print(f"Blue guesser history: {blue_guesser.guess_history}")

# Reset for next game
blue_guesser.reset()
red_guesser.reset()