aient 1.2.2__tar.gz → 1.2.4__tar.gz

{aient-1.2.2 → aient-1.2.4}/PKG-INFO

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

{aient-1.2.2 → aient-1.2.4}/aient/architext/architext/core.py

@@ -142,6 +142,15 @@ class Texts(ContextProvider):
     async def render(self) -> Optional[str]:
         return self.content
 
+    def __eq__(self, other):
+        if not isinstance(other, Texts):
+            return NotImplemented
+        # If either object is dynamic, they are only equal if they are the exact same object.
+        if self._is_dynamic or (hasattr(other, '_is_dynamic') and other._is_dynamic):
+            return self is other
+        # For static content, compare the actual content.
+        return self.content == other.content
+
 class Tools(ContextProvider):
     def __init__(self, tools_json: Optional[List[Dict]] = None, name: str = "tools"):
         super().__init__(name)
@@ -154,6 +163,11 @@ class Tools(ContextProvider):
             return None
         return f"<tools>{str(self._tools_json)}</tools>"
 
+    def __eq__(self, other):
+        if not isinstance(other, Tools):
+            return NotImplemented
+        return self._tools_json == other._tools_json
+
 class Files(ContextProvider):
     def __init__(self, *paths: Union[str, List[str]], name: str = "files"):
         super().__init__(name)
@@ -230,6 +244,11 @@ class Files(ContextProvider):
         if not self._files: return None
         return "<latest_file_content>" + "\n".join([f"<file><file_path>{p}</file_path><file_content>{c}</file_content></file>" for p, c in self._files.items()]) + "\n</latest_file_content>"
 
+    def __eq__(self, other):
+        if not isinstance(other, Files):
+            return NotImplemented
+        return self._files == other._files
+
 class Images(ContextProvider):
     def __init__(self, url: str, name: Optional[str] = None):
         super().__init__(name or url)
@@ -250,6 +269,11 @@ class Images(ContextProvider):
             logging.warning(f"Image file not found: {self.url}. Skipping.")
             return None # Or handle error appropriately
 
+    def __eq__(self, other):
+        if not isinstance(other, Images):
+            return NotImplemented
+        return self.url == other.url
+
 # 3. 消息类 (已合并 MessageContent)
 class Message(ABC):
     def __init__(self, role: str, *initial_items: Union[ContextProvider, str, list]):
@@ -356,26 +380,36 @@ class Message(ABC):
             return type(self)(*new_items)
         return NotImplemented
 
-    def __getitem__(self, key: str) -> Any:
+    def __getitem__(self, key: Union[str, int]) -> Any:
         """
-        使得 Message 对象支持字典风格的访问 (e.g., message['content'])。
+        使得 Message 对象支持字典风格的访问 (e.g., message['content'])
+        和列表风格的索引访问 (e.g., message[-1])。
         """
-        if key == 'role':
-            return self.role
-        elif key == 'content':
-            # 直接调用 to_dict 并提取 'content'，确保逻辑一致
-            rendered_dict = self.to_dict()
-            return rendered_dict.get('content') if rendered_dict else None
-        # 对于 tool_calls 等特殊属性，也通过 to_dict 获取
-        elif hasattr(self, key):
-            rendered_dict = self.to_dict()
-            if rendered_dict and key in rendered_dict:
-                return rendered_dict[key]
-
-        # 如果在对象本身或其 to_dict() 中都找不到，则引发 KeyError
-        if hasattr(self, key):
-             return getattr(self, key)
-        raise KeyError(f"'{key}'")
+        if isinstance(key, str):
+            if key == 'role':
+                return self.role
+            elif key == 'content':
+                # 直接调用 to_dict 并提取 'content'，确保逻辑一致
+                rendered_dict = self.to_dict()
+                return rendered_dict.get('content') if rendered_dict else None
+            # 对于 tool_calls 等特殊属性，也通过 to_dict 获取
+            elif hasattr(self, key):
+                rendered_dict = self.to_dict()
+                if rendered_dict and key in rendered_dict:
+                    return rendered_dict[key]
+
+            # 如果在对象本身或其 to_dict() 中都找不到，则引发 KeyError
+            if hasattr(self, key):
+                 return getattr(self, key)
+            raise KeyError(f"'{key}'")
+        elif isinstance(key, int):
+            return self._items[key]
+        else:
+            raise TypeError(f"Message indices must be integers or strings, not {type(key).__name__}")
+
+    def __len__(self) -> int:
+        """返回消息中 provider 的数量。"""
+        return len(self._items)
 
     def __repr__(self): return f"Message(role='{self.role}', items={[i.name for i in self._items]})"
     def __bool__(self) -> bool:

{aient-1.2.2 → aient-1.2.4}/aient/architext/test/test.py

@@ -1136,6 +1136,28 @@ Current time: {Texts(lambda: datetime.now().strftime("%Y-%m-%d %H:%M:%S"))}
             await p.refresh()
         self.assertEqual(role_message.content, "通过工厂创建的内容")
 
+    async def test_za_message_indexing_and_length(self):
+        """测试 Message 对象是否支持通过索引访问 provider 以及获取长度"""
+        # 1. 创建一个 UserMessage
+        mess = UserMessage(
+            Texts("some instruction"),
+            Texts("hi", name="done")
+        )
+
+        # 2. 测试获取长度
+        # 这在实现 __len__ 之前会失败
+        self.assertEqual(len(mess), 2)
+
+        # 3. 测试通过索引访问
+        # 这在修改 __getitem__ 之前会失败
+        self.assertEqual(mess[-1].name, "done")
+        self.assertEqual(mess[0].name, Texts("some instruction").name)
+        self.assertEqual(mess[0], Texts("some instruction"))
+
+        # 4. 测试索引越界
+        with self.assertRaises(IndexError):
+            _ = mess[2]
+
 # ==============================================================================
 # 6. 演示
 # ==============================================================================

{aient-1.2.2 → aient-1.2.4}/aient/models/chatgpt.py

@@ -256,6 +256,12 @@ class chatgpt(BaseLLM):
             "image": True
         }
 
+        done_message = self.conversation[convo_id].provider("done")
+        if self.check_done and done_message:
+            done_message.visible = False
+            if self.conversation[convo_id][-1][-1].name == "done":
+                done_message[-1].visible = True
+
         # 构造请求数据
         request_data = {
             "model": model or self.engine,

{aient-1.2.2 → aient-1.2.4}/aient.egg-info/PKG-INFO

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

{aient-1.2.2 → aient-1.2.4}/pyproject.toml

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