pycityagent 1.0.0__py3-none-any.whl → 2.0.0a1__py3-none-any.whl

pycityagent/brain/reason/social.py

@@ -1,148 +0,0 @@
-import json
-
-async def handleConve(wk, retrives):
-    """
-    对话处理
-    Conversation handler
-
-    Args:
-    - wk (WorkingMemory)
-    - retrives (dict): the MemoryRetrive output collection
-
-    Returns:
-    - (Tuple([list[int], list[str]])): (id, message)
-    """
-    soul = wk._agent.Soul
-    familiers = retrives['familiers']
-    ids = []
-    messages = []
-    if wk.enable_social:
-        # * 接受其他agent发送过来的消息
-        if wk.msg_agent_unhandle:
-            for from_id, content_dialog in wk.msg_agent_unhandle.items():
-                if from_id in wk.agent_converse_buffer.keys():
-                    # * 正在进行对话
-                    if content_dialog[0]['content'] == 'End':
-                        # * 接受到End, 说明对方认为对话已结束
-                        if from_id in wk.msg_agent_still.keys():
-                            del wk.msg_agent_still[from_id]
-                    else:
-                        # * 正常进行对话
-                        wk.agent_converse_buffer[from_id] += content_dialog
-                        dialog = wk.agent_converse_buffer[from_id]
-                        print(dialog)
-                        resp = soul.text_request(dialog)
-                        if resp == 'Finished':
-                            del wk.msg_agent_still[from_id]
-                            ids.append(from_id)
-                            messages.append('End')
-                        else:
-                            wk.agent_converse_buffer[from_id].append({
-                                'role': 'assistant',
-                                'content': resp
-                            })
-                            ids.append(from_id)
-                            messages.append(resp)
-                            print(f'回复: {resp}')
-                            wk.msg_agent_still[from_id] = 20
-                else:
-                    # * 之前没有进行过对话 - 他人主动问好
-                    # * 判断是否进行对话 - 目前只支持熟人对话
-                    if from_id in familiers.keys():
-                        wk.agent_converse_buffer[from_id].append({
-                            'role': 'system',
-                            'content': f'''和你对话的人是: {familiers[from_id]['name']}, 和你的关系是{familiers[from_id]['relation']} {familiers[from_id]['learned']}'''
-                        })
-                        wk.agent_converse_buffer[from_id] += content_dialog
-                        dialog = wk.agent_converse_buffer[from_id]
-                        resp = soul.text_request(dialog)
-                        if resp == 'Finished':
-                            # * 不理睬
-                            ids.append(from_id)
-                            messages.append('End')
-                        else:
-                            # * 回复
-                            wk.agent_converse_buffer[from_id].append({
-                                'role': 'assistant',
-                                'content': resp
-                            })
-                            ids.append(from_id)
-                            messages.append(resp)
-                            wk.msg_agent_still[from_id] = 20
-            wk.msg_agent_unhandle.clear()  # 清空
-        else:
-            # * 主动问好
-            # * 判断周围是否有熟人
-            familier_around = []
-            for person in wk.sence['persons']:
-                if person['id'] in familiers.keys():
-                    familier_around.append((person, familiers[person['id']]))
-            if len(familier_around) > 0:
-                # 周围有熟人
-                # * 感兴趣人物筛选
-                dialog = wk.most_interest_dialog+[{'role': 'user', 'content': f'''我的基本信息为: {wk._agent.Image.get_profile()}'''}]
-                familier_text = ''''''
-                for person in familier_around:
-                    familier_text += f'''[人物id: {person[0]['id']}, 姓名: {person[1]['name']}, 和我的关系: {person[1]['relation']}, 其他信息: {person[1]['learned']}]\n'''
-                dialog += [{'role': 'user', 'content': f'''周围人物信息: \n{familier_text}'''}]
-                resp = soul.text_request(dialog)
-                try:
-                    resp_dict = json.loads(resp)
-                    interested_id = familiers[resp_dict['id']]
-                    conve = resp_dict['conve']
-                    # * 是否发起对话
-                    if conve == '是':
-                        # 发起对话
-                        wk.agent_converse_buffer[interested_id].append({
-                            'role': 'system',
-                            'content': f'''和你对话的人是: {familiers[interested_id]['name']}, 和你的关系是{familiers[interested_id]['relation']} {familiers[interested_id]['learned']}'''
-                        })
-                        hello_dialog = wk.agent_converse_buffer[interested_id] + [{'role': 'user', 'content': f'''假设你主动向对方问好, 你会说什么?'''}]
-                        hello = soul.text_request(hello_dialog)
-                        wk.agent_converse_buffer[interested_id].append({'role': 'system', 'content': f'你主动问好说: {hello}'})
-                        ids.append(interested_id)
-                        messages.append(hello)
-                        wk.msg_agent_still[interested_id] = 20
-                except:
-                    print("社交信息理解失败")
-
-        # * 处理对话对象挂断或突然终止对话的情况
-        to_del = []
-        for id in wk.msg_agent_still.keys():
-            wk.msg_agent_still[id] -= 1
-            if wk.msg_agent_still[id] <= 0:
-                to_del.append(id)
-        for id in to_del:
-            del wk.msg_agent_still[id]
-
-    return (ids, messages)
-
-async def startConve(wk):
-    """
-    判断是否进入/保持对话状态
-    Whether enter or stay in 'conve' state
-
-    Args:
-    - wk (WorkingMemory)
-
-    Returns:
-    - (bool): if True, enter 'conve' state, else not
-    """
-    if len(wk.msg_agent_still) > 0:
-        return True
-    return False
-
-async def endConve(wk):
-    """
-    判断是否结束对话(所有对话)
-    Whther end conversations (all conversations)
-
-    Args:
-    - wk (WorkingMemory)
-
-    Returns:
-    - (bool): if True, all conversations are ended, else not
-    """
-    if len(wk.msg_agent_still) > 0:
-        return False
-    return True

