aient 1.1.98__py3-none-any.whl → 1.2.0__py3-none-any.whl

aient/architext/architext/core.py

@@ -35,13 +35,30 @@ class ContentBlock:
 # 2. 上下文提供者 (带缓存)
 class ContextProvider(ABC):
     def __init__(self, name: str):
-        self.name = name; self._cached_content: Optional[str] = None; self._is_stale: bool = True
+        self.name = name
+        self._cached_content: Optional[str] = None
+        self._is_stale: bool = True
+        self._visible: bool = True
 
     def __str__(self):
         # This allows the object to be captured when used inside an f-string.
         return _register_provider(self)
 
     def mark_stale(self): self._is_stale = True
+
+    @property
+    def visible(self) -> bool:
+        """Gets the visibility of the provider."""
+        return self._visible
+
+    @visible.setter
+    def visible(self, value: bool):
+        """Sets the visibility of the provider."""
+        if self._visible != value:
+            self._visible = value
+            # Content needs to be re-evaluated, but the source data hasn't changed,
+            # so just marking it stale is enough for the renderer to reconsider it.
+            self.mark_stale()
     async def refresh(self):
         if self._is_stale:
             self._cached_content = await self.render()
@@ -51,7 +68,8 @@ class ContextProvider(ABC):
     @abstractmethod
     def update(self, *args, **kwargs): raise NotImplementedError
     def get_content_block(self) -> Optional[ContentBlock]:
-        if self._cached_content is not None: return ContentBlock(self.name, self._cached_content)
+        if self.visible and self._cached_content is not None:
+            return ContentBlock(self.name, self._cached_content)
         return None
 
 class Texts(ContextProvider):

aient/architext/test/test.py

@@ -1021,6 +1021,38 @@ Current time: {Texts(lambda: datetime.now().strftime("%Y-%m-%d %H:%M:%S"))}
         empty_text = Texts("")
         self.assertEqual(empty_text.content, "")
 
+    async def test_z7_provider_visibility(self):
+        """测试 provider 的可见性标志是否能正常工作"""
+        # 1. 初始化 provider，visible 默认为 True
+        text_provider = Texts("Hello, World!", name="greeting")
+        self.assertTrue(text_provider.visible)
+
+        messages = Messages(SystemMessage(text_provider))
+
+        # 2. 初始渲染，内容应该可见
+        rendered_visible = await messages.render_latest()
+        self.assertEqual(len(rendered_visible), 1)
+        self.assertEqual(rendered_visible[0]['content'], "Hello, World!")
+
+        # 3. 设置为不可见
+        provider = messages.provider("greeting")
+        provider.visible = False
+        self.assertFalse(provider.visible)
+
+        # 4. 再次渲染，内容应该消失
+        # 因为 visibility 变化会 mark_stale，所以需要 render_latest
+        rendered_invisible = await messages.render_latest()
+        self.assertEqual(len(rendered_invisible), 0, "设置为不可见后，消息应该不被渲染")
+
+        # 5. 再次设置为可见
+        provider.visible = True
+        self.assertTrue(provider.visible)
+
+        # 6. 渲染，内容应该再次出现
+        rendered_visible_again = await messages.render_latest()
+        self.assertEqual(len(rendered_visible_again), 1)
+        self.assertEqual(rendered_visible_again[0]['content'], "Hello, World!")
+
 
 # ==============================================================================
 # 6. 演示

aient/models/chatgpt.py

