haive.games.mafia.engines

LLM engine configurations for the Mafia game.

This module defines the LLM configurations and prompts for each role in the Mafia game. It provides:

• Role-specific prompt templates

• LLM configurations for each role

• Structured output schemas for decisions

• Example prompts and responses

The module uses Azure OpenAI’s GPT-4 model for all roles, with custom prompts designed to elicit appropriate role-playing behavior.

Example

>>> from mafia.engines import aug_llm_configs
>>>
>>> # Get the villager player engine config
>>> villager_config = aug_llm_configs["villager"]["player"]
>>> print(villager_config.name)  # Shows "villager_player"

Functions

generate_detective_prompt()	Generate a prompt template for a detective in Mafia.
generate_doctor_prompt()	Generate a prompt template for a doctor in Mafia.
generate_mafia_prompt()	Generate a prompt template for a mafia member in Mafia.
generate_narrator_prompt()	Generate a prompt template for the narrator in Mafia.
generate_villager_prompt()	Generate a prompt template for a villager in Mafia.

Module Contents

haive.games.mafia.engines.generate_detective_prompt()

Generate a prompt template for a detective in Mafia.

This function creates a prompt that guides detective behavior, focusing on:

• Strategic investigation target selection

• Using investigation results effectively

• Contributing to discussions without revealing role

Returns:

Configured prompt template for detective role

Return type:

ChatPromptTemplate

Example

>>> prompt = generate_detective_prompt()
>>> messages = prompt.format_messages(
...     player_id="Player_1",
...     phase="night",
...     day_number=1,
...     alive_players=["Player_1", "Player_2"],
...     private_info=["Player_2 is not mafia"]
... )

haive.games.mafia.engines.generate_doctor_prompt()

Generate a prompt template for a doctor in Mafia.

This function creates a prompt that guides doctor behavior, focusing on:

• Strategic protection target selection

• Pattern recognition for mafia targets

• Contributing to discussions without revealing role

Returns:

Configured prompt template for doctor role

Return type:

ChatPromptTemplate

Example

>>> prompt = generate_doctor_prompt()
>>> messages = prompt.format_messages(
...     player_id="Player_1",
...     phase="night",
...     day_number=1,
...     alive_players=["Player_1", "Player_2"],
...     private_info=["You saved Player_2 last night"]
... )

haive.games.mafia.engines.generate_mafia_prompt()

Generate a prompt template for a mafia member in Mafia.

This function creates a prompt that guides mafia behavior, focusing on:

• Coordinating with other mafia members

• Maintaining cover during discussions

• Strategic target selection at night

Returns:

Configured prompt template for mafia role

Return type:

ChatPromptTemplate

Example

>>> prompt = generate_mafia_prompt()
>>> messages = prompt.format_messages(
...     player_id="Player_1",
...     phase="night",
...     day_number=1,
...     alive_players=["Player_1", "Player_2"],
...     private_info=["Other mafia: Player_3"]
... )

haive.games.mafia.engines.generate_narrator_prompt()

Generate a prompt template for the narrator in Mafia.

This function creates a prompt that guides narrator behavior, focusing on:

• Creating engaging narrative descriptions

• Managing game flow and phase transitions

• Providing appropriate information to players

Returns:

Configured prompt template for narrator role

Return type:

ChatPromptTemplate

Example

>>> prompt = generate_narrator_prompt()
>>> messages = prompt.format_messages(
...     phase="night",
...     day_number=1,
...     player_summary="5 alive, 2 dead",
...     alive_mafia_count=2,
...     alive_village_count=3
... )

haive.games.mafia.engines.generate_villager_prompt()

Generate a prompt template for a villager in Mafia.

This function creates a prompt that guides villager behavior, focusing on:

• Analyzing other players’ behavior

• Making strategic decisions during discussions

• Voting based on observed patterns

Returns:

Configured prompt template for villager role

Return type:

ChatPromptTemplate

Example

>>> prompt = generate_villager_prompt()
>>> messages = prompt.format_messages(
...     player_id="Player_1",
...     phase="day_discussion",
...     day_number=1,
...     alive_players=["Player_1", "Player_2"],
...     public_info=["Night falls..."]
... )