pycityagent/brain/reason/trip.py

@@ -1,67 +0,0 @@
-from ..scheduler import *
-import time
-
-async def startTrip(wk):
-    """
-    判断是否开始行程
-    Whether start trip
-
-    Args:
-    - wk (WorkingMemory)
-
-    Returns:
-    - (bool): if True, start target trip, else not
-    """
-    now = wk.scheduler.now
-    if now == None:
-        return False
-    elif now.type.value != ScheduleType.TRIP.value:
-        return False
-    elif now.is_set:
-        return False
-    else:
-        time_now = wk.sence['time']
-        if now.time <= time_now:
-            return True
-        else:
-            return False
-        
-async def tripArrived(wk):
-    """
-    判断是否已到达目的地
-    Whether arrived target location
-
-    Args:
-    - wk (WorkingMemory)
-
-    Returns:
-    - (bool): if True, has arrived, else not
-    """
-    now = wk.scheduler.now
-    if now == None:
-        return False
-    elif now.type.value != ScheduleType.TRIP.value:
-        return False
-    else:
-        if now.arrived:
-            return True
-        return False
-    
-async def routeFailed(wk):
-    """
-    判断导航是否失败
-    Whether the route request failed
-
-    Args:
-    - wk (WorkingMemory)
-
-    Returns:
-    - (bool): if True, failed, else not
-    """
-    now = wk.scheduler.now
-    if now == None or now.type.value != ScheduleType.TRIP.value:
-        return False
-    elif now.route_failed:
-        wk._agent.Hub.Update([AgentMessage(wk._agent.Hub._agent_id, int(time.time()*1000), '抱歉, 当前行程导航请求失败, 已跳过该行程', None, None)])
-        return True
-    return False

pycityagent/brain/reason/user.py

