pycityagent 2.0.0a61__cp312-cp312-macosx_11_0_arm64.whl → 2.0.0a62__cp312-cp312-macosx_11_0_arm64.whl

pycityagent/cityagent/message_intercept.py

@@ -1,4 +1,5 @@
 import asyncio
+import json
 
 from pycityagent.llm import LLM
 from pycityagent.message import MessageBlockBase, MessageBlockListenerBase
@@ -12,7 +13,7 @@ async def check_message(
     返回: (是否合规, from_uuid, to_uuid)
     """
     print(f"\n检查消息: {from_uuid} -> {to_uuid}: {content}")
-
+    is_valid = True
     prompt = f"""
         请判断以下消息是否具有情绪煽动性：
         发送者ID: {from_uuid}
@@ -21,11 +22,21 @@ async def check_message(
         
         如果消息具有情绪煽动性，请返回 False；如果消息正常，请返回 True。
         """
-
-    if "test" in content.lower():
-        is_valid = False
+    for _ in range(10):
+        try:
+            response: str = await llm_client.atext_request(
+                prompt, timeout=300
+            )  # type:ignore
+            if "false" in response.lower():
+                is_valid = False
+                break
+            elif "true" in response.lower():
+                is_valid = True
+                break
+        except:
+            pass
     else:
-        is_valid = True
+        raise RuntimeError(f"Request for message interception prompt=`{prompt}` failed")
     print(f"消息检查结果: {'合规' if is_valid else '不合规'}")
     return is_valid
 

{pycityagent-2.0.0a61.dist-info → pycityagent-2.0.0a62.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: pycityagent
-Version: 2.0.0a61
+Version: 2.0.0a62
 Summary: LLM-based city environment agent building library
 Author-email: Yuwei Yan <pinkgranite86@gmail.com>, Junbo Yan <yanjb20thu@gmali.com>, Jun Zhang <zhangjun990222@gmali.com>
 License: MIT License

{pycityagent-2.0.0a61.dist-info → pycityagent-2.0.0a62.dist-info}/RECORD

@@ -82,7 +82,7 @@ pycityagent/cityagent/firmagent.py,sha256=UVlNN0lpa4cC4PZVqYzQhbc5VJ2oGsA1731mhb
 pycityagent/cityagent/nbsagent.py,sha256=WIXW__6dZ5IrqBqDCjvGbrCshpXzuFRV3Ww6gkYw7p4,4387
 pycityagent/cityagent/initial.py,sha256=7hgCt_tGdnVTXGfEQOn1GTW5dAs1b-ru_FwXxRLI6tM,4549
 pycityagent/cityagent/societyagent.py,sha256=DKkPtCPXTpHs83hi3bvXxkQYxKqZI0YldYCyFu4yjsk,24811
-pycityagent/cityagent/message_intercept.py,sha256=2RI4--5SVQy9aLXulxxldzmwOynwAKFqtmBaFUnEtsk,3511
+pycityagent/cityagent/message_intercept.py,sha256=KqwW19zpPG8WMpiU91GAhJc9Gox3YgjFkTDokUFWAkw,3940
 pycityagent/cityagent/governmentagent.py,sha256=HJLuhvEmllu_1KnFEJsYCIasaBJT0BV9Cn_4Y2QGPqg,2791
 pycityagent/cityagent/blocks/dispatcher.py,sha256=mEa1r3tRS3KI1BMZR_w_sbUGzOj6aUJuiUrsHv1n2n0,2943
 pycityagent/cityagent/blocks/needs_block.py,sha256=s8LikgtKORfo_Sw9SQ5_3biNPTof15QuUs4cDynXCyM,15332
@@ -97,9 +97,9 @@ pycityagent/cityagent/blocks/mobility_block.py,sha256=xWbARMfJ3-6fddrW3VOUKJrXfM
 pycityagent/survey/models.py,sha256=YE50UUt5qJ0O_lIUsSY6XFCGUTkJVNu_L1gAhaCJ2fs,3546
 pycityagent/survey/__init__.py,sha256=rxwou8U9KeFSP7rMzXtmtp2fVFZxK4Trzi-psx9LPIs,153
 pycityagent/survey/manager.py,sha256=S5IkwTdelsdtZETChRcfCEczzwSrry_Fly9MY4s3rbk,1681
-pycityagent-2.0.0a61.dist-info/RECORD,,
-pycityagent-2.0.0a61.dist-info/LICENSE,sha256=n2HPXiupinpyHMnIkbCf3OTYd3KMqbmldu1e7av0CAU,1084
-pycityagent-2.0.0a61.dist-info/WHEEL,sha256=VujM3ypTCyUW6hcTDdK2ej0ARVMxlU1Djlh_zWnDgqk,109
-pycityagent-2.0.0a61.dist-info/entry_points.txt,sha256=BZcne49AAIFv-hawxGnPbblea7X3MtAtoPyDX8L4OC4,132
-pycityagent-2.0.0a61.dist-info/top_level.txt,sha256=yOmeu6cSXmiUtScu53a3s0p7BGtLMaV0aff83EHCTic,43
-pycityagent-2.0.0a61.dist-info/METADATA,sha256=hvkHrK65VERp64_mKCo6cwULjsAl_GTwsui2NCwOek4,9110
+pycityagent-2.0.0a62.dist-info/RECORD,,
+pycityagent-2.0.0a62.dist-info/LICENSE,sha256=n2HPXiupinpyHMnIkbCf3OTYd3KMqbmldu1e7av0CAU,1084
+pycityagent-2.0.0a62.dist-info/WHEEL,sha256=VujM3ypTCyUW6hcTDdK2ej0ARVMxlU1Djlh_zWnDgqk,109
+pycityagent-2.0.0a62.dist-info/entry_points.txt,sha256=BZcne49AAIFv-hawxGnPbblea7X3MtAtoPyDX8L4OC4,132
+pycityagent-2.0.0a62.dist-info/top_level.txt,sha256=yOmeu6cSXmiUtScu53a3s0p7BGtLMaV0aff83EHCTic,43
+pycityagent-2.0.0a62.dist-info/METADATA,sha256=FhgUzevhPEKaFM0lB6XPQ3Fyoswe2V_7tjPsGzDjV-M,9110