pycityagent 2.0.0a6__py3-none-any.whl → 2.0.0a7__py3-none-any.whl

pycityagent/agent.py

@@ -77,7 +77,7 @@ class Agent(ABC):
     def __getstate__(self):
         state = self.__dict__.copy()
         # 排除锁对象
-        del state['_llm_client']
+        del state["_llm_client"]
         return state
 
     async def bind_to_simulator(self):
@@ -278,12 +278,14 @@ class Agent(ABC):
     def get_interview_history(self) -> List[Dict]:
         """获取采访历史记录"""
         return self._interview_history
-    
+
     async def handle_message(self, payload: str):
         """处理收到的消息，识别发送者"""
         # 从消息中解析发送者 ID 和消息内容
         message, sender_id = payload.split("|from:")
-        print(f"Agent {self._agent_id} received message: '{message}' from Agent {sender_id}")
+        print(
+            f"Agent {self._agent_id} received message: '{message}' from Agent {sender_id}"
+        )
 
     async def send_message(self, to_agent_id: int, message: str):
         """通过 Messager 发送消息，附带发送者的 ID"""

pycityagent/environment/interact/interact.py

@@ -1,9 +1,11 @@
 """环境相关的Interaction定义"""
+
 from enum import Enum
 from typing import Callable, Optional, Any
 from abc import ABC, abstractmethod
 from typing import Callable, Any
 
+
 class ActionType(Enum):
     """
     行动类型枚举 所有行动本质上为数据推送
@@ -14,24 +16,33 @@ class ActionType(Enum):
     - Hub = 2, 用于表示与AppHub(前端)对接的行动
     - Comp = 3, 表示综合类型 (可能同时包含与Sim以及Hub的交互)
     """
+
     Sim = 1
     Hub = 2
     Comp = 3
-    
+
+
 class Action:
     """
     - Action
     """
-    def __init__(self, agent, type:ActionType, source: Optional[str] = None, before:Optional[Callable[[list], Any]] = None) -> None:
-        '''
+
+    def __init__(
+        self,
+        agent,
+        type: ActionType,
+        source: Optional[str] = None,
+        before: Optional[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
@@ -51,16 +62,30 @@ class Action:
 
     @abstractmethod
     async def Forward(self):
-        '''接口函数'''
+        """接口函数"""
+
 
 class SimAction(Action):
     """SimAction: 模拟器关联Action"""
-    def __init__(self, agent, source: Optional[str] = None, before:Optional[Callable[[list], Any]] = None) -> None:
+
+    def __init__(
+        self,
+        agent,
+        source: Optional[str] = None,
+        before: Optional[Callable[[list], Any]] = None,
+    ) -> None:
         super().__init__(agent, ActionType.Sim, source, before)
 
+
 class HubAction(Action):
     """HubAction: Apphub关联Action"""
-    def __init__(self, agent, source: Optional[str] = None, before:Optional[Callable[[list], Any]] = None) -> None:
+
+    def __init__(
+        self,
+        agent,
+        source: Optional[str] = None,
+        before: Optional[Callable[[list], Any]] = None,
+    ) -> None:
         super().__init__(agent, ActionType.Hub, source, before)
 
 
@@ -69,46 +94,70 @@ class SetSchedule(SimAction):
     用于将agent的行程信息同步至模拟器 —— 仅对citizen类型agent适用
     Synchronize agent's schedule to simulator —— only avalable for citizen type of agent
     """
-    def __init__(self, agent, source: Optional[str] = None, before:Optional[Callable[[list], Any]] = None) -> None:
+
+    def __init__(
+        self,
+        agent,
+        source: Optional[str] = None,
+        before: Optional[Callable[[list], Any]] = None,
+    ) -> None:
         super().__init__(agent, source, before)
 
-    async def Forward(self, schedule = None):
+    async def Forward(self, schedule=None):
         """
         如果当前行程已经同步至模拟器: 跳过同步, 否则同步至模拟器
         If current schedule has been synchronized to simulator: skip, else sync
         """
         if not schedule == None:
             if not schedule.is_set:
-                '''同步schedule至模拟器'''
+                """同步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}}
+                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}]
+                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}
+                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至模拟器'''
+                """同步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}}
+                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}]
+                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}
+                req = {"person_id": self._agent._id, "schedules": set_schedule}
                 await self._agent._client.person_service.SetSchedule(req)
 
 
@@ -117,25 +166,33 @@ class SendAgentMessage(SimAction):
     发送信息给其他agent
     Send messages to other agents
     """
-    def __init__(self, agent, source: Optional[str] = None, before:Optional[Callable[[list], Any]] = None) -> None:
+
+    def __init__(
+        self,
+        agent,
+        source: Optional[str] = None,
+        before: Optional[Callable[[list], Any]] = None,
+    ) -> None:
         super().__init__(agent, source, before)
 
     async def Forward(self, messages: Optional[dict] = None):
         if not messages == None and len(messages) > 0:
