pycityagent 1.0.0__py3-none-any.whl → 1.1.0__py3-none-any.whl

pycityagent/__init__.py

@@ -1,4 +1,6 @@
 from .simulator import Simulator
 from .agent import Agent
+from .agent_citizen import CitizenAgent
+from .agent_func import FuncAgent
 
-__all__ = [Simulator, Agent]
+__all__ = [Simulator, Agent, CitizenAgent, FuncAgent]

pycityagent/ac/__init__.py

@@ -2,5 +2,7 @@
 
 from .ac import *
 from .action import *
+from .hub_actions import *
+from .sim_actions import *
 
-__all__ = [ActionController, Action]
+__all__ = [ActionController, Action, HubAction, SimAction, SendUserMessage, SendStreetview, SendPop, PositionUpdate, ShowPath, ShowPosition, SetSchedule, SendAgentMessage]

pycityagent/ac/ac.py

@@ -1,9 +1,10 @@
-from typing import Any
-from .trip import TripAction
-from .shop import ShopAction
-from .converse import ConverseAction
-from .controled import ControledAction
-from .idle import IdleAction
+from typing import Any, Optional, Union
+from .action import Action
+from .citizen_actions.trip import TripAction
+from .citizen_actions.shop import ShopAction
+from .citizen_actions.converse import ConverseAction
+from .citizen_actions.controled import ControledAction
+from .citizen_actions.idle import IdleAction
 
 class ActionController:
     """
@@ -11,40 +12,56 @@ class ActionController:
     Agent Controller: Connect with AppHub and Simulator
     Note: Actions are predefined. By now, user defined actions are not supported
     """
-    def __init__(self, agent, config=None) -> None:
+    def __init__(self, agent) -> None:
         self._agent = agent
-        self._trip = TripAction(agent)
-        self._shop = ShopAction(agent)
-        self._converse = ConverseAction(agent)
-        self._control = ControledAction(agent)
-        self._idle = IdleAction(agent)
-        self._config = config
+        self.action_chain = {}
+        self.reset_ac()
+        self._init_actions()
 
     async def Run(self):
-        # TODO: 后期补充相关扩展接口，应该根据state-action关联模块进行（为了方便用户定义更加丰富的状态）
-        if self._agent.state == 'idle':
-            await self._idle()
-        elif self._agent.state == 'trip':
-            await self._trip()
-        elif self._agent.state == 'conve':
-            await self._converse()
-        elif self._agent.state == 'shop':
-            await self._shop()
-        elif self._agent.state == 'controled':
-            await self._control()
+        agent_state = self._agent.state
+        if agent_state in self.action_chain.keys():
+            for action in self.action_chain[agent_state]:
+                await action.Forward()
 
-    @property
-    def TripAction(self):
-        return self._trip
-    
-    @property
-    def ShopAction(self):
-        return self._shop
-    
-    @property
-    def ConveAction(self):
-        return self._converse
-    
-    @property
-    def ControlAction(self):
-        return self._controls
+    def clear_ac(self):
+        """
+        清空AC
+        """
+        self.action_chain.clear()
+
+    def reset_ac(self):
+        """
+        重置Action Controller
+        Reset action controller
+        """
+        self.action_chain.clear()
+        for state in self._agent._st.machine.states.keys():
+            self.action_chain[state] = []
+        
+    def set_ac(self, states:list[str]):
+        """
+        设置ac
+        """
+        self.action_chain.clear()
+        for state in states:
+            self.action_chain[state] = []
+
+    def add_actions(self, state:str, actions:Union[Action, list[Action]]):
+        """
+        添加Action到目标动作链
+        """
+        if issubclass(type(actions), Action) or isinstance(actions, Action):
+            self.action_chain[state].append(actions)
+        else:
+            self.action_chain[state] += actions
+
+    def _init_actions(self):
+        """
+        构建初始化动作链——适用于citizen类型的agent
+        """
+        self.action_chain['trip'] = [TripAction(self._agent)]
+        self.action_chain['shop'] = [ShopAction(self._agent)]
+        self.action_chain['conve'] = [ConverseAction(self._agent)]
+        self.action_chain['controled'] = [ControledAction(self._agent)]
+        self.action_chain['idle'] = [IdleAction(self._agent)]

pycityagent/ac/action.py

@@ -1,14 +1,55 @@
 from abc import ABC, abstractclassmethod
-from typing import Any
+from typing import Callable, Any
 
