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

pycityagent/image/__init__.py

@@ -1,3 +0,0 @@
-from .image import *
-
-__all__ = [AgentImage, AgentScratch]

pycityagent/image/image.py

@@ -1,158 +0,0 @@
-from typing import Optional
-import json
-from abc import ABC, abstractclassmethod
-
-class ProfileTemplate:
-    """
-    profile模板类
-    The template class of Agent Profile
-    """
-    def __init__(self) -> None:
-        pass
-
-    @abstractclassmethod
-    def to_dialog(self) -> list[dict]:
-        """
-        将Profile转化为LLM可用的dialog的抽象函数
-        The abstract method that turns Agent's profile to LLM standard dialog
-        """
-
-class AgentImage:
-    """
-    Agent Image模块: 用户画像控制
-    Agent's Image module: Defines the uniqueness of Agent
-    """
-    def __init__(self, agent, scratch_file:str=None, selfie:bool=False) -> None:
-        self._agent = agent
-        self.scratch = AgentScratch()
-        """
-        Agent的Scratch信息: 用于描述Agent的基本属性
-        Scratch: Used to describe the basic info of Agent
-        - name (str): the name of Agent
-        - age (int): the age of Agent
-        - education (str): the education level, ['低等教育水平', '中等教育水平', '高等教育水平']
-        - gender (str): the gender of Agent, ['男', '女']
-        - consumption (str): the consumption level , ['低等消费水平', '中等消费水平', '高等消费水平']
-        - innate (str): the innate of Agent, '乐观积极' for instance
-        - learned (str): the learned info of Agent, '我是一名教师, 就职于xxx' for instance
-        - currently (str): the currently plan of Agent, '写一篇关于社会科学的研究报告' for instance
-        - lifestyle (str): the lifestyle of Agent, '早上7点起床, 晚上11点睡觉' for instance
-        """
-        self.selfie = None
-        """
-        Agent自画像
-        Selfie of Agent
-        Not implement yet!!!
-        """
-        profile = self._agent.base['profile']
-        if scratch_file != None:
-            self.scratch.load(scratch_file)
-        else:
-            self.scratch.name = self._agent.agent_name
-            self.scratch.age = profile['age']
-            education = profile['education']
-            if education < 3:
-                self.scratch.education = '低等教育水平'
-            elif education < 5:
-                self.scratch.education = '中等教育水平'
-            else:
-                self.scratch.education = '高等教育水平'
-            if profile['gender'] == 1:
-                self.scratch.gender = '男'
-            else:
-                self.scratch.gender = '女'
-            consumption = profile['consumption']
-            if consumption < 3:
-                self.scratch.consumption = '低等消费水平'
-            elif consumption < 5:
-                self.scratch.consumption = '高等消费水平'
-            else:
-                self.scratch.consumption = '高等消费水平'
-        if selfie:
-            # TODO: 补充自拍
-            pass
-
-    def get_profile(self):
-        """
-        获取Agent的Scratch Profile信息
-        Get the Scratch Profile message
-
-        Returns:
-        - (str)
-        """
-        return self.scratch.get_profile_content()
-    
-    def load_scratch(self, scratch_file:str):
-        """
-        加载基于文件的Profile加载
-        Load Profile based on file
-
-        Args:
-        - scratch_file (str): the path of scratch_file
-        """
-        self.scratch.load(scratch_file)
-
-    def load_selfie(self, selfie_file:str):
-        """
-        基于图像的Selfie加载
-        Load Selfie based on selfie file
-        Not implemented yet!!!
-
-        Args:
-        - selfie_file (str): the path of selfie_file
-        """
-        print("Not Implemented")
-
-class AgentScratch():
-    def __init__(self, scratch:Optional[dict]=None) -> None:
-        self.name = None
-        self.age = None
-        self.education = None
-        self.gender = None
-        self.consumption = None
-        self.innate = ''
-        self.learned = ''
-        self.currently = ''
-        self.lifestyle = ''
-        if scratch != None:
-            self.forward(scratch)
-
-    def forward(self, x:dict):
-        """Scratch记忆更新"""
-        for k in x.keys():
-            self.__dict__[k] = x[k]
-
-    def save(self, output_path:str):
-        """Scratch记忆持久化"""
-        with open(output_path, 'w') as f:
-            json.dumps(self.__dict__, f, 4)
-    
-    def load(self, x:str):
-        """Scratch记忆恢复"""
-        with open(x, 'r') as f:
-            json_str = f.read()
-            scratch = json.loads(json_str)
-            self.forward(scratch)
-    
-    def get_str_lifestyle(self):
-        return self.lifestyle
-    
-    def get_str_firstname(self):
-        return self.first_name
-    
-    def get_profile_content(self):
-        text = ''''''
-        text += f'姓名: {self.name}\n'
-        text += f'年龄: {self.age}\n'
-        text += f'性别: {self.gender}\n'
-        text += f'受教育水平: {self.education}\n'
-        text += f'消费能力: {self.consumption}\n'
-        text += f'性格特点: {self.innate}\n'
-        text += f'爱好与习惯: {self.learned}\n'
-        text += f'近期状态: {self.currently}\n'
-        text += f'生活习惯: {self.lifestyle}\n'
-        return text
-
-
-    def __str__(self):
-        return str(self.__dict__)