@@ -1,122 +0,0 @@
-from pycitysim.apphub import AgentMessage
-import time
-import json
-import copy
-
-async def hasUserControl(wk) -> bool:
-    """
-    判断是否有用户指令
-    Whether there have user commond
-
-    Args:
-    - wk (WorkingMemory)
-
-    Returns:
-    - (bool): if True, have user commond, else not
-    """
-    soul = wk._agent.Soul
-    hub = wk._agent.Hub
-    if len(wk.msg_user_unhandle) > 0:
-        # * 理解用户对话目的
-        understand_dialog = copy.deepcopy(wk.understanding_prompt)
-        user_dialog = []
-        contin_flag = True
-        for user_message in wk.msg_user_unhandle:
-            content = user_message.content
-            sub = [{'role': 'user', 'content': content}]
-            user_dialog += sub
-            print(f'[用户消息]: {content}')
-            if '开' in content or '启' in content or 'enable' in content :
-                if '用户交互' in content or 'interaction' in content:
-                    wk._agent.enable_user_interaction()
-                    hub.Update([AgentMessage(hub._agent_id, int(time.time()*1000), '已启动用户交互功能', None, None)])
-                    contin_flag = False
-                elif wk.enable_user_interaction and ('街景' in content or 'streetview' in content):
-                    wk._agent.enable_streetview()
-                    hub.Update([AgentMessage(hub._agent_id, int(time.time()*1000), '已启动街景功能(周围存在街景信息时自动呈现于屏幕右下角)', None, None)])
-                    contin_flag = False
-                elif wk.enable_user_interaction and ('经济' in content or 'economy' in content):
-                    wk._agent.enable_economy_behavior()
-                    hub.Update([AgentMessage(hub._agent_id, int(time.time()*1000), '已启动经济行为模拟', None, None)])
-                    contin_flag = False
-                elif wk.enable_user_interaction and ('社交' in content or 'social' in content):
-                    wk._agent.enable_social_behavior()
-                    hub.Update([AgentMessage(hub._agent_id, int(time.time()*1000), '已启动社交行为模拟', None, None)])
-                    contin_flag = False
-            elif '关' in content or 'diable' in content or '闭' in content:
-                if '用户交互' in content or 'interaction' in content:
-                    wk._agent.disable_user_interaction()
-                    hub.Update([AgentMessage(hub._agent_id, int(time.time()*1000), '已关闭用户交互功能', None, None)])
-                    contin_flag = False
-                elif wk.enable_user_interaction and ('街景' in content or 'streetview' in content):
-                    wk._agent.disable_streetview()
-                    hub.Update([AgentMessage(hub._agent_id, int(time.time()*1000), '已关闭街景功能', None, None)])
-                    contin_flag = False
-                elif wk.enable_user_interaction and ('经济' in content or 'economy' in content):
-                    wk._agent.disable_economy_behavior()
-                    hub.Update([AgentMessage(hub._agent_id, int(time.time()*1000), '已关闭经济行为模拟', None, None)])
-                    contin_flag = False
-                elif wk.enable_user_interaction and ('社交' in content or 'social' in content):
-                    wk._agent.disable_social_behavior()
-                    hub.Update([AgentMessage(hub._agent_id, int(time.time()*1000), '已关闭社交行为模拟', None, None)])
-                    contin_flag = False
-                
-        if not contin_flag:
-            return False
-        if not wk.enable_user_interaction:
-            # * 关闭了交互功能
-            hub.Update([AgentMessage(hub._agent_id, int(time.time()*1000), '抱歉, 我无法处理您的信息, 可能是因为您关闭了用户交互通道或该Agent不支持用户交互', None, None)])
-            return False
-        understand_dialog += user_dialog
-        understand = soul.text_request(understand_dialog)
-
-        if understand == '对话':
-            wk.set_user_converse_background()
-            dialog = copy.deepcopy(wk.user_converse_buffer)
-            if wk.current_target != None:
-                dialog.insert(3, {'role': 'system', 'content': f'你目前的计划: {wk.current_target}'})
-            dialog += user_dialog
-            print(dialog)
-            resp = soul.text_request(dialog)
-            if "Error" in resp:
-                hub.Update([AgentMessage(hub._agent_id, int(time.time()*1000), "抱歉, 我暂时无法理解您的意思", None, None)])
-            else:
-                wk.user_converse_buffer += user_dialog
-                wk.user_converse_buffer += [{'role': 'assistant', 'content': resp}]
-                print(wk.user_converse_buffer)  # 添加到记忆中
-                hub.Update([AgentMessage(hub._agent_id, int(time.time()*1000), resp, None, None)])
-                wk.msg_user_unhandle = []
-            return False
-        elif understand == '控制':
-            commond_to_target_dialog = [
-                {'role': 'system', 'content': '你是一名语义理解大师, 你需要理解用户输入的目的, 并从第一视角[我]出发返回理解结果'},
-                {'role': 'system', 'content': '例如用户输入: 你现在去买菜, 你将输出: 我现在要去买菜'},
-                {'role': 'system', 'content': '例如用户输入: 你现在去公司工作, 你将输出: 我现在准备去公司工作'}
-            ]
-            for user_message in wk.msg_user_unhandle:
-                content = user_message.content
-                commond_to_target_dialog += [{'role': 'user', 'content': content}]
-            wk.current_target = soul.text_request(commond_to_target_dialog)
-            wk.has_user_command = True
-            wk.msg_user_unhandle = []
-            return True
-        else:
-            # * 无法理解
-            hub.Update([AgentMessage(hub._agent_id, int(time.time()*1000), '抱歉，我无法理解您的话，您可以尝试换一种表达方式以帮助我理解', None, None)])
-            wk.msg_user_unhandle = []
-            return False
-
-async def endUserControl(wk) -> bool:
-    """
-    判断用户控制是否已结束
-    Whether user control has ended
-
-    Args:
-    - wk (WorkingMemory)
-
-    Returns:
-    - (bool): if True, ended, else not
-    """
-    if wk.has_user_command == False:
-        return True
-    return False

pycityagent/brain/retrive/social.py

@@ -1,6 +0,0 @@
-async def getfamilier(social):
-    """
-    获取社交网络信息
-    Get Social Information
-    """
-    return social.familiers_dict