aient 1.1.92__tar.gz → 1.1.94__tar.gz

{aient-1.1.92 → aient-1.1.94}/PKG-INFO

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

{aient-1.1.92 → aient-1.1.94}/aient/architext/architext/core.py

@@ -6,7 +6,7 @@ import hashlib
 import mimetypes
 from dataclasses import dataclass
 from abc import ABC, abstractmethod
-from typing import List, Dict, Any, Optional, Union
+from typing import List, Dict, Any, Optional, Union, Callable
 
 # 1. 核心数据结构: ContentBlock
 @dataclass
@@ -32,20 +32,39 @@ class ContextProvider(ABC):
         return None
 
 class Texts(ContextProvider):
-    def __init__(self, text: str, name: Optional[str] = None):
+    def __init__(self, text: Optional[Union[str, Callable[[], str]]] = None, name: Optional[str] = None):
+        if text is None and name is None:
+            raise ValueError("Either 'text' or 'name' must be provided.")
+
         self._text = text
+        self._is_dynamic = callable(self._text)
+
         if name is None:
-            h = hashlib.sha1(self._text.encode()).hexdigest()
-            _name = f"text_{h[:8]}"
+            if self._is_dynamic:
+                import uuid
+                _name = f"dynamic_text_{uuid.uuid4().hex[:8]}"
+            else:
+                # Handle the case where text is None during initialization
+                h = hashlib.sha1(self._text.encode() if self._text else b'').hexdigest()
+                _name = f"text_{h[:8]}"
         else:
             _name = name
         super().__init__(_name)
 
-    def update(self, text: str):
+    async def refresh(self):
+        if self._is_dynamic:
+            self._is_stale = True
+        await super().refresh()
+
+    def update(self, text: Union[str, Callable[[], str]]):
         self._text = text
+        self._is_dynamic = callable(self._text)
         self.mark_stale()
 
-    async def render(self) -> str: return self._text
+    async def render(self) -> Optional[str]:
+        if self._is_dynamic:
+            return self._text()
+        return self._text
 
 class Tools(ContextProvider):
     def __init__(self, tools_json: List[Dict]): super().__init__("tools"); self._tools_json = tools_json

{aient-1.1.92 → aient-1.1.94}/aient/architext/test/test.py

@@ -819,6 +819,86 @@ class TestContextManagement(unittest.IsolatedAsyncioTestCase):
             if os.path.exists(test_file):
                 os.remove(test_file)
 
+    async def test_z_dynamic_texts_provider(self):
+        """测试 Texts provider 是否支持可调用对象以实现动态内容"""
+        import time
+        from datetime import datetime
+
+        # 1. 使用 lambda 函数创建一个动态的 Texts provider
+        # 每次调用 render 时，它都应该返回当前时间
+        dynamic_text_provider = Texts(lambda: datetime.now().strftime("%Y-%m-%d %H:%M:%S"))
+        messages = Messages(UserMessage(dynamic_text_provider))
+
+        # 2. 第一次渲染
+        rendered1 = await messages.render_latest()
+        time1_str = rendered1[0]['content']
+        self.assertIsNotNone(time1_str)
+
+        # 3. 等待一秒钟
+        time.sleep(1)
+
+        # 4. 第二次渲染，并期望内容已更新
+        rendered2 = await messages.render_latest()
+        time2_str = rendered2[0]['content']
+        self.assertIsNotNone(time2_str)
+
+        # 5. 验证两次渲染的时间戳不同
+        self.assertNotEqual(time1_str, time2_str, "动态 Texts provider 的内容在两次渲染之间应该更新")
+
+    async def test_z2_dynamic_texts_with_prefix(self):
+        """测试动态 Texts provider 包含静态前缀时也能正确更新"""
+        import time
+        from datetime import datetime
+        import platform
+
+        # 1. 创建一个包含静态前缀和动态内容的 provider
+        # 正确的用法是将整个表达式放入 lambda
+        dynamic_provider = Texts(lambda: f"平台信息：{platform.platform()}, 时间：{datetime.now().isoformat()}")
+        messages = Messages(UserMessage(dynamic_provider))
+
+        # 2. 第一次渲染
+        rendered1 = await messages.render_latest()
+        content1 = rendered1[0]['content']
+        self.assertIn("平台信息：", content1)
+
+        # 3. 等待一秒
+        time.sleep(1)
+
+        # 4. 第二次渲染
+        rendered2 = await messages.render_latest()
+        content2 = rendered2[0]['content']
+        self.assertIn("平台信息：", content2)
+
+        # 5. 验证两次内容不同（因为时间戳变了）
+        self.assertNotEqual(content1, content2, "包含静态前缀的动态 provider 内容应该更新")
+
+    async def test_z3_deferred_text_update_via_provider(self):
+        """测试 Texts(name=...) 初始化, 然后通过 provider 更新内容"""
+        # This test is expected to fail with a TypeError on the next line
+        # because the current Texts.__init__ requires 'text'.
+        deferred_text_provider = Texts(name="deferred_content")
+
+        messages = Messages(UserMessage(deferred_text_provider))
+
+        # Initial render: with no text, it should probably render to an empty string.
+        # If there's no content, the message itself might not be rendered.
+        # Let's assume an empty provider results in the message not rendering.
+        await deferred_text_provider.refresh()
+        self.assertIsNone(deferred_text_provider.get_content_block())
+
+        rendered_initial = await messages.render_latest()
+        self.assertEqual(len(rendered_initial), 0)
+
+        # 3. Get provider and update content
+        provider = messages.provider("deferred_content")
+        self.assertIsNotNone(provider)
+        provider.update("This is the new content.")
+
+        # 4. Re-render and validate
+        rendered_updated = await messages.render_latest()
+        self.assertEqual(len(rendered_updated), 1)
+        self.assertEqual(rendered_updated[0]['content'], "This is the new content.")
+
 
 # ==============================================================================
 # 6. 演示

{aient-1.1.92 → aient-1.1.94}/aient.egg-info/PKG-INFO

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

{aient-1.1.92 → aient-1.1.94}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "aient"
-version = "1.1.92"
+version = "1.1.94"
 description = "Aient: The Awakening of Agent."
 readme = "README.md"
 requires-python = ">=3.11"