-            req = {'messages': []}
+            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})
+                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': []}
+                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})
+                    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)

pycityagent/environment/sence/static.py

@@ -1,6 +1,7 @@
 """
 静态数据支持; Static Resources: Poi Type association; Type prefix.
 """
+
 POI_TYPE_DICT = {
     "100000": "\u7f8e\u98df",
     "101000": "\u7f8e\u98df:\u4e2d\u9910\u5385",
@@ -409,7 +410,7 @@ POI_TYPE_DICT = {
     "801012": "\u5ba4\u5185\u53ca\u9644\u5c5e\u8bbe\u65bd:\u901a\u884c\u8bbe\u65bd\u7c7b:\u7535\u68af",
     "801099": "\u5ba4\u5185\u53ca\u9644\u5c5e\u8bbe\u65bd:\u901a\u884c\u8bbe\u65bd\u7c7b:\u5176\u5b83\u901a\u884c\u8bbe\u65bd\u7c7b",
     "809900": "\u5ba4\u5185\u53ca\u9644\u5c5e\u8bbe\u65bd:\u5176\u5b83\u5ba4\u5185\u53ca\u9644\u5c5e\u8bbe\u65bd",
-    "990000": "\u5176\u5b83"
+    "990000": "\u5176\u5b83",
 }
 
-LEVEL_ONE_PRE = ["1", "2", "8", "9"]
+LEVEL_ONE_PRE = ["1", "2", "8", "9"]

pycityagent/environment/sim/aoi_service.py

@@ -25,7 +25,7 @@ class AoiService:
         """
         获取AOI信息
         get AOI information
-        
+
         Args:
         - req (dict): https://cityproto.sim.fiblab.net/#city.map.v2.GetAoiRequest
 

pycityagent/environment/sim/economy_services.py

@@ -29,7 +29,7 @@ class EconomyPersonService:
         """
         批量查询人的经济情况（资金、雇佣关系）
         Query person’s economic situation (funds, employment relationship) in batches
-        
+
         Args:
         - req (dict): https://cityproto.sim.fiblab.net/#city.economy.v1.GetPersonRequest
 

pycityagent/environment/sim/road_service.py

@@ -25,7 +25,7 @@ class RoadService:
         """
         查询道路信息
         Query road information
-        
+
         Args:
         - req (dict): https://cityproto.sim.fiblab.net/#city.map.v2.GetRoadRequest
 

pycityagent/environment/sim/social_service.py

@@ -43,7 +43,7 @@ class SocialService:
         """
         接收消息
         Receive message
-        
+
         Args:
         - req (dict): https://cityproto.sim.fiblab.net/#city.social.v1.ReceiveRequest
 

pycityagent/environment/simulator.py

@@ -52,8 +52,8 @@ class Simulator:
             with open(_map_pb_path, "wb") as f:
                 f.write(_map_pb.SerializeToString())
 
-        if 'simulator' in config:
-            if 'server' not in config["simulator"]:
+        if "simulator" in config:
+            if "server" not in config["simulator"]:
                 self._sim_env = sim_env = ControlSimEnv(
                     task_name=config["simulator"].get("task", "citysim"),
                     map_file=_map_pb_path,
@@ -71,9 +71,11 @@ class Simulator:
                 - grpc client of simulator
                 """
             else:
-                self._client = CityClient(config['simulator']['server'], secure=False)
+                self._client = CityClient(config["simulator"]["server"], secure=False)
         else:
-            logging.warning("No simulator config found, no simulator client will be used")
+            logging.warning(
+                "No simulator config found, no simulator client will be used"
+            )
         self.map = SimMap(
             mongo_uri=_mongo_uri,
             mongo_db=_mongo_db,

pycityagent/environment/utils/__init__.py

@@ -7,4 +7,8 @@ from .geojson import wrap_feature_collection
 from .base64 import encode_to_base64
 from .port import find_free_port
 
-__all__ = ["wrap_feature_collection","find_free_port","find_free_port",]
+__all__ = [
+    "wrap_feature_collection",
+    "find_free_port",
+    "find_free_port",
+]

pycityagent/llm/__init__.py

@@ -3,4 +3,4 @@
 from .llm import LLM, LLMConfig
 from .embedding import SimpleEmbedding
 
-__all__ = ["LLM", "LLMConfig", "SimpleEmbedding"]
+__all__ = ["LLM", "LLMConfig", "SimpleEmbedding"]

pycityagent/llm/embedding.py

@@ -5,16 +5,17 @@ from typing import List, Dict, Optional
 import hashlib
 import json
 
+
 class SimpleEmbedding:
     """简单的基于内存的embedding实现
-    
+
     使用简单的词袋模型(Bag of Words)和TF-IDF来生成文本的向量表示。
     所有向量都保存在内存中，适用于小规模应用。
     """
-    
+
     def __init__(self, vector_dim: int = 128, cache_size: int = 1000):
         """初始化
-        
+
         Args:
             vector_dim: 向量维度
             cache_size: 缓存大小，超过此大小将清除最早的缓存
@@ -25,29 +26,29 @@ class SimpleEmbedding:
         self._vocab: Dict[str, int] = {}  # 词汇表
         self._idf: Dict[str, float] = {}  # 逆文档频率
         self._doc_count = 0  # 文档总数
-        
+
     def _text_to_hash(self, text: str) -> str:
         """将文本转换为hash值"""
         return hashlib.md5(text.encode()).hexdigest()
-        
+
     def _tokenize(self, text: str) -> List[str]:
         """简单的分词"""
         # 这里使用简单的空格分词，实际应用中可以使用更复杂的分词方法
         return text.lower().split()
-        
+
     def _update_vocab(self, tokens: List[str]):
         """更新词汇表"""
         for token in set(tokens):  # 使用set去重
             if token not in self._vocab:
                 self._vocab[token] = len(self._vocab)
-                
+
     def _update_idf(self, tokens: List[str]):
         """更新IDF值"""
         self._doc_count += 1
         unique_tokens = set(tokens)
         for token in unique_tokens:
             self._idf[token] = self._idf.get(token, 0) + 1
-            
+
     def _calculate_tf(self, tokens: List[str]) -> Dict[str, float]:
         """计算词频(TF)"""
         tf = {}
@@ -58,31 +59,31 @@ class SimpleEmbedding:
         for token in tf:
             tf[token] /= total_tokens
         return tf
-        
+
     def _calculate_tfidf(self, tokens: List[str]) -> np.ndarray:
         """计算TF-IDF向量"""
         vector = np.zeros(self.vector_dim)
         tf = self._calculate_tf(tokens)
-        
+
         for token, tf_value in tf.items():
             if token in self._idf:
                 idf = np.log(self._doc_count / self._idf[token])
                 idx = self._vocab[token] % self.vector_dim  # 使用取模运算来控制向量维度
                 vector[idx] += tf_value * idf
-                
+
         # L2归一化
         norm = np.linalg.norm(vector)
         if norm > 0:
             vector /= norm
-            
+
         return vector
-        
+
     async def embed(self, text: str) -> np.ndarray:
         """生成文本的向量表示
-        
+
         Args:
             text: 输入文本
-            
+
         Returns:
             np.ndarray: 文本的向量表示
         """
@@ -90,47 +91,47 @@ class SimpleEmbedding:
         text_hash = self._text_to_hash(text)
         if text_hash in self._cache:
             return self._cache[text_hash]
-            
+
         # 分词
         tokens = self._tokenize(text)
         if not tokens:
             return np.zeros(self.vector_dim)
-            
+
         # 更新词汇表和IDF
         self._update_vocab(tokens)
         self._update_idf(tokens)
-        
+
         # 计算向量
         vector = self._calculate_tfidf(tokens)
-        
+
         # 更新缓存
         if len(self._cache) >= self.cache_size:
             # 删除最早的缓存
             oldest_key = next(iter(self._cache))
             del self._cache[oldest_key]
         self._cache[text_hash] = vector
-        
+
         return vector
-        
+
     def save(self, file_path: str):
         """保存模型"""
         state = {
-            'vector_dim': self.vector_dim,
-            'cache_size': self.cache_size,
-            'vocab': self._vocab,
-            'idf': self._idf,
-            'doc_count': self._doc_count
+            "vector_dim": self.vector_dim,
+            "cache_size": self.cache_size,
+            "vocab": self._vocab,
+            "idf": self._idf,
+            "doc_count": self._doc_count,
         }
-        with open(file_path, 'w') as f:
+        with open(file_path, "w") as f:
             json.dump(state, f)
-            
+
     def load(self, file_path: str):
         """加载模型"""
-        with open(file_path, 'r') as f:
+        with open(file_path, "r") as f:
             state = json.load(f)
-        self.vector_dim = state['vector_dim']
-        self.cache_size = state['cache_size']
-        self._vocab = state['vocab']
-        self._idf = state['idf']
-        self._doc_count = state['doc_count']
-        self._cache = {}  # 清空缓存 
+        self.vector_dim = state["vector_dim"]
+        self.cache_size = state["cache_size"]
+        self._vocab = state["vocab"]
+        self._idf = state["idf"]
+        self._doc_count = state["doc_count"]
+        self._cache = {}  # 清空缓存