+class ActionType:
+    """
+    行动类型枚举 所有行动本质上为数据推送
+    Action Type enumeration, all actions are essentially data push
+
+    Types:
+    - Sim = 1, 用于表示与模拟器对接的行动
+    - Hub = 2, 用于表示与AppHub(前端)对接的行动
+    - Comp = 3, 表示综合类型 (可能同时包含与Sim以及Hub的交互)
+    """
+    Sim = 1
+    Hub = 2
+    Comp = 3
 class Action:
-    def __init__(self, agent) -> None:
-        '''默认初始化'''
+    def __init__(self, agent, type:ActionType, source: str = None, before:Callable[[list], Any] = None) -> None:
+        '''
+        默认初始化
+        
+        Args:
+        - agent (Agent): the related agent
+        - type (ActionType)
+        - source (str): 数据来源, 默认为None, 如果为None则会从接收用户传入的数据作为Forward函数参数, 否则从WM.Reason数据缓存中取对应数据作为参数
+        - before (function): 数据处理方法, 用于当Reason缓存中的参数与标准格式不符时使用
+        '''
         self._agent = agent
+        self._type = type
+        self._source = source
+        self._before = before
+
+    def get_source(self):
+        """
+        获取source数据
+        """
+        if self._source != None:
+            source = self._agent.Brain.Memory.Working.Reason[self._source]
+            if self._before != None:
+                source = self._before(source)
+            return source
+        else:
+            return None
 
     @abstractclassmethod
     async def Forward(self):
         '''接口函数'''
 
-    async def __call__(self) -> Any:
-        await self.Forward()
+class SimAction(Action):
+    def __init__(self, agent, source: str = None, before: Callable[[list], Any] = None) -> None:
+        super().__init__(agent, ActionType.Sim, source, before)
+
+class HubAction(Action):
+    def __init__(self, agent, source: str = None, before: Callable[[list], Any] = None) -> None:
+        super().__init__(agent, ActionType.Hub, source, before)

pycityagent/ac/action_stream.py

@@ -0,0 +1,25 @@
+from abc import ABC, abstractclassmethod
+from typing import Any, Optional
+from action import Action
+
+class ActionStream:
+    """
+    可以将多个Action聚合在一起构成一个ActionStream, Stream中的Action会顺序执行
+    You can collect multiple Actions into a ActionStream, actions in stream will be activated in sequence
+    注意: 该部分内容目前没有使用
+    """
+    def __init__(self, actions:Optional[list[Action]]=[]) -> None:
+        self.actions = actions
+
+    def add_actions(self, actions:list[Action]):
+        """
+        添加Action至当前stream
+
+        Args:
+        - actions (list(Action)): a list of action
+        """
+        self.actions += actions
+
+    def __call__(self) -> Any:
+        for action in self.actions:
+            action()

pycityagent/ac/citizen_actions/controled.py

@@ -0,0 +1,16 @@
+import time
+from typing import Callable, Any
+
+from pycityagent.ac.action import ActionType
+from ..action import Action
+from pycitysim.apphub import AgentMessage
+
+class ControledAction(Action):
+    '''Converse行为控制器'''
+    def __init__(self, agent, sources: list[str] = None, before: Callable[[list], Any] = None) -> None:
+        super().__init__(agent, ActionType.Comp, sources, before)
+
+    async def Forward(self):
+        req = {'person_id': self._agent._id, 'schedules': []}
+        await self._agent._client.person_service.SetSchedule(req)
+        self._agent.Hub.Update([AgentMessage(self._agent.Hub._agent_id, int(time.time()*1000), '我已理解您的意思，正在修改我的行程', None, None)])

pycityagent/ac/citizen_actions/converse.py

@@ -0,0 +1,34 @@
+import time
+from typing import Callable, Any
+
+from pycityagent.ac.action import ActionType
+from ..action import Action
+from pycitysim.apphub import AgentMessage
+
+class ConverseAction(Action):
+    '''Converse行为控制器'''
+    def __init__(self, agent, sources: list[str] = None, before: Callable[[list], Any] = None) -> None:
+        super().__init__(agent, ActionType.Comp, sources, before)
+
+    async def Forward(self):
+        target_agent_ids = self._agent.Brain.Memory.Working.Reason['agent_message_handle_resp'][0]
+        if len(target_agent_ids) == 0:
+            return
+        messages = self._agent.Brain.Memory.Working.Reason['agent_message_handle_resp'][1]
+        req = {'messages': []}
+        if len(target_agent_ids) != len(messages):
+            print("Warning: the number of target agent and message are not aligned, only sends matched messages")
+        rng = min(len(target_agent_ids), len(messages))
+        for i in range(rng):
+            dic = {}
+            dic['from'] = self._agent._id
+            dic['to'] = target_agent_ids[i]
+            dic['message'] = messages[i]
+            req['messages'].append(dic)
+        # * 发送至模拟器
+        await self._agent._client.social_service.Send(req=req)
+
+        # * 发送至AppHub
+        if self._agent.Hub != None and messages[0] != 'End':
+            # * 将信息中的第一条不同至pop
+            self._agent.Hub.Update(pop=messages[0])

