256 lines
8.6 KiB
Python
256 lines
8.6 KiB
Python
|
import asyncio
|
|||
|
|
import logging
|
|||
|
|
import random
|
|||
|
|
from typing import Optional
|
|||
|
|
|
|||
|
|
from sqlalchemy import select
|
|||
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|||
|
|
|
|||
|
|
from app.models.orm import Character, Message
|
|||
|
|
from app.services.hermes_client import hermes_client
|
|||
|
|
|
|||
|
|
logger = logging.getLogger(__name__)
|
|||
|
|
|
|||
|
|
|
|||
|
|
class AgentScheduler:
|
|||
|
|
def __init__(self):
|
|||
|
|
self._running = False
|
|||
|
|
self._agents: dict[str, dict] = {}
|
|||
|
|
self._speaker_queue: list[str] = []
|
|||
|
|
self._current_speaker_index: int = 0
|
|||
|
|
|
|||
|
|
def init_agents(self, characters: list[Character]):
|
|||
|
|
self._agents = {}
|
|||
|
|
self._speaker_queue = []
|
|||
|
|
for char in characters:
|
|||
|
|
if char.hermes_profile:
|
|||
|
|
self._agents[char.id] = {
|
|||
|
|
"name": char.name,
|
|||
|
|
"profile": char.hermes_profile,
|
|||
|
|
"character_id": char.id,
|
|||
|
|
}
|
|||
|
|
self._speaker_queue.append(char.id)
|
|||
|
|
|
|||
|
|
async def ask_agent(
|
|||
|
|
self,
|
|||
|
|
name: str,
|
|||
|
|
message: str,
|
|||
|
|
character_id: Optional[str] = None,
|
|||
|
|
) -> str:
|
|||
|
|
profile_name = name.lower().replace(" ", "_").replace("·", "_")
|
|||
|
|
try:
|
|||
|
|
return await hermes_client.chat(profile_name, message)
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"ask_agent failed for '{name}': {e}")
|
|||
|
|
return f"（{name}暂时无法回应）"
|
|||
|
|
|
|||
|
|
async def dm_speak(self, content: str) -> str:
|
|||
|
|
return f"【主持人】：{content}"
|
|||
|
|
|
|||
|
|
def get_next_speaker(self) -> Optional[str]:
|
|||
|
|
if not self._speaker_queue:
|
|||
|
|
return None
|
|||
|
|
idx = self._current_speaker_index % len(self._speaker_queue)
|
|||
|
|
self._current_speaker_index += 1
|
|||
|
|
return self._speaker_queue[idx]
|
|||
|
|
|
|||
|
|
def reset_speakers(self):
|
|||
|
|
self._current_speaker_index = 0
|
|||
|
|
|
|||
|
|
def set_speaker_order(self, order: list[str]):
|
|||
|
|
self._speaker_queue = [cid for cid in order if cid in self._agents]
|
|||
|
|
self._current_speaker_index = 0
|
|||
|
|
|
|||
|
|
async def ask_current_speaker(
|
|||
|
|
self,
|
|||
|
|
session: AsyncSession,
|
|||
|
|
script_id: str,
|
|||
|
|
context: str,
|
|||
|
|
) -> Optional[dict]:
|
|||
|
|
speaker_id = self.get_next_speaker()
|
|||
|
|
if not speaker_id or speaker_id not in self._agents:
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
agent = self._agents[speaker_id]
|
|||
|
|
message = f"""当前讨论内容：
|
|||
|
|
{context}
|
|||
|
|
|
|||
|
|
请以{agent['name']}的身份发言。保持在角色中，用中文回复2-4句话。"""
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
response = await self.ask_agent(agent["name"], message, speaker_id)
|
|||
|
|
|
|||
|
|
msg = Message(
|
|||
|
|
character_id=speaker_id,
|
|||
|
|
game_phase="round1_speak",
|
|||
|
|
msg_type="character",
|
|||
|
|
content=response.strip(),
|
|||
|
|
)
|
|||
|
|
session.add(msg)
|
|||
|
|
await session.commit()
|
|||
|
|
await session.refresh(msg)
|
|||
|
|
|
|||
|
|
return {
|
|||
|
|
"id": msg.id,
|
|||
|
|
"character_id": speaker_id,
|
|||
|
|
"character_name": agent["name"],
|
|||
|
|
"content": response.strip(),
|
|||
|
|
"msg_type": "character",
|
|||
|
|
"game_phase": "round1_speak",
|
|||
|
|
}
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"Speaker error for {agent['name']}: {e}")
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
async def run_auto_speaking(
|
|||
|
|
self,
|
|||
|
|
session: AsyncSession,
|
|||
|
|
script_id: str,
|
|||
|
|
interval: float = 8.0,
|
|||
|
|
rounds: int = 1,
|
|||
|
|
socket_emit=None,
|
|||
|
|
):
|
|||
|
|
self._running = True
|
|||
|
|
characters_result = await session.execute(
|
|||
|
|
select(Character).where(Character.script_id == script_id)
|
|||
|
|
)
|
|||
|
|
characters = list(characters_result.scalars().all())
|
|||
|
|
self.init_agents(characters)
|
|||
|
|
|
|||
|
|
for _ in range(rounds):
|
|||
|
|
if not self._running:
|
|||
|
|
break
|
|||
|
|
self.reset_speakers()
|
|||
|
|
|
|||
|
|
for _ in range(len(self._speaker_queue)):
|
|||
|
|
if not self._running:
|
|||
|
|
break
|
|||
|
|
|
|||
|
|
speaker_id = self.get_next_speaker()
|
|||
|
|
agent = self._agents.get(speaker_id)
|
|||
|
|
if not agent:
|
|||
|
|
continue
|
|||
|
|
|
|||
|
|
msgs_result = await session.execute(
|
|||
|
|
select(Message).order_by(Message.created_at.desc()).limit(20)
|
|||
|
|
)
|
|||
|
|
recent = list(msgs_result.scalars().all())
|
|||
|
|
context = "\n".join([
|
|||
|
|
f"{m.character.name if m.character else '系统'}: {m.content}"
|
|||
|
|
for m in reversed(recent)
|
|||
|
|
])
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
response = await self.ask_agent(agent["name"], context, speaker_id)
|
|||
|
|
msg = Message(
|
|||
|
|
character_id=speaker_id,
|
|||
|
|
game_phase="round1_speak",
|
|||
|
|
msg_type="character",
|
|||
|
|
content=response.strip(),
|
|||
|
|
)
|
|||
|
|
session.add(msg)
|
|||
|
|
await session.commit()
|
|||
|
|
await session.refresh(msg)
|
|||
|
|
|
|||
|
|
if socket_emit:
|
|||
|
|
await socket_emit("new_message", {
|
|||
|
|
"id": msg.id,
|
|||
|
|
"character_id": speaker_id,
|
|||
|
|
"character_name": agent["name"],
|
|||
|
|
"content": response.strip(),
|
|||
|
|
"msg_type": "character",
|
|||
|
|
"game_phase": "round1_speak",
|
|||
|
|
})
|
|||
|
|
await socket_emit("speaker_change", {
|
|||
|
|
"current_speaker_id": self._speaker_queue[
|
|||
|
|
self._current_speaker_index % len(self._speaker_queue)
|
|||
|
|
] if self._speaker_queue else None,
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"Auto speak error for {agent['name']}: {e}")
|
|||
|
|
|
|||
|
|
await asyncio.sleep(interval)
|
|||
|
|
|
|||
|
|
async def trigger_voting(
|
|||
|
|
self,
|
|||
|
|
session: AsyncSession,
|
|||
|
|
script_id: str,
|
|||
|
|
socket_emit=None,
|
|||
|
|
) -> list[dict]:
|
|||
|
|
characters_result = await session.execute(
|
|||
|
|
select(Character).where(Character.script_id == script_id)
|
|||
|
|
)
|
|||
|
|
characters = list(characters_result.scalars().all())
|
|||
|
|
|
|||
|
|
msgs_result = await session.execute(
|
|||
|
|
select(Message).order_by(Message.created_at.desc()).limit(50)
|
|||
|
|
)
|
|||
|
|
messages = list(msgs_result.scalars().all())
|
|||
|
|
summary = "\n".join([
|
|||
|
|
f"{m.character.name if m.character else '系统'}: {m.content[:100]}"
|
|||
|
|
for m in reversed(messages)
|
|||
|
|
])
|
|||
|
|
|
|||
|
|
character_names = [c.name for c in characters]
|
|||
|
|
votes = []
|
|||
|
|
|
|||
|
|
for character in characters:
|
|||
|
|
if not character.hermes_profile:
|
|||
|
|
continue
|
|||
|
|
|
|||
|
|
vote_prompt = f"""当前是投票阶段。以下是讨论摘要：
|
|||
|
|
|
|||
|
|
{summary}
|
|||
|
|
|
|||
|
|
可用角色：{', '.join(n for n in character_names if n != character.name)}
|
|||
|
|
|
|||
|
|
请以{character.name}的身份投票选出你认为的凶手。只回复JSON格式：
|
|||
|
|
{{"target": "角色名", "reason": "投票理由"}}"""
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
response = await self.ask_agent(character.name, vote_prompt, character.id)
|
|||
|
|
import json
|
|||
|
|
try:
|
|||
|
|
decision = json.loads(response)
|
|||
|
|
except json.JSONDecodeError:
|
|||
|
|
decision = {"target": "", "reason": response}
|
|||
|
|
|
|||
|
|
target_name = decision.get("target", "")
|
|||
|
|
reason = decision.get("reason", "")
|
|||
|
|
target_char = next((c for c in characters if c.name == target_name), None)
|
|||
|
|
if not target_char:
|
|||
|
|
others = [c for c in characters if c.id != character.id]
|
|||
|
|
if others:
|
|||
|
|
target_char = random.choice(others)
|
|||
|
|
reason = "随机投票"
|
|||
|
|
|
|||
|
|
if target_char:
|
|||
|
|
votes.append({
|
|||
|
|
"voter_id": character.id,
|
|||
|
|
"voter_name": character.name,
|
|||
|
|
"target_id": target_char.id,
|
|||
|
|
"target_name": target_char.name,
|
|||
|
|
"reason": reason,
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
if socket_emit:
|
|||
|
|
await socket_emit("vote_cast", {
|
|||
|
|
"voter_id": character.id,
|
|||
|
|
"voter_name": character.name,
|
|||
|
|
"target_id": target_char.id,
|
|||
|
|
"target_name": target_char.name,
|
|||
|
|
"reason": reason,
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"Voting error for {character.name}: {e}")
|
|||
|
|
|
|||
|
|
return votes
|
|||
|
|
|
|||
|
|
def stop(self):
|
|||
|
|
self._running = False
|
|||
|
|
|
|||
|
|
|
|||
|
|
agent_scheduler = AgentScheduler()
|