Spaces:
Build error
Build error
File size: 1,797 Bytes
01523b5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 |
from __future__ import annotations
from typing import TYPE_CHECKING, List
from agentverse.message import Message
from . import selector_registry as SelectorRegistry
from .base import BaseSelector
if TYPE_CHECKING:
from agentverse.environments import BaseEnvironment
@SelectorRegistry.register("classroom")
class ClassroomSelector(BaseSelector):
def select_message(
self, environment: BaseEnvironment, messages: List[Message]
) -> List[Message]:
selected = []
for message in messages:
if message.sender.startswith("Student"):
if message.content.startswith("[RaiseHand]"):
message.content = "[RaiseHand]"
selected.append(message)
elif message.content != "" or len(message.tool_response) > 0:
selected.append(message)
elif message.sender.startswith("Professor"):
# If the professor launch a group discussion, then we
# brutely discard the student's message in this turn
if message.content.startswith("[GroupDiscuss]"):
return [message]
selected.append(message)
# If some student speak while the professor is speaking, then
# we brutely discard the student's message in this turn
if (
len(selected) > 1
and selected[0].sender.startswith("Professor")
and selected[0].content != ""
):
filtered_selected = []
filtered_selected.append(selected[0])
for message in selected[1:]:
if message.content.startswith("[RaiseHand]"):
filtered_selected.append(message)
selected = filtered_selected
return selected
|