pycityagent/ac/citizen_actions/idle.py

@@ -0,0 +1,20 @@
+import time
+from typing import Callable, Any
+
+from pycityagent.ac.action import ActionType
+from ..action import Action
+from pycitysim.apphub import AgentMessage
+
+class IdleAction(Action):
+    '''idle行为控制器'''
+    def __init__(self, agent, sources: list[str] = None, before: Callable[[list], Any] = None) -> None:
+        super().__init__(agent, ActionType.Comp, sources, before)
+
+    async def Forward(self):
+        if len(self._agent.base['schedules']) > 0:
+            req = {'person_id': self._agent._id, 'schedules': []}
+            await self._agent._client.person_service.SetSchedule(req)
+        if self._agent.Hub != None:
+            self._agent.Hub.Update()
+
+    

pycityagent/ac/citizen_actions/shop.py

@@ -0,0 +1,82 @@
+from typing import Callable, Any
+from pycityagent.ac.action import ActionType
+from ..action import Action
+from pycitysim.apphub import AgentMessage
+import time
+
+def encap_msg(msg, role='user', **kwargs):
+    dialog = {'role': role, 'content': msg}
+    dialog.update(kwargs)
+    return dialog
+
+class ShopAction(Action):
+    '''Shop行为控制器'''
+    def __init__(self, agent, sources: list[str] = None, before: Callable[[list], Any] = None) -> None:
+        super().__init__(agent, ActionType.Comp, sources, before)
+
+    async def Forward(self):
+        # * 与模拟器对接 - 暂时没有
+        # * 与AppHub对接
+        profile = self._agent.Image.get_profile()
+        self.consumption(profile)
+
+    def consumption(self, profile, mall_info):
+        dialogs = []
+        system_prompt = f'''
+                            你是一个在北京工作和生活的人。 
+                            {profile} 
+                            现在是2024年1月。
+                            你需要为下一周的基本生活购买必需品。
+                        '''
+        dialogs.append(encap_msg(system_prompt, 'system'))
+        actions_format = ['''{{'商品': 购买的商品的列表,'购买量': 每个商品的购买量，一个列表, '解释': '这种购买方式的原因'}}''']
+        actions_candidates = ['''【食品】
+                                米：10元/公斤
+                                面粉：7.5元/公斤
+                                新鲜蔬菜（如菠菜）：7元/500克
+                                水果（如苹果）：15元/公斤
+                                猪肉：30元/公斤
+                                鸡肉：20元/公斤
+                                鸡蛋：1.5元/个
+                                牛奶：10元/升''',
+                                '''【清洁用品】
+                                    洗衣液：30元/瓶
+                                    洗洁精：20元/瓶
+                                    垃圾袋：0.3元/个''',
+                                '''【个人护理用品】
+                                    牙膏：10元/支
+                                    洗发水：30元/瓶
+                                    沐浴露：35元/瓶
+                                    面巾纸：5元/包''',
+                                '''【其他】
+                                    矿泉水：1.7元/瓶
+                                    面包：8元/个
+                                    辣条：3元/包''']
+
+        user_prompt = f'''
+                        首先确定你的消费预算。以如下格式回答，不要有冗余的文本！
+                        {{'消费预算': 一个代表购买必需品需要消耗的钱的数量的数字}}
+                        '''
+        dialogs.append(encap_msg(user_prompt))
+        # * 对接一：确定消费预算
+        self._agent.Hub.Update([AgentMessage(self._agent.Hub._agent_id, int(time.time()*1000), '我正在确定消费预算......', None, None)])
+        msg = self._agent._soul.text_request(dialogs)
+        self._agent.Hub.Update([AgentMessage(self._agent.Hub._agent_id, int(time.time()*1000), f'我的消费预算是: {msg}'), None, None])
+
+        dialogs.append(encap_msg(msg, 'assistant'))
+
+        # * 对接二：购物选择
+        for cand in actions_candidates:
+            user_prompt = f'''
+                            购物中心里有
+                            {cand}
+                            你要买哪些商品，以如下格式回答，不要有冗余的文本！
+                            {actions_format[0]}
+                        '''
+            self._agent.Hub.Update([AgentMessage(self._agent.Hub._agent_id, int(time.time()*1000), f'我看到了\n {cand}', None, None)])
+            dialogs.append(encap_msg(user_prompt))
+            msg = self._agent._soul.text_request(dialogs)
+            self._agent.Hub.Update([AgentMessage(self._agent.Hub._agent_id, int(time.time()*1000), f'我的购买选择是: {msg}', None, None)])
+            dialogs.append(encap_msg(msg, 'assistant'))
+        
+        self._agent.Hub.Update([AgentMessage(self._agent.Hub._agent_id, int(time.time()*1000), '购物完成', None, None)])