@@ -97,13 +97,8 @@ class chatgpt(BaseLLM):
         Initialize Chatbot with API key (from https://platform.openai.com/account/api-keys)
         """
         super().__init__(api_key, engine, api_url, system_prompt, proxy, timeout, max_tokens, temperature, top_p, presence_penalty, frequency_penalty, reply_count, truncate_limit, use_plugins=use_plugins, print_log=print_log)
-        self.conversation: dict[str, list[dict]] = {
-            "default": [
-                {
-                    "role": "system",
-                    "content": self.system_prompt,
-                },
-            ],
+        self.conversation: dict[str, Messages] = {
+            "default": Messages(SystemMessage(self.system_prompt)),
         }
         if cache_messages:
             self.conversation["default"] = cache_messages

{aient-1.1.98.dist-info → aient-1.2.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aient
-Version: 1.1.98
+Version: 1.2.0
 Summary: Aient: The Awakening of Agent.
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown

{aient-1.1.98.dist-info → aient-1.2.0.dist-info}/RECORD

@@ -1,8 +1,8 @@
 aient/__init__.py,sha256=SRfF7oDVlOOAi6nGKiJIUK6B_arqYLO9iSMp-2IZZps,21
 aient/architext/architext/__init__.py,sha256=79Ih1151rfcqZdr7F8HSZSTs_iT2SKd1xCkehMsXeXs,19
-aient/architext/architext/core.py,sha256=pgnJ5Q9dlxMpp4ilBuHzScxbTlVYihubpo6uYqNV1lU,20976
+aient/architext/architext/core.py,sha256=RGUKPrTm0mXrC7FweIkPXRAXCDE1xxU08pZo1n-2tsA,21555
 aient/architext/test/openai_client.py,sha256=Dqtbmubv6vwF8uBqcayG0kbsiO65of7sgU2-DRBi-UM,4590
-aient/architext/test/test.py,sha256=5xV8nrOBf2Gn49EGnBvGLNXQRbNGi_hD0PYESs7tl0g,49091
+aient/architext/test/test.py,sha256=CSjXZbY-cgJP1xXY-vA56iM5XXMfwsW2sZQLwAwWS7k,50477
 aient/architext/test/test_save_load.py,sha256=o8DqH6gDYZkFkQy-a7blqLtJTRj5e4a-Lil48pJ0V3g,3260
 aient/core/__init__.py,sha256=NxjebTlku35S4Dzr16rdSqSTWUvvwEeACe8KvHJnjPg,34
 aient/core/log_config.py,sha256=kz2_yJv1p-o3lUQOwA3qh-LSc3wMHv13iCQclw44W9c,274
@@ -17,7 +17,7 @@ aient/core/test/test_payload.py,sha256=8jBiJY1uidm1jzL-EiK0s6UGmW9XkdsuuKFGrwFhF
 aient/models/__init__.py,sha256=ZTiZgbfBPTjIPSKURE7t6hlFBVLRS9lluGbmqc1WjxQ,43
 aient/models/audio.py,sha256=kRd-8-WXzv4vwvsTGwnstK-WR8--vr9CdfCZzu8y9LA,1934
 aient/models/base.py,sha256=-nnihYnx-vHZMqeVO9ljjt3k4FcD3n-iMk4tT-10nRQ,7232
-aient/models/chatgpt.py,sha256=EqvIcB6R_JGmvD5kEfR1ZRQYLfyGpmy4PvfRx8hMIKE,42019
+aient/models/chatgpt.py,sha256=_fYGUnSciKSWYhPuS0lECtU6-C42GGGhFiqbVxejaJg,41919
 aient/plugins/__init__.py,sha256=p3KO6Aa3Lupos4i2SjzLQw1hzQTigOAfEHngsldrsyk,986
 aient/plugins/arXiv.py,sha256=yHjb6PS3GUWazpOYRMKMzghKJlxnZ5TX8z9F6UtUVow,1461
 aient/plugins/config.py,sha256=TGgZ5SnNKZ8MmdznrZ-TEq7s2ulhAAwTSKH89bci3dA,7079
@@ -35,8 +35,8 @@ aient/plugins/write_file.py,sha256=Jt8fOEwqhYiSWpCbwfAr1xoi_BmFnx3076GMhuL06uI,3
 aient/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 aient/utils/prompt.py,sha256=UcSzKkFE4-h_1b6NofI6xgk3GoleqALRKY8VBaXLjmI,11311
 aient/utils/scripts.py,sha256=VqtK4RFEx7KxkmcqG3lFDS1DxoNlFFGErEjopVcc8IE,40974
-aient-1.1.98.dist-info/licenses/LICENSE,sha256=XNdbcWldt0yaNXXWB_Bakoqnxb3OVhUft4MgMA_71ds,1051
-aient-1.1.98.dist-info/METADATA,sha256=Ovyrvqq57HzewzZke5vlu0OuV0E8TQm2s0FGMxlrKcA,4842
-aient-1.1.98.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-aient-1.1.98.dist-info/top_level.txt,sha256=3oXzrP5sAVvyyqabpeq8A2_vfMtY554r4bVE-OHBrZk,6
-aient-1.1.98.dist-info/RECORD,,
+aient-1.2.0.dist-info/licenses/LICENSE,sha256=XNdbcWldt0yaNXXWB_Bakoqnxb3OVhUft4MgMA_71ds,1051
+aient-1.2.0.dist-info/METADATA,sha256=YPHfQb4kRpMzEehuqTUOm36MKNkFfcc4VrDJQbNga00,4841
+aient-1.2.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+aient-1.2.0.dist-info/top_level.txt,sha256=3oXzrP5sAVvyyqabpeq8A2_vfMtY554r4bVE-OHBrZk,6
+aient-1.2.0.dist-info/RECORD,,