update

.idea/workspace.xml

@@ -3,7 +3,6 @@
   <component name="ChangeListManager">
     <list default="true" id="be0bdb17-6f52-4720-b934-341fa7978d29" name="Changes" comment="">
       <change beforePath="$PROJECT_DIR$/.idea/workspace.xml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/workspace.xml" afterDir="false" />
-      <change beforePath="$PROJECT_DIR$/requirements.txt" beforeDir="false" afterPath="$PROJECT_DIR$/requirements.txt" afterDir="false" />
     </list>
     <option name="SHOW_DIALOG" value="false" />
     <option name="HIGHLIGHT_CONFLICTS" value="true" />
@@ -71,7 +70,7 @@
       <option name="number" value="Default" />
       <option name="presentableId" value="Default" />
       <updated>1745132074997</updated>
-      <workItem from="1745132078229" duration="3033000" />
+      <workItem from="1745132078229" duration="4105000" />
     </task>
     <servers />
   </component>
@@ -101,6 +100,6 @@
     </breakpoint-manager>
   </component>
   <component name="com.intellij.coverage.CoverageDataManagerImpl">
-    <SUITE FILE_PATH="coverage/werewolf_1$app.coverage" NAME="app Coverage Results" MODIFIED="1745137162588" SOURCE_PROVIDER="com.intellij.coverage.DefaultCoverageFileProvider" RUNNER="coverage.py" COVERAGE_BY_TEST_ENABLED="true" COVERAGE_TRACING_ENABLED="false" WORKING_DIRECTORY="$PROJECT_DIR$/werewolf" />
+    <SUITE FILE_PATH="coverage/werewolf_1$app.coverage" NAME="app Coverage Results" MODIFIED="1745138641306" SOURCE_PROVIDER="com.intellij.coverage.DefaultCoverageFileProvider" RUNNER="coverage.py" COVERAGE_BY_TEST_ENABLED="true" COVERAGE_TRACING_ENABLED="false" WORKING_DIRECTORY="$PROJECT_DIR$/werewolf" />
   </component>
 </project>

werewolf/app.py

@@ -11,9 +11,9 @@ from agent_build_sdk.sdk.werewolf_agent import WerewolfAgent
 if __name__ == '__main__':
     name = 'spy'
     agent = WerewolfAgent(name, model_name=os.getenv('MODEL_NAME'))
-    agent.register_role_agent(ROLE_VILLAGER, VillagerAgent())
-    agent.register_role_agent(ROLE_WOLF, WolfAgent())
-    agent.register_role_agent(ROLE_SEER, SeerAgent())
-    agent.register_role_agent(ROLE_WITCH, WitchAgent())
+    agent.register_role_agent(ROLE_VILLAGER, VillagerAgent(model_name=os.getenv('MODEL_NAME')))
+    agent.register_role_agent(ROLE_WOLF, WolfAgent(model_name=os.getenv('MODEL_NAME')))
+    agent.register_role_agent(ROLE_SEER, SeerAgent(model_name=os.getenv('MODEL_NAME')))
+    agent.register_role_agent(ROLE_WITCH, WitchAgent(model_name=os.getenv('MODEL_NAME')))
     agent_builder = AgentBuilder(name, agent=agent)
     agent_builder.start()

werewolf/seer/seer_agent.py

@@ -11,8 +11,8 @@ from seer.prompt import DESC_PROMPT, VOTE_PROMPT, SKILL_PROMPT
 class SeerAgent(BasicRoleAgent):
     """预言家角色Agent"""
 
-    def __init__(self, llm_caller=None, memory=None):
-        super().__init__(ROLE_SEER)
+    def __init__(self, model_name):
+        super().__init__(ROLE_SEER, model_name=model_name)
         self.memory.set_variable("checked_players", {})  # 存储已查验的玩家信息
 
     def perceive(self, req=AgentReq):

werewolf/villager/villager_agent.py

@@ -11,8 +11,8 @@ from agent_build_sdk.sdk.agent import format_prompt
 class VillagerAgent(BasicRoleAgent):
     """平民角色Agent"""
 
-    def __init__(self):
-        super().__init__(ROLE_VILLAGER)
+    def __init__(self, model_name):
+        super().__init__(ROLE_VILLAGER, model_name=model_name)
 
     def perceive(self, req=AgentReq):
         if req.status == STATUS_START:

werewolf/witch/witch_agent.py

@@ -11,8 +11,8 @@ from witch.prompt import DESC_PROMPT, VOTE_PROMPT, SKILL_PROMPT
 class WitchAgent(BasicRoleAgent):
     """女巫角色Agent"""
 
-    def __init__(self, llm_caller=None, memory=None):
-        super().__init__(ROLE_WITCH)
+    def __init__(self, model_name):
+        super().__init__(ROLE_WITCH, model_name=model_name)
         # 初始化女巫的两瓶药
         self.memory.set_variable("has_poison", True)
         self.memory.set_variable("has_antidote", True)

werewolf/wolf/wolf_agent.py

@@ -11,8 +11,8 @@ from wolf.prompt import DESC_PROMPT, VOTE_PROMPT, KILL_PROMPT, WOLF_SPEECH_PROMP
 class WolfAgent(BasicRoleAgent):
     """狼人角色Agent"""
 
-    def __init__(self, llm_caller=None, memory=None):
-        super().__init__(ROLE_WOLF)
+    def __init__(self, model_name):
+        super().__init__(ROLE_WOLF, model_name=model_name)
         self.memory.set_variable("teammates", [])  # 存储队友信息
 
     def perceive(self, req=AgentReq):