pycityagent/ac/citizen_actions/trip.py

@@ -0,0 +1,41 @@
+import time
+from typing import Callable, Any
+
+from pycityagent.ac.action import ActionType
+from ..action import Action
+from pycitysim.apphub import AgentMessage
+
+class TripAction(Action):
+    '''Trip行为控制器'''
+    def __init__(self, agent, sources: list[str] = None, before: Callable[[list], Any] = None) -> None:
+        super().__init__(agent, ActionType.Comp, sources, before)
+
+    async def Forward(self):
+        now = self._agent.Scheduler.now
+        if now.is_set:
+            # TODO: 目前仅支持传输一张图片至前端
+            '''之前已经将schedule同步至模拟器了'''
+            if self._agent.Hub != None:
+                self._agent.Hub.Update(streetview=self._agent.Brain.Sence.sence_buffer['streetview'][0])
+        else:
+            '''同步schedule至模拟器'''
+            self._agent.Scheduler.now.is_set = True
+            departure_time = now.time
+            mode = now.mode
+            aoi_id = now.target_id_aoi
+            poi_id = now.target_id_poi
+            end = {'aoi_position': {'aoi_id': aoi_id, 'poi_id': poi_id}}
+            activity = now.description
+            trips = [{'mode': mode, 'end': end, 'departure_time': departure_time, 'activity': activity}]
+            set_schedule = [{'trips': trips, 'loop_count': 1, 'departure_time': departure_time}]
+
+            # * 与模拟器对接
+            req = {'person_id': self._agent._id, 'schedules': set_schedule}
+            await self._agent._client.person_service.SetSchedule(req)
+
+            # * 与AppHub对接
+            if self._agent.Hub != None:
+                messages = [AgentMessage(self._agent.Hub._agent_id, int(time.time()*1000), f'已到达出发时间, {activity}', None, None)]
+                self._agent.Hub.Update(messages)
+
+    

pycityagent/ac/hub_actions.py

@@ -0,0 +1,93 @@
+from typing import Callable, Optional, Any
+from pycitysim.apphub import AgentMessage
+from .action import HubAction
+from PIL.Image import Image
+
+class SendUserMessage(HubAction):
+    """
+    发送用户可见信息
+    Send messages to user
+    """
+    def __init__(self, agent, source: str = None, before: Callable[[list], Any] = None) -> None:
+        super().__init__(agent, source, before)
+
+    async def Forward(self, messages: Optional[list[AgentMessage]] = None):
+        if not messages == None:
+            self._agent.Hub.Update(messages = messages)
+        elif self._source != None:
+            messages = self.get_source()
+            if messages != None:
+                self._agent.Hub.Update(messages = messages)
+
+class SendStreetview(HubAction):
+    """
+    发送街景图片至前端
+    Send streetview to frontend
+    """
+    def __init__(self, agent, source: str = None, before: Callable[[list], Any] = None) -> None:
+        super().__init__(agent, source, before)
+
+    async def Forward(self, images: Optional[list[Image]] = None):
+        # TODO: 目前只支持发送一张图片
+        if not images == None:
+            self._agent.Hub.Update(streetview = images[0])
+        elif self._source != None:
+            images = self.get_source()
+            if images != None:
+                self._agent.Hub.Update(streetview = images[0])
+
+class SendPop(HubAction):
+    """
+    发送Pop信息 - agent头顶气泡信息
+    Send pop message to frontend, pop message shows above the target agent
+    """
+    def __init__(self, agent, source: str = None, before: Callable[[list], Any] = None) -> None:
+        super().__init__(agent, source, before)
+
+    async def Forward(self, pop:Optional[str] = None):
+        if not pop == None:
+            self._agent.Hub.Update(pop = pop)
+        elif self._source != None:
+            pop = self.get_source()
+            if pop != None:
+                self._agent.Hub.Update(pop = pop)
+
+class PositionUpdate(HubAction):
+    """
+    同步当前agent的位置信息
+    Send the position of current agent to frontend
+    """
+    def __init__(self, agent, source: str = None, before: Callable[[list], Any] = None) -> None:
+        super().__init__(agent, source, before)
+
+    async def Forward(self, longlat: Optional[list[float]] = None):
+        if longlat != None:
+            self._agent.Hub.Update(longlat = longlat)
+        elif self._source != None:
+            longlat = self.get_source()
+            if longlat != None:
+                self._agent.Hub.Update(longlat = longlat)
+
+class ShowPath(HubAction):
+    """
+    在地图上展示路径信息
+    Show a path in frontend map
+    """
+    def __init__(self, agent, source: str = None, before: Callable[[list], Any] = None) -> None:
+        super().__init__(agent, source, before)
+
+    async def Forward(self):
+        # TODO: 相关功能完善
+        pass
+
+class ShowPosition(HubAction):
+    """
+    在地图上展示特定地点
+    Show a position in frontend map
+    """
+    def __init__(self, agent, source: str = None, before: Callable[[list], Any] = None) -> None:
+        super().__init__(agent, source, before)
+
+    async def Forward(self):
+        # TODO: 相关功能完善
+        pass

pycityagent/ac/sim_actions.py

@@ -0,0 +1,80 @@
+import time
+from typing import Callable, Optional, Any
+from .action import SimAction
+from ..brain.scheduler import TripSchedule
+
+class SetSchedule(SimAction):
+    """
+    用于将agent的行程信息同步至模拟器 —— 仅对citizen类型agent适用
+    Synchronize agent's schedule to simulator —— only avalable for citizen type of agent
+    """
+    def __init__(self, agent, sources: list[str] = None, before: Callable[[list], Any] = None) -> None:
+        super().__init__(agent, sources, before)
+
+    async def Forward(self, schedule: Optional[TripSchedule] = None):
+        """
+        如果当前行程已经同步至模拟器: 跳过同步, 否则同步至模拟器
+        If current schedule has been synchronized to simulator: skip, else sync
+        """
+        if not schedule == None:
+            if not schedule.is_set:
+                '''同步schedule至模拟器'''
+                self._agent.Scheduler.now.is_set = True
+                departure_time = schedule.time
+                mode = schedule.mode
+                aoi_id = schedule.target_id_aoi
+                poi_id = schedule.target_id_poi
+                end = {'aoi_position': {'aoi_id': aoi_id, 'poi_id': poi_id}}
+                activity = schedule.description
+                trips = [{'mode': mode, 'end': end, 'departure_time': departure_time, 'activity': activity}]
+                set_schedule = [{'trips': trips, 'loop_count': 1, 'departure_time': departure_time}]
+
+                # * 与模拟器对接
+                req = {'person_id': self._agent._id, 'schedules': set_schedule}
+                await self._agent._client.person_service.SetSchedule(req)
+        elif self._source != None:
+            schedule = self.get_source()
+            if schedule != None and not schedule.is_set:
+                '''同步schedule至模拟器'''
+                self._agent.Scheduler.now.is_set = True
+                departure_time = schedule.time
+                mode = schedule.mode
+                aoi_id = schedule.target_id_aoi
+                poi_id = schedule.target_id_poi
+                end = {'aoi_position': {'aoi_id': aoi_id, 'poi_id': poi_id}}
+                activity = schedule.description
+                trips = [{'mode': mode, 'end': end, 'departure_time': departure_time, 'activity': activity}]
+                set_schedule = [{'trips': trips, 'loop_count': 1, 'departure_time': departure_time}]
+
+                # * 与模拟器对接
+                req = {'person_id': self._agent._id, 'schedules': set_schedule}
+                await self._agent._client.person_service.SetSchedule(req)
+
+
+class SendAgentMessage(SimAction):
+    """
+    发送信息给其他agent
+    Send messages to other agents
+    """
+    def __init__(self, agent, sources: list[str] = None, before: Callable[[list], Any] = None) -> None:
+        super().__init__(agent, sources, before)
+
+    async def Forward(self, messages: Optional[dict] = None):
+        if not messages == None and len(messages) > 0:
+            req = {'messages': []}
+            for message in messages:
+                from_id = self._agent._id
+                to_id = message['id']
+                mes = message['message']
+                req['messages'].append({'from': from_id, 'to': to_id, 'message': mes})
+            await self._agent._client.social_service.Send(req)
+        elif self._source != None:
+            messages = self.get_source()
+            if not messages == None and len(messages) > 0:
+                req = {'messages': []}
+                for message in messages:
+                    from_id = self._agent._id
+                    to_id = message['id']
+                    mes = message['message']
+                    req['messages'].append({'from': from_id, 'to': to_id, 'message': mes})
+                await self._agent._client.social_service.Send(req)