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

pycityagent/workflow/trigger.py

@@ -0,0 +1,66 @@
+import asyncio
+from typing import Any, Callable, Dict, List, Optional, Union
+import socket
+from ..memory import Memory
+    
+class EventTrigger:
+    """Base class for event triggers that wait for specific conditions to be met."""
+    
+    async def wait_for_trigger(self) -> None:
+        """Wait for the event trigger to be activated.
+        
+        Raises:
+            NotImplementedError: Subclasses must implement this method.
+        """
+        raise NotImplementedError
+    
+class MemoryChangeTrigger(EventTrigger):
+    """Event trigger that activates when a specific key in memory changes."""
+    
+    def __init__(self, memory: Memory, key: str) -> None:
+        """Initialize the memory change trigger.
+
+        Args:
+            memory (Memory): The memory object to watch.
+            key (str): The key in memory to monitor for changes.
+        """
+        self.memory = memory
+        self.key = key
+        self.trigger_event = asyncio.Event()
+        asyncio.create_task(self.memory.add_watcher(key, self.trigger_event.set))
+
+    async def wait_for_trigger(self) -> None:
+        """Wait for the memory change trigger to be activated."""
+        await self.trigger_event.wait()
+        self.trigger_event.clear()
+
+class PortMessageTrigger(EventTrigger):
+    """Event trigger that activates upon receiving a message on a specified UDP port."""
+    
+    def __init__(self, port: int) -> None:
+        """Initialize the port message trigger.
+
+        Args:
+            port (int): The UDP port to listen for incoming messages.
+        """
+        self.port = port
+        self.trigger_event = asyncio.Event()
+        self.message: Optional[str] = None
+        asyncio.create_task(self._listen_on_port())
+
+    async def _listen_on_port(self) -> None:
+        """Listen for incoming UDP messages on the specified port.
+
+        Activates the trigger event when a message is received.
+        """
+        with socket.socket(socket.AF_INET, socket.SOCK_DGRAM) as s:
+            s.bind(("", self.port))
+            while True:
+                data, _ = s.recvfrom(1024)
+                self.message = data.decode()
+                self.trigger_event.set()
+
+    async def wait_for_trigger(self) -> None:
+        """Wait for the port message trigger to be activated."""
+        await self.trigger_event.wait()
+        self.trigger_event.clear()

pycityagent-2.0.0a1.dist-info/METADATA

@@ -0,0 +1,208 @@
+Metadata-Version: 2.1
+Name: pycityagent
+Version: 2.0.0a1
+Summary: LLM-based城市环境agent构建库
+License: MIT
+Author: Yuwei Yan
+Author-email: pinkgranite86@gmail.com
+Requires-Python: >=3.10,<4.0
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Requires-Dist: Pillow (==11.0.0)
+Requires-Dist: Requests (==2.32.3)
+Requires-Dist: Shapely (==2.0.6)
+Requires-Dist: aiohttp (==3.10.10)
+Requires-Dist: citystreetview (==1.2.4)
+Requires-Dist: dashscope (==1.14.1)
+Requires-Dist: geojson (==3.1.0)
+Requires-Dist: gradio (>=5.7.1,<6.0.0)
+Requires-Dist: grpcio (==1.67.1)
+Requires-Dist: matplotlib (==3.8.3)
+Requires-Dist: mosstool (==1.0.24)
+Requires-Dist: networkx (==3.2.1)
+Requires-Dist: numpy (>=1.20.0,<2.0.0)
+Requires-Dist: openai (==1.54.3)
+Requires-Dist: poetry (>=1.2.2)
+Requires-Dist: protobuf (<=4.24.0)
+Requires-Dist: pycitydata (==1.0.0)
+Requires-Dist: pycityproto (==2.0.7)
+Requires-Dist: pyyaml (>=6.0.2,<7.0.0)
+Requires-Dist: sidecar (==0.7.0)
+Requires-Dist: zhipuai (>=2.1.5.20230904,<3.0.0.0)
+Description-Content-Type: text/markdown
+
+
+
+<div style="text-align: center; background-color: white; padding: 20px; border-radius: 30px;">
+  <img src="./static/cityagent_logo.png" alt="CityAgent Logo" width="200" style="display: block; margin: 0 auto;">
+  <h1 style="color: black; margin: 0; font-size: 3em;">CityAgent: LLM Agents in City</h1>
+</div>
+
+
+# 🚀 CityAgent
+![License](https://img.shields.io/badge/license-MIT-green) &ensp;
+[![Online Documentation](https://img.shields.io/badge/docs-online-blue)](https://docs.fiblab.net/pycityagent) &ensp;
+
+
+CityAgent is an advanced framework specifically designed for building intelligent agents in urban simulation environments. With CityAgent, you can easily create and manage agents, enabling complex urban scenarios to be modeled and simulated efficiently.
+
+## 🌟 Features
+- **Modular Design**: Plug-and-play components for agent behavior.
+- **Urban Environment Simulation**: Built to simulate diverse urban scenarios.
+- **LLM Integration**: Connects to language models for enriched agent behavior.
+- **Flexible Configuration**: YAML-based configuration for easy customization.
+
+## 📑 Table of Contents
+
+1. [News](#news)
+2. [Framework](#framework)
+3. [Setup](#setup)
+4. [QuickStart](#quickstart)
+5. [Contributing](#contributing)
+6. [License](#license)
+
+<a id="news"></a>
+## 📰 News
+
+- 📢 **11.10** - Initial update is now live!
+- 📢 **2.x version is not compatible* with 1.x version**
+
+Stay tuned for upcoming updates!
+
+<a id="framework"></a>
+## 🛠️ Framework
+
+CityAgent is built with a multi-layered architecture that allows users to create and manage intelligent agents for urban environments in a scalable and flexible manner. The framework is divided into several key layers, each responsible for different functionalities as depicted in the diagram below:
+
+<img src="./static/framework.png" alt="CityAgent Framework Overview" width="600" style="display: block; margin: 20px auto;">
+
+### Architectural Layers
+- **Model Layer**: Handles agent configuration, task definitions, logging setup, and result aggregation.
+  - **Task Configuration**: Defines agent behaviors and objectives.
+  - **Unified Execution**: Centralized entry point for agent processes.
+
+- **Agent Layer**: Implements multi-head workflows to manage agent actions.
+  - **Memory**: Stores agent-related information, such as location and motion.
+    - **Static Profiles**: Maintains unchanging agent attributes.
+    - **Custom Data Pool**: Functions as a working memory.
+  - **Multi-Head Workflow**: Supports both normal and event-driven modes.
+    - **Reason Block**: Utilizes LLMs to determine decisions based on context and tools.
+    - **Route Block**: Chooses the best path based on defined criteria using LLMs or rules.
+    - **Action Block**: Executes actions as per defined contexts and tools.
+
+- **Message Layer**: Facilitates agent communication through peer-to-peer (P2P), peer-to-group (P2G), and group chats.
+
+- **Environment Layer**: Manages interaction with the urban environment.
+  - **Environment Sensing**: Reads data from the environment.
+  - **Interaction Handling**: Writes or modifies environmental states.
+  - **Message Management**: Handles incoming and outgoing agent messages.
+
+- **LLM Layer**: Provides configuration and integration for using LLMs in the agent's workflow.
+  - **Prompting & Execution**: Supports model invocation and monitoring.
+  - **Model Support**: Compatible with various LLMs, such as OpenAI, Qwen, Deepseek, etc.
+
+- **Tool Layer**: Provides additional utilities to the agents.
+  - **String Processing**: Handles parsing and formatting.
+  - **Result Analysis**: Parses responses in formats like JSON or dictionaries.
+  - **Data Storage & Retrieval**: Includes ranking and search tools.
+
+<a id="setup"></a>
+## ⚙️ Setup
+
+You can set up CityAgent in two different ways:
+
+### 1. From Scratch
+
+Follow these steps to set up CityAgent from scratch by cloning the repository. The project is built using Python and managed with Poetry.
+
+1. **Clone the Repository**
+   ```bash
+   git clone [This Repository]
+   ```
+2. **Navigate to the Project Directory**
+   ```bash
+   cd pycityagent
+   ```
+3. **Install Poetry** (if not installed)
+   ```bash
+   curl -sSL https://install.python-poetry.org | python3 -
+   ```
+4. **Install Dependencies**
+   ```bash
+   poetry install
+   ```
+5. **Activate the Virtual Environment**
+   ```bash
+   poetry shell
+   ```
+
+### 2. Install via pip
+
+This method is not yet available. Stay tuned for future updates!
+
+<a id="quickstart"></a>
+## 🚀 QuickStart
+
+Get started with CityAgent in just a few minutes!
+
+### 1. Config Configuration
+CityAgent uses a configuration file written in `.yaml` format to manage settings for various components. Below is a sample configuration file (`config_template.yaml`) that showcases the structure:
+
+```yaml
+llm_request:
+  text_request:
+    request_type: openai
+    api_key: <YOUR_API_KEY>
+    model: gpt-4o
+  img_understand_request:
+    request_type: none
+    api_key: none
+    model: none
+  img_generate_request:
+    request_type: none
+    api_key: none
+    model: none
+
+simulator_request:
+  simulator:
+    server: https://api-opencity-2x.fiblab.net:58081
+  map_request:
+    mongo_uri: <MONGO_URI>
+    mongo_db: llmsim
+    mongo_coll: map_beijing5ring_withpoi_0424
+    cache_dir: ./cache
+  route_request:
+    server: http://api-opencity-2x.fiblab.net:58082
+  streetview_request:
+    engine: baidumap / googlemap
+    mapAK: baidumap api-key (if you use baidumap engine)
+    proxy: googlemap proxy (if you use googlemap engine)
+```
+
+### 2. Example Usage
+To get started quickly, please refer to the `examples` folder in the repository. It contains sample scripts and configurations to help you understand how to create and use agents in an urban simulation environment.
+
+<a id="contributing"></a>
+## 🤝 Contributing
+We welcome contributions from the community!.
+
+<a id="license"></a>
+## 📄 License
+
+CityAgent is licensed under the MIT License. See the [LICENSE](LICENSE) file for more details.
+
+---
+
+Feel free to reach out if you have any questions, suggestions, or want to collaborate!
+
+---
+
+> **Follow us**: Stay updated with the latest news and features by watching the repository.
+
+---
+

pycityagent-2.0.0a1.dist-info/RECORD

@@ -0,0 +1,65 @@
+pycityagent/__init__.py,sha256=n56bWkAUEcvjDsb7LcJpaGjlrriSKPnR0yBhwRfEYBA,212
+pycityagent/agent.py,sha256=hfJbvZ5kxJJ3AAWjGdEzy52Rid3T5fVq8Qm0djnfxUY,6005
+pycityagent/config.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+pycityagent/economy/__init__.py,sha256=aonY4WHnx-6EGJ4WKrx4S-2jAkYNLtqUA04jp6q8B7w,75
+pycityagent/economy/econ_client.py,sha256=qQb_kZneEXGBRaS_y5Jdoi95I8GyjKEsDSC4s6V6R7w,10829
+pycityagent/environment/__init__.py,sha256=awHxlOud-btWbk0FCS4RmGJ13W84oVCkbGfcrhKqihA,240
+pycityagent/environment/interact/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+pycityagent/environment/interact/interact.py,sha256=LNl_HlasoW5H72GiBnbQkCP51qDIp5dpk-EShA7RPWk,5946
+pycityagent/environment/message/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+pycityagent/environment/sence/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+pycityagent/environment/sence/static.py,sha256=fdBjHKacNiDCKhvQkc9WgEYYSO0peMC5lb8CcXl9iNc,29101
+pycityagent/environment/sidecar/__init__.py,sha256=RFbOf40aYBP4WwRpFkua5tlRE_OtMcMNyp1Lew_aaAU,235
+pycityagent/environment/sidecar/sidecarv2.py,sha256=beKlYZgt38EQbV1x6NWQo7xVXyB-5QHfbwJexyXu7Tg,3252
+pycityagent/environment/sim/__init__.py,sha256=tr0lFAUsbyfAN5W5GehTlJHTSDGLeseUqVJgZfWn6PY,668
+pycityagent/environment/sim/aoi_service.py,sha256=pYSsqdGBpfQvp0-umNJir_7Yph75OJqhLjFQ2QWhSP8,1211
+pycityagent/environment/sim/client.py,sha256=MHR7Hhu-Cr7B61d3K_QDpSJh4HrUU9JQf6Cu7_8pdsE,3493
+pycityagent/environment/sim/clock_service.py,sha256=YG_A_IA7iJp9rOQE3bCPgppzuzzUwWMCzlxSUdOy088,1326
+pycityagent/environment/sim/economy_services.py,sha256=_WirbF-vv4U1VK5oACiEC3Bse9Pia9HWNEw21suEnAs,7093
+pycityagent/environment/sim/lane_service.py,sha256=6ljWiX_GbZd-a-fEcktDZncxBkraLA2LDE_pVV99uLs,3896
+pycityagent/environment/sim/light_service.py,sha256=ezYGcV_PkZ6I-yv48LNEjsytwwwQx-1Qp2QCWXWIhdQ,4215
+pycityagent/environment/sim/person_service.py,sha256=nIvOsoBoqOTDYtsiThg07-4ZBgkTUDEbb3dHyOjzyb8,10652
+pycityagent/environment/sim/road_service.py,sha256=Pab182YRcrjLw3UcfoD1Hdd6O8XEdi6Q2hJKzFcpSWE,1272
+pycityagent/environment/sim/social_service.py,sha256=a9mGZm95EFUIKQJUwQi9f8anmtf2SK4XqGfE2W9IXSQ,2001
+pycityagent/environment/simulator.py,sha256=zF6jBJLYtv-xqmFPKfrO_Z2u3FHDjRZ6segZhJbe7M8,13289
+pycityagent/environment/utils/__init__.py,sha256=sYbR0Przx6eekKXoTDDNB6u-cOaKfqsF0NYypppHuK8,116
+pycityagent/environment/utils/geojson.py,sha256=Ieg8Bzw63kKhJlhDIOVDoh-wQO4Sbtoe47FtIOy5wWg,686
+pycityagent/environment/utils/grpc.py,sha256=6EJwKXXktIWb1NcUiJzIRmfrY0S03QAXXGcCDHqAT00,1998
+pycityagent/environment/utils/map_utils.py,sha256=oqrRgQICC3SYw6gwjjPe_MAif7_t6dlrQpY8E32Fexs,5777
+pycityagent/environment/utils/protobuf.py,sha256=0jBvK_s96y_n7tuMbG22TOtQmg71SGV4ONDy2IGsU9o,1148
+pycityagent/llm/__init__.py,sha256=0-_q-StY0g4p9mdUNve_kqzHNXwVxedNsvxGXSQYq6o,144
+pycityagent/llm/embedding.py,sha256=vtjMUuInrtn0WRPXp3cmYEhwHXJ0q6zszdXsjTsgNeQ,4499
+pycityagent/llm/llm.py,sha256=PFbGCVQOLNJEfd5KaZdXABYIoPTFOGrNXIAE2eDRCP4,19191
+pycityagent/llm/llmconfig.py,sha256=WcyyfjP6XVLCDj2aemEM6ogQCWNxbTXQBiR0WjN4E8E,457
+pycityagent/llm/utils.py,sha256=emio-WhYh6vYb5Sp7KsnW9hKKe_jStjJsGBBJfcAPgM,153
+pycityagent/memory/__init__.py,sha256=Hs2NhYpIG-lvpwPWwj4DydB1sxtjz7cuA4iDAzCXnjI,243
+pycityagent/memory/const.py,sha256=zsyz282_8gH7MiFwBlnfN8RUh7Z_JcbP7H_dtJDmnK8,888
+pycityagent/memory/memory.py,sha256=I5GJ77eQZBcKcWevZQcHUBTd2XNT4CotE6mmvB5mA8I,18170
+pycityagent/memory/memory_base.py,sha256=U6odnR_5teaT3N2XF0W2McCxPyxtCpw61DjRVIO6p6I,5639
+pycityagent/memory/profile.py,sha256=dJPeZ91DZcYdR7zmsfSSQx7epvvE1mrvj4-4WAT0tkM,5217
+pycityagent/memory/self_define.py,sha256=vH2PrRSIykAOc5FebKN2JFQdx17nfR-IWv4s_ZS0YJk,5214
+pycityagent/memory/state.py,sha256=xqOxo69sZX1M9eCUMbg5BUlSFe9LdMDH3oTWDXC0y64,5148
+pycityagent/memory/utils.py,sha256=97lkenn-36wgt7uWb3Z39BXdJ5zlEQTQnQBFpoND1gg,879
+pycityagent/message/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+pycityagent/simulation/__init__.py,sha256=SZQzjcGR-zkTDrE81bQuEdKn7yjk4BcZ9m7o1wTc7EE,111
+pycityagent/simulation/interview.py,sha256=mY4Vpz0vgJo4rrMy3TZnwwM-iVDL6J0LgjOxbEuV27E,1173
+pycityagent/simulation/simulation.py,sha256=RXB0esX_0kS_gB8_zaVjPVQRHJYZ6iSYH44d_Uksf2s,10016
+pycityagent/simulation/survey/__init__.py,sha256=hFJ0Q1yo4jwKAIXP17sznBSWwm2Lyh3F3W3Lly40wr8,172
+pycityagent/simulation/survey/manager.py,sha256=DkNrb12Ay7TiGURoyJTFFeUdV1zh6TgRpTmpZOblADw,2158
+pycityagent/simulation/survey/models.py,sha256=-3EKe-qvkUJ2TH24ow0A_Lc4teGet7pueN2T5mOR_Qc,1308
+pycityagent/simulation/ui/__init__.py,sha256=4aevYl09KSku2NEpFZhyFNxI1qlswbrAFwhF_YhxFTo,62
+pycityagent/simulation/ui/interface.py,sha256=cSQRQhuA5Gj-X7anuN9tmRpG6-I8QYOuswjIWUeSvio,21636
+pycityagent/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+pycityagent/utils/decorators.py,sha256=v21QD_cKMTQ-t1UorXjgbWDFzeRehvprDalhQHIQxgg,3160
+pycityagent/utils/parsers/__init__.py,sha256=p5RZ4azJIpMgmKIU2evIp2moNzFabIR9X1LVNjT6Fq4,279
+pycityagent/utils/parsers/code_block_parser.py,sha256=Cs2Z_hm9VfNCpPPll1TwteaJF-HAQPs-3RApsOekFm4,1173
+pycityagent/utils/parsers/json_parser.py,sha256=FZ3XN1g8z4Dr2TFraUOoah1oQcze4fPd2m01hHoX0Mo,2917
+pycityagent/utils/parsers/parser_base.py,sha256=k6DVqwAMK3jJdOP4IeLE-aFPm3V2F-St5qRBuRdx4aU,1742
+pycityagent/workflow/__init__.py,sha256=dQCK7diOAQAn8Fo73rb6cnw-7u2NC55JWLW02bLdHLU,495
+pycityagent/workflow/block.py,sha256=rRL4ajiRZq3el4W0n6U4Ytb4WlN_4tH9OvVgFEuh73c,5093
+pycityagent/workflow/prompt.py,sha256=cmzKEmlzjdwg50uwfnTnN_6xNJA8OVjo5fdmcsaTbdU,2886
+pycityagent/workflow/tool.py,sha256=4sYbH-OVNffXbK3m58f8ap6WrS_uMzhK1c_Rl2er1B0,8981
+pycityagent/workflow/trigger.py,sha256=vweibjzGFex8IyRH7BbU-yarXcIgo6lt9v0UN3e7l5w,2352
+pycityagent-2.0.0a1.dist-info/METADATA,sha256=84-4cu5JguETBEEV63jJgKr_0XtYx78zS0ZDqpE79FE,7537
+pycityagent-2.0.0a1.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+pycityagent-2.0.0a1.dist-info/RECORD,,

{pycityagent-1.0.0.dist-info → pycityagent-2.0.0a1.dist-info}/WHEEL

@@ -1,5 +1,4 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.42.0)
+Generator: poetry-core 1.9.1
 Root-Is-Purelib: true
 Tag: py3-none-any
-

pycityagent/ac/__init__.py

@@ -1,6 +0,0 @@
-'''Action Control - 最终进行simulator以及AppHub对接'''
-
-from .ac import *
-from .action import *
-
-__all__ = [ActionController, Action]

pycityagent/ac/ac.py

@@ -1,50 +0,0 @@
-from typing import Any
-from .trip import TripAction
-from .shop import ShopAction
-from .converse import ConverseAction
-from .controled import ControledAction
-from .idle import IdleAction
-
-class ActionController:
-    """
-    Agent行为控制器: 与AppHub(simulator)对接
-    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:
-        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
-
-    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()
-
-    @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

pycityagent/ac/action.py

@@ -1,14 +0,0 @@
-from abc import ABC, abstractclassmethod
-from typing import Any
-
-class Action:
-    def __init__(self, agent) -> None:
-        '''默认初始化'''
-        self._agent = agent
-
-    @abstractclassmethod
-    async def Forward(self):
-        '''接口函数'''
-
-    async def __call__(self) -> Any:
-        await self.Forward()

pycityagent/ac/controled.py

@@ -1,13 +0,0 @@
-import time
-from .action import Action
-from pycitysim.apphub import AgentMessage
-
-class ControledAction(Action):
-    '''Converse行为控制器'''
-    def __init__(self, agent) -> None:
-        super().__init__(agent)
-
-    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/converse.py

@@ -1,31 +0,0 @@
-import time
-from .action import Action
-from pycitysim.apphub import AgentMessage
-
-class ConverseAction(Action):
-    '''Converse行为控制器'''
-    def __init__(self, agent) -> None:
-        super().__init__(agent)
-
-    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/idle.py

@@ -1,17 +0,0 @@
-import time
-from .action import Action
-from pycitysim.apphub import AgentMessage
-
-class IdleAction(Action):
-    '''idle行为控制器'''
-    def __init__(self, agent) -> None:
-        super().__init__(agent)
-
-    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/shop.py

@@ -1,80 +0,0 @@
-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) -> None:
-        super().__init__(agent)
-
-    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/trip.py

@@ -1,37 +0,0 @@
-import time
-from .action import Action
-from pycitysim.apphub import AgentMessage
-
-class TripAction(Action):
-    '''Trip行为控制器'''
-    def __init__(self, agent) -> None:
-        super().__init__(agent)
-
-    async def Forward(self):
-        now = self._agent.Scheduler.now
-        if now.is_set:
-            '''之前已经将schedule同步至模拟器了'''
-            if self._agent.Hub != None:
-                self._agent.Hub.Update(streetview=self._agent.Brain.Sence.sence_buffer['streetview'])
-        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/brain/__init__.py

@@ -1,10 +0,0 @@
-'''Agent大脑功能组织单位'''
-
-from .brain import *
-from .brainfc import *
-from .memory import *
-from .scheduler import *
-from .sence import *
-from .static import *
-
-__all__ = [Brain, BrainFunction, MemoryController, Memory, MemoryPersistence, MemoryRetrive, MemoryReason, MemoryType, ScheduleType, Scheduler, Sence, SencePlug, POI_TYPE_DICT]

pycityagent/brain/brain.py

@@ -1,52 +0,0 @@
-from abc import ABC, abstractmethod
-from typing import Optional
-from .scheduler import Scheduler
-from .sence import Sence
-from .memory import MemoryController
-
-class Brain:
-    """
-    大脑模块
-    Brain Module
-    """
-    def __init__(self, agent) -> None:
-        self._agent = agent
-        self._simulator = agent._simulator
-        self._sence = Sence(agent)
-        """
-        感知模块
-        Sence module
-        """
-        self._memory = MemoryController(agent)
-        """
-        记忆模块
-        Memory module
-        """
-
-    async def Run(self):
-        """
-        大脑主工作流
-        The main workflow of Brain
-        """
-        await self._sence.Sence()
-        await self._memory.Forward()
-
-    @property
-    def Agent(self):
-        return self._agent
-    
-    @property
-    def Sence(self):
-        return self._sence
-    
-    @property
-    def Memory(self):
-        return self._memory
-    
-    @property
-    def Simulator(self):
-        return self._simulator
-    
-    @property
-    def Hub(self):
-        return self._hub