cs commited on
Commit
7264072
·
1 Parent(s): 713c200

Fix import errors - add required exports to prompt files

Browse files
werewolf/guard/prompt.py CHANGED
@@ -3,10 +3,25 @@ Guard Agent Prompts - Simplified
3
  """
4
 
5
  from werewolf.common_prompts import (
 
 
6
  OUTPUT_FORMAT_NAME_ONLY,
7
  OUTPUT_FORMAT_SPEECH,
8
  )
9
 
 
 
 
 
 
 
 
 
 
 
 
 
 
10
  DESC_PROMPT = f"""{{history}}
11
  You are {{name}}, the Guard. Protect key players.
12
 
 
3
  """
4
 
5
  from werewolf.common_prompts import (
6
+ GAME_RULE_PROMPT,
7
+ CLEAN_USER_PROMPT,
8
  OUTPUT_FORMAT_NAME_ONLY,
9
  OUTPUT_FORMAT_SPEECH,
10
  )
11
 
12
+ __all__ = [
13
+ "GAME_RULE_PROMPT",
14
+ "CLEAN_USER_PROMPT",
15
+ "DESC_PROMPT",
16
+ "VOTE_PROMPT",
17
+ "SKILL_PROMPT",
18
+ "SHERIFF_ELECTION_PROMPT",
19
+ "SHERIFF_SPEECH_PROMPT",
20
+ "SHERIFF_VOTE_PROMPT",
21
+ "SHERIFF_SPEECH_ORDER_PROMPT",
22
+ "SHERIFF_TRANSFER_PROMPT",
23
+ ]
24
+
25
  DESC_PROMPT = f"""{{history}}
26
  You are {{name}}, the Guard. Protect key players.
27
 
werewolf/seer/prompt.py CHANGED
@@ -3,10 +3,25 @@ Seer Agent Prompts - Simplified
3
  """
4
 
5
  from werewolf.common_prompts import (
 
 
6
  OUTPUT_FORMAT_NAME_ONLY,
7
  OUTPUT_FORMAT_SPEECH,
8
  )
9
 
 
 
 
 
 
 
 
 
 
 
 
 
 
10
  DESC_PROMPT = f"""{{history}}
11
  You are {{name}}, the Seer. Check players to find wolves.
12
 
 
3
  """
4
 
5
  from werewolf.common_prompts import (
6
+ GAME_RULE_PROMPT,
7
+ CLEAN_USER_PROMPT,
8
  OUTPUT_FORMAT_NAME_ONLY,
9
  OUTPUT_FORMAT_SPEECH,
10
  )
11
 
12
+ __all__ = [
13
+ "GAME_RULE_PROMPT",
14
+ "CLEAN_USER_PROMPT",
15
+ "DESC_PROMPT",
16
+ "VOTE_PROMPT",
17
+ "SKILL_PROMPT",
18
+ "SHERIFF_ELECTION_PROMPT",
19
+ "SHERIFF_SPEECH_PROMPT",
20
+ "SHERIFF_VOTE_PROMPT",
21
+ "SHERIFF_SPEECH_ORDER_PROMPT",
22
+ "SHERIFF_TRANSFER_PROMPT",
23
+ ]
24
+
25
  DESC_PROMPT = f"""{{history}}
26
  You are {{name}}, the Seer. Check players to find wolves.
27
 
werewolf/villager/prompt.py CHANGED
@@ -3,10 +3,24 @@ Villager Agent Prompts - Simplified
3
  """
4
 
5
  from werewolf.common_prompts import (
 
 
6
  OUTPUT_FORMAT_NAME_ONLY,
7
  OUTPUT_FORMAT_SPEECH,
8
  )
9
 
 
 
 
 
 
 
 
 
 
 
 
 
10
  DESC_PROMPT = f"""{{history}}
11
  You are {{name}}, a Villager. Find wolves through logic.
12
 
 
3
  """
4
 
5
  from werewolf.common_prompts import (
6
+ GAME_RULE_PROMPT,
7
+ CLEAN_USER_PROMPT,
8
  OUTPUT_FORMAT_NAME_ONLY,
9
  OUTPUT_FORMAT_SPEECH,
10
  )
11
 
12
+ __all__ = [
13
+ "GAME_RULE_PROMPT",
14
+ "CLEAN_USER_PROMPT",
15
+ "DESC_PROMPT",
16
+ "VOTE_PROMPT",
17
+ "SHERIFF_ELECTION_PROMPT",
18
+ "SHERIFF_SPEECH_PROMPT",
19
+ "SHERIFF_VOTE_PROMPT",
20
+ "SHERIFF_SPEECH_ORDER_PROMPT",
21
+ "SHERIFF_TRANSFER_PROMPT",
22
+ ]
23
+
24
  DESC_PROMPT = f"""{{history}}
25
  You are {{name}}, a Villager. Find wolves through logic.
26
 
werewolf/wolf/prompt.py CHANGED
@@ -3,10 +3,26 @@ Wolf Agent Prompts - Focused Strategy
3
  """
4
 
5
  from werewolf.common_prompts import (
 
 
6
  OUTPUT_FORMAT_NAME_ONLY,
7
  OUTPUT_FORMAT_SPEECH,
8
  )
9
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
10
  DESC_PROMPT = f"""{{history}}
11
  You are {{name}}, a Villager. (Don't reveal you're a wolf!)
12
 
 
3
  """
4
 
5
  from werewolf.common_prompts import (
6
+ GAME_RULE_PROMPT,
7
+ CLEAN_USER_PROMPT,
8
  OUTPUT_FORMAT_NAME_ONLY,
9
  OUTPUT_FORMAT_SPEECH,
10
  )
11
 
12
+ __all__ = [
13
+ "GAME_RULE_PROMPT",
14
+ "CLEAN_USER_PROMPT",
15
+ "DESC_PROMPT",
16
+ "VOTE_PROMPT",
17
+ "WOLF_SPEECH_PROMPT",
18
+ "KILL_PROMPT",
19
+ "SHERIFF_ELECTION_PROMPT",
20
+ "SHERIFF_SPEECH_PROMPT",
21
+ "SHERIFF_VOTE_PROMPT",
22
+ "SHERIFF_SPEECH_ORDER_PROMPT",
23
+ "SHERIFF_TRANSFER_PROMPT",
24
+ ]
25
+
26
  DESC_PROMPT = f"""{{history}}
27
  You are {{name}}, a Villager. (Don't reveal you're a wolf!)
28