pycityagent/simulator.py

@@ -1,161 +0,0 @@
-from pycitysim import *
-from pycitysim.sidecar import OnlyClientSidecar
-from pycitysim.sim import CityClient
-from typing import Optional, Union
-from datetime import datetime, timedelta
-import asyncio
-from .agent import Agent
-
-class SimPerceive:
-    """
-    模拟器感知
-    Simulator Perceive
-    """
-    def __init__(self, simualtor) -> None:
-        self._simulator=simualtor
-    
-    async def PerceiveAoisByIds(self, ids:Optional[list[int]]):
-        """
-        Simulator视角下的AOI感知
-        Perceive AOI from Simulator
-
-        Args:
-        - ids list[int]: list of aoi id
-
-        Returns:
-        - https://cityproto.sim.fiblab.net/#city.map.v2.GetAoiResponse
-        """
-        req = {'aoi_ids': ids}
-        resp = await self._simulator._client.aoi_service.GetAoi(req)
-        return resp
-
-class Simulator:
-    """
-    模拟器
-    Simulator
-    """
-    def __init__(self, config) -> None:
-        self.config = config
-        self._client = CityClient(self.config['simulator']['server'], secure=True)
-        self._perceive = SimPerceive(self)
-        self.map = map.Map(
-            mongo_uri = "mongodb://sim:FiblabSim1001@mgo.db.fiblab.tech:8635/",
-            mongo_db = "srt",
-            mongo_coll = config['map_request']['mongo_coll'],
-            cache_dir = config['map_request']['cache_dir'],
-        )
-        self.time = 0
-
-    # * Agent相关
-    def FindAgentsByArea(self, req: dict, status=None):
-        """
-        通过区域范围查找agent/person
-        Get agents/persons in the provided area
-
-        Args:
-        - req (dict): 用于描述区域的请求 https://cityproto.sim.fiblab.net/#city.person.1.GetPersonByLongLatBBoxRequest
-        - status (int): 用于限制agent/person状态 if 'status' is not None, then you get those persons in 'status' https://cityproto.sim.fiblab.net/#city.agent.v2.Status
-
-        Returns:
-        - https://cityproto.sim.fiblab.net/#city.person.1.GetPersonByLongLatBBoxResponse
-        """
-        loop = asyncio.get_event_loop()
-        resp = loop.run_until_complete(self._client.person_service.GetPersonByLongLatBBox(req=req))
-        loop.close()
-        if status == None:
-            return resp
-        else:
-            motions = []
-            for agent in resp.motions:
-                if agent.status in status:
-                    motions.append(agent)
-            resp.motions = motions
-            return resp
-
-    async def GetAgent(self, name:str=None, id:int=None):
-        """
-        获取agent
-        Get Agent
-
-        Args:
-        - name str: 你为agent取的名字 the name of your agent
-        - id int: 即绑定的person的id the id of person that you try to bind with
-        
-        Returns:
-        - Agent
-        """
-        await self.GetTime()
-        name_ = "张三"
-        if name != None:
-            name_ = name
-        if id == None:
-            base = self._client.person_service.default_person()
-            agent = Agent(
-                name_, 
-                self.config['simulator']['server'], 
-                simulator=self, 
-                id=id, 
-                base=base,
-            )
-        else:
-            resp = await self._client.person_service.GetPerson({"person_id": id})
-            base = resp['base']
-            motion = resp['motion']
-            agent = Agent(
-                name_, 
-                self.config['simulator']['server'], 
-                simulator=self, 
-                id=id, 
-                base=base,
-                motion=motion,
-            )
-        if 'streetview_request' in self.config.keys():
-            agent.Brain.Sence._streetview_engine = self.config['streetview_request']['engine']
-            if agent.Brain.Sence._streetview_engine == 'baidumap':
-                agent.Brain.Sence._streetviewAK = self.config['streetview_request']['mapAK']
-            elif agent.Brain.Sence._streetview_engine == 'googlemap':
-                if 'proxy' in self.config['streetview_request'].keys():
-                    agent.Brain.Sence._streetviewProxy = self.config['streetview_request']['proxy']
-            else:
-                agent.Brain.Sence._streetview_engine = ""
-                print("Wrong Streetview Engine")
-        return agent
-
-    def InsertAgent(self, profile):
-        """
-        插入agent
-        Insert Agent
-        Not implemented yet
-        """
-        print("Not Implemented Yet")
-        pass
-        
-    async def GetTime(self, format_time:bool=False, format:Optional[str]="%H:%M:%S") -> Union[int, str]:
-        """
-        获取模拟器当前时间 Get current time of simulator
-        默认返回以00:00:00为始的, 以s为单位的时间(int)
-        支持格式化时间
-
-        Args:
-        - format_time (bool): 是否格式化 format or not
-        - format (str): 格式化模板，默认为"Hour:Minute:Second" the formation
-
-        Returns:
-        - time Union[int, str]: 时间 time in second(int) or formated time(str)
-        """
-        t_sec = await self._client.clock_service.Now({})
-        self.time = t_sec['t']
-        if format_time:
-            current_date = datetime.now().date()
-            start_of_day = datetime.combine(current_date, datetime.min.time())
-            current_time = start_of_day + timedelta(seconds=t_sec['t'])
-            formatted_time = current_time.strftime(format)
-            return formatted_time
-        else:
-            return t_sec['t']
-
-            
-    @property
-    def Perceive(self):
-        """模拟器感知模块"""
-        return self._perceive

pycityagent/st/__init__.py

@@ -1,4 +0,0 @@
-'''State Transformer - Agent状态控制机'''
-from .st import StateTransformer
-
-__all__ = [StateTransformer]

pycityagent/st/st.py

@@ -1,96 +0,0 @@
-from transitions import Machine
-from abc import ABC, abstractmethod
-
-class StateTransformer:
-    """
-    State Transformer模块: 控制agent状态转移
-    The State Transformer module: used to control the transformation of agent's state
-
-    Doc:
-    - 基于transitions()构建 https://github.com/tyarkoni/transitions
-    - Agent的状态不仅表征着用户当前的行为, 同时也起到了串联/控制Agent行为的功能 
-    - 状态信息:
-        - idle: 空闲状态
-        - trip: 出行状态
-        - shop: 购物状态
-        - conve: 社交状态
-        - paused: 暂停状态
-        - controled: 受控状态, 即用户通过OpenCity website console控制Agent
-    - 状态转移:
-        - idle:
-            - gotrip: idle -> trip
-            - goshop: idel -> shop
-            - goconverse: idel -> conve
-            - pause: idle -> paused
-            - gousercontrol: idle -> controled
-        - trip:
-            - arrived: trip -> idle
-            - routefailed: trip -> idle (导航失败)
-            - pause: idle -> paused
-            - gousercontrol: idle -> controled
-        - shop:
-            - shopdone: shop -> idle
-            - pause: idle -> paused
-            - gousercontrol: idle -> controled
-        - conve:
-            - convedone: conve -> idle
-            - pause: idle -> paused
-            - gousercontrol: idle -> controled
-        - paused:
-            - active*: paused -> *
-        - controled:
-            - controlback: controled -> idle
-    - Note: 目前不支持状态扩展
-    """
-    states = ['idle', 'trip', 'shop', 'conve', 'paused', 'controled']
-
-    def __init__(self, config=None):
-        self.pre_state = None
-        # TODO: 后续添加扩展接口
-        # Initialize the state machine
-        self.machine = Machine(model=self, states=StateTransformer.states, initial='idle')
-
-        # idle
-        self.machine.add_transition(trigger='gotrip', source='idle', dest='trip')
-        self.machine.add_transition(trigger='goshop', source='idle', dest='shop')
-        self.machine.add_transition(trigger='goconverse', source='idle', dest='conve')
-
-        # trip
-        self.machine.add_transition(trigger='arrived', source='trip', dest='idle')
-        self.machine.add_transition(trigger='routefailed', source='trip', dest='idle')
-
-        # shop
-        self.machine.add_transition(trigger='shopdone', source='shop', dest='idle')
-
-        # conve
-        self.machine.add_transition(trigger='convedone', source='conve', dest='idle')
-
-        # pause
-        self.machine.add_transition(trigger='pause', source='*', dest='paused', before='state_remember')
-
-        # active
-        self.machine.add_transition(trigger='active_trip', source='paused', dest='trip')
-        self.machine.add_transition(trigger='active_idle', source='paused', dest='idle')
-        self.machine.add_transition(trigger='active_shop', source='paused', dest='shop')
-        self.machine.add_transition(trigger='active_conve', source='paused', dest='conve')
-
-        # nothing
-        self.machine.add_transition(trigger='nothing', source='*', dest='=')
-
-        # controled
-        self.machine.add_transition(trigger='gousercontrol', source='*', dest='controled')
-        self.machine.add_transition(trigger='controlback', source='controled', dest='idle')
-
-    def state_remember(self):
-        if self.state != 'paused':  # 避免重复Pause
-            self.pre_state = self.state
-
-    def pause_back(self):
-        if self.pre_state == 'trip':
-            self.active_trip()
-        elif self.pre_state == 'shop':
-            self.active_shop()
-        elif self.pre_state == 'conve':
-            self.active_conve()
-        elif self.pre_sate == 'idle':
-            self.active_idle()

pycityagent/urbanllm/__init__.py

@@ -1,3 +0,0 @@
-from .urbanllm import *
-
-__all__ = [LLMConfig, UrbanLLM]

pycityagent/urbanllm/urbanllm.py

@@ -1,132 +0,0 @@
-from openai import OpenAI
-from http import HTTPStatus
-import dashscope
-from urllib.parse import urlparse, unquote
-from pathlib import PurePosixPath
-import requests
-from dashscope import ImageSynthesis
-from PIL import Image
-from io import BytesIO
-
-class LLMConfig:
-    """
-    大语言模型相关配置
-    The config of LLM
-    """
-    def __init__(
-        self, 
-        config: dict
-    ) -> None:
-        self.config = config
-        self.text = config['text_request']
-        self.image_u = config['img_understand_request']
-        self.image_g = config['img_generate_request']
-
-class UrbanLLM:
-    """
-    大语言模型对象
-    The LLM Object used by Agent(Soul)
-    """
-    def __init__(self, config: LLMConfig) -> None:
-        self.config = config
-
-    def text_request(self, dialog:list[dict]) -> str:
-        """
-        文本相关请求
-        Text request
-
-        Args:
-        - dialog (list[dict]): 标准的LLM文本dialog. The standard text LLM dialog
-
-        Returns:
-        - (str): the response content
-        """
-        if self.config.text['request_type'] == 'openai':
-            client = OpenAI(
-                api_key=self.config.text['api_key'], 
-                base_url=self.config.text['api_base']
-            )
-            response = client.chat.completions.create(
-                model=self.config.text['model'],
-                messages=dialog,
-                temperature=1.0,
-            )
-            return response.choices[0].message.content
-        elif self.config.text['request_type'] == 'qwen':
-
-            response = dashscope.Generation.call(
-                model=self.config.text['model'],
-                api_key=self.config.text['api_key'],
-                messages=dialog,
-                result_format='message'
-            )
-            if response.status_code == HTTPStatus.OK:
-                return response.output.choices[0]['message']['content']
-            else:
-                return "Error: {}".format(response.status_code)
-        else:
-            print("ERROR: Wrong Config")
-            return "wrong config"
-
-    def img_understand(self, img_path:str, prompt:str=None) -> str:
-        """
-        图像理解
-        Image understanding
-
-        Args:
-        - img_path: 图像的绝对路径. The absolute path of selected Image
-        - prompt: 理解提示词 - 例如理解方向. The understanding prompts
-
-        Returns:
-        - (str): the understanding content
-        """
-        ppt = "如何理解这幅图像？"
-        if prompt != None:
-            ppt += prompt
-        dialog = [{
-            'role': 'user',
-            'content': [
-                {'image': 'file://' + img_path},
-                {'text': ppt}
-            ]
-        }]
-        response = dashscope.MultiModalConversation.call(
-                    model=self.config.image_u['model'],
-                    api_key=self.config.image_u['api_key'],
-                    messages=dialog
-                )
-        if response.status_code == HTTPStatus.OK:
-            return response.output.choices[0]['message']['content']
-        else:
-            print(response.code)  # The error code.
-            return "Error"
-
-    def img_generate(self, prompt, size:str='512*512', quantity:int = 1):
-        """
-        图像生成
-        Image generation
-
-        Args:
-        - prompt (str): 图像生成提示词. The image generation prompts
-        - size (str): 生成图像尺寸, 默认为'512*512'. The image size, default: '512*512'
-        - quantity (int): 生成图像数量, 默认为1. The quantity of generated images, default: 1
-
-        Returns:
-        - (list[PIL.Image.Image]): 生成的图像列表. The list of generated Images.
-        """
-        rsp = ImageSynthesis.call(
-                model=self.config.image_g['model'],
-                api_key=self.config.image_g['api_key'],
-                prompt=prompt,
-                n=quantity,
-                size=size
-        )
-        if rsp.status_code == HTTPStatus.OK:
-            res = []
-            for result in rsp.output.results:
-                res.append(Image.open(BytesIO(requests.get(result.url).content)))
-            return res
-        else:
-            print('Failed, status_code: %s, code: %s, message: %s' %
-                (rsp.status_code, rsp.code, rsp.message))
-            return None

pycityagent-1.0.0.dist-info/LICENSE

@@ -1,21 +0,0 @@
-MIT License
-
-Copyright (c) 2024 FIBLAB
-
-Permission is hereby granted, free of charge, to any person obtaining a copy
-of this software and associated documentation files (the "Software"), to deal
-in the Software without restriction, including without limitation the rights
-to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
-copies of the Software, and to permit persons to whom the Software is
-furnished to do so, subject to the following conditions:
-
-The above copyright notice and this permission notice shall be included in all
-copies or substantial portions of the Software.
-
-THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
-OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
-SOFTWARE.