aient 1.2.2__py3-none-any.whl → 1.2.3__py3-none-any.whl

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/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.dist-info → aient-1.2.3.dist-info}/METADATA

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

{aient-1.2.2.dist-info → aient-1.2.3.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=MCO9VDcHaLflWfCyCxLsbodN03isVzllOcHDRMatzHE,24300
+aient/architext/architext/core.py,sha256=qSRiIfx9Cl_OZIDfGMFr7AKbwRWTBrJNt_jqJeOr5Lw,25636
 aient/architext/test/openai_client.py,sha256=Dqtbmubv6vwF8uBqcayG0kbsiO65of7sgU2-DRBi-UM,4590
-aient/architext/test/test.py,sha256=imNgYfAGTJWodxLg1gFTSf6qRfnBjkKG5oqZ68Gp2jw,54522
+aient/architext/test/test.py,sha256=efWn_qYdgPZO-Wz68o33QoWhWiJjU-cAzPbIVL0sjhw,55316
 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
@@ -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.2.2.dist-info/licenses/LICENSE,sha256=XNdbcWldt0yaNXXWB_Bakoqnxb3OVhUft4MgMA_71ds,1051
-aient-1.2.2.dist-info/METADATA,sha256=0tPSEeenP8co0F3rLx9quhkSrJMJmmNQpGaVKueMtVk,4841
-aient-1.2.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-aient-1.2.2.dist-info/top_level.txt,sha256=3oXzrP5sAVvyyqabpeq8A2_vfMtY554r4bVE-OHBrZk,6
-aient-1.2.2.dist-info/RECORD,,
+aient-1.2.3.dist-info/licenses/LICENSE,sha256=XNdbcWldt0yaNXXWB_Bakoqnxb3OVhUft4MgMA_71ds,1051
+aient-1.2.3.dist-info/METADATA,sha256=HupXkB_iUMlrUCSb45q3nBmmHPENpd52JAZjNGudCTo,4841
+aient-1.2.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+aient-1.2.3.dist-info/top_level.txt,sha256=3oXzrP5sAVvyyqabpeq8A2_vfMtY554r4bVE-OHBrZk,6
+aient-1.2.3.dist-info/RECORD,,