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

pycityagent/workflow/trigger.py

@@ -0,0 +1,150 @@
+import asyncio
+from typing import Any, Callable, Dict, List, Optional, Union, Type
+import socket
+from ..memory import Memory
+from ..environment import Simulator
+
+KEY_TRIGGER_COMPONENTS = [Memory, Simulator]
+    
+class EventTrigger:
+    """Base class for event triggers that wait for specific conditions to be met."""
+    # 定义该trigger需要的组件类型
+    required_components: List[Type] = []
+    
+    def __init__(self, block=None):
+        self.block = block
+        if block is not None:
+            self.initialize()
+    
+    def initialize(self) -> None:
+        """Initialize the trigger with necessary dependencies."""
+        if not self.block:
+            raise RuntimeError("Block not set for trigger")
+            
+        # 检查所需组件是否都存在
+        missing_components = []
+        for component_type in self.required_components:
+            component_name = component_type.__name__.lower()
+            if not hasattr(self.block, component_name):
+                missing_components.append(component_type.__name__)
+        
+        if missing_components:
+            raise RuntimeError(
+                f"Block is missing required components for {self.__class__.__name__}: "
+                f"{', '.join(missing_components)}"
+            )
+    
+    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."""
+    required_components = [Memory]
+    
+    def __init__(self, key: str) -> None:
+        """Initialize the memory change trigger.
+
+        Args:
+            key (str): The key in memory to monitor for changes.
+        """
+        self.key = key
+        self.trigger_event = asyncio.Event()
+        self._initialized = False
+        super().__init__()
+
+    def initialize(self) -> None:
+        """Initialize the trigger with memory from block."""
+        super().initialize()  # 首先检查必需组件
+        self.memory = self.block.memory
+        asyncio.create_task(self.memory.add_watcher(self.key, self.trigger_event.set))
+        self._initialized = True
+
+    async def wait_for_trigger(self) -> None:
+        """Wait for the memory change trigger to be activated."""
+        if not self._initialized:
+            raise RuntimeError("Trigger not properly initialized")
+        await self.trigger_event.wait()
+        self.trigger_event.clear()
+
+
+class TimeTrigger(EventTrigger):
+    """Event trigger that activates periodically based on time intervals."""
+    required_components = [Simulator]
+    
+    def __init__(self, 
+                 days: Optional[int] = None, 
+                 hours: Optional[int] = None, 
+                 minutes: Optional[int] = None) -> None:
+        """Initialize the time trigger with interval settings.
+
+        Args:
+            days (Optional[int]): Execute every N days
+            hours (Optional[int]): Execute every N hours
+            minutes (Optional[int]): Execute every N minutes
+
+        Raises:
+            ValueError: If all interval parameters are None or negative
+        """
+        if all(param is None for param in (days, hours, minutes)):
+            raise ValueError("At least one time interval must be specified")
+        
+        # 验证参数有效性
+        for param_name, param_value in [('days', days), ('hours', hours), ('minutes', minutes)]:
+            if param_value is not None and param_value < 0:
+                raise ValueError(f"{param_name} cannot be negative")
+            
+        # 将所有时间间隔转换为秒
+        self.interval = 0
+        if days is not None:
+            self.interval += days * 24 * 60 * 60
+        if hours is not None:
+            self.interval += hours * 60 * 60
+        if minutes is not None:
+            self.interval += minutes * 60
+            
+        self.trigger_event = asyncio.Event()
+        self._initialized = False
+        self._monitoring_task = None
+        self._last_trigger_time = None
+        super().__init__()
+
+    def initialize(self) -> None:
+        """Initialize the trigger with necessary dependencies."""
+        super().initialize()  # 首先检查必需组件
+        self.memory = self.block.memory
+        self.simulator = self.block.simulator
+        # 启动时间监控任务
+        self._monitoring_task = asyncio.create_task(self._monitor_time())
+        self._initialized = True
+
+    async def _monitor_time(self):
+        """持续监控时间并在达到间隔时触发事件"""
+        # 第一次调用时直接触发
+        self.trigger_event.set()
+        
+        while True:
+            try:
+                current_time = await self.simulator.GetTime()
+                
+                # 如果是第一次或者已经过了指定的时间间隔
+                if (self._last_trigger_time is None or 
+                    current_time - self._last_trigger_time >= self.interval):
+                    self._last_trigger_time = current_time
+                    self.trigger_event.set()
+                
+                await asyncio.sleep(5)  # 避免过于频繁的检查
+            except Exception as e:
+                print(f"Error in time monitoring: {e}")
+                await asyncio.sleep(10)  # 发生错误时等待较长时间
+
+    async def wait_for_trigger(self) -> None:
+        """Wait for the time trigger to be activated."""
+        if not self._initialized:
+            raise RuntimeError("Trigger not properly initialized")
+        await self.trigger_event.wait()
+        self.trigger_event.clear()

pycityagent-2.0.0a2.dist-info/METADATA

@@ -0,0 +1,208 @@
+Metadata-Version: 2.1
+Name: pycityagent
+Version: 2.0.0a2
+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.0a2.dist-info/RECORD

@@ -0,0 +1,69 @@
+pycityagent/__init__.py,sha256=n56bWkAUEcvjDsb7LcJpaGjlrriSKPnR0yBhwRfEYBA,212
+pycityagent/agent.py,sha256=QzVBgTdlwZEY0pQQB9nTruWiIc3t8mZxLzv5cId1IbY,5951
+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=JVG6sSD2Hbohl1TtKjuQi7_M7tKMrFh9vl3QV3VA5O0,724
+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/sim_env.py,sha256=HI1LcS_FotDKQ6vBnx0e49prXSABOfA20aU9KM-ZkCY,4625
+pycityagent/environment/sim/social_service.py,sha256=a9mGZm95EFUIKQJUwQi9f8anmtf2SK4XqGfE2W9IXSQ,2001
+pycityagent/environment/simulator.py,sha256=lw6-T6HFeE6lOtbTQdKZLre2cX7PZkXftRt7wC4mE-8,11858
+pycityagent/environment/utils/__init__.py,sha256=PUx8etr2p_AA7F50ZR7g27odkgv-nOqFZa61ER8-DLg,221
+pycityagent/environment/utils/base64.py,sha256=hoREzQo3FXMN79pqQLO2jgsDEvudciomyKii7MWljAM,374
+pycityagent/environment/utils/const.py,sha256=3RMNy7_bE7-23K90j9DFW_tWEzu8s7hSTgKbV-3BFl4,5327
+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/port.py,sha256=3OM6kSUt3PxvDUOlgyiendBtETaWU8Mzk_8H0TzTmYg,295
+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=XiUlW9zmnvf4RRkDI6HghmueeOfIpkIKuTBzp9z6uB0,12669
+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=EyCcjB6LyBim-5iAOPe4m2qfvghEPqu1ZdGfy4KPeZ8,551
+pycityagent/workflow/block.py,sha256=IXfarqIax6yVP_DniU6ZsPTT8QA4aIDnvZbwP_MtRaw,6054
+pycityagent/workflow/prompt.py,sha256=cmzKEmlzjdwg50uwfnTnN_6xNJA8OVjo5fdmcsaTbdU,2886
+pycityagent/workflow/tool.py,sha256=4sYbH-OVNffXbK3m58f8ap6WrS_uMzhK1c_Rl2er1B0,8981
+pycityagent/workflow/trigger.py,sha256=u2aPBOzTWvdYKb45nUsQfA_uRunsmbLzB7ztSU2mlrQ,5736
+pycityagent-2.0.0a2.dist-info/METADATA,sha256=IrHbMpcjWubG-yxzVCPB50qjlaeQXq2UYxyksRd3P6U,7537
+pycityagent-2.0.0a2.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+pycityagent-2.0.0a2.dist-info/RECORD,,

{pycityagent-1.0.0.dist-info → pycityagent-2.0.0a2.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)])