aient 1.2.6__py3-none-any.whl → 1.2.8__py3-none-any.whl

aient/architext/architext/core.py

@@ -632,6 +632,35 @@ class Messages:
             logging.error(f"Could not deserialize file {file_path}: {e}")
             return cls()
 
-    def __getitem__(self, index: int) -> Message: return self._messages[index]
+    def __getitem__(self, index: Union[int, slice]) -> Union[Message, 'Messages']:
+        if isinstance(index, slice):
+            return Messages(*self._messages[index])
+        return self._messages[index]
+
+    def __setitem__(self, index: slice, value: 'Messages'):
+        if not isinstance(index, slice) or not isinstance(value, Messages):
+            raise TypeError("Unsupported operand type(s) for slice assignment")
+
+        # Basic slice assignment logic.
+        # A more robust implementation would handle step and negative indices.
+        start, stop, step = index.indices(len(self._messages))
+
+        if step != 1:
+            raise ValueError("Slice assignment with step is not supported.")
+
+        # Remove old providers from the index
+        for i in range(start, stop):
+            for provider in self._messages[i].provider():
+                self._notify_provider_removed(provider)
+
+        # Replace the slice in the list
+        self._messages[start:stop] = value._messages
+
+        # Add new providers to the index and set parent
+        for msg in value:
+            msg._parent_messages = self
+            for provider in msg.provider():
+                self._notify_provider_added(provider, msg)
+
     def __len__(self) -> int: return len(self._messages)
     def __iter__(self): return iter(self._messages)

aient/architext/test/test.py

@@ -1182,6 +1182,83 @@ Current time: {Texts(lambda: datetime.now().strftime("%Y-%m-%d %H:%M:%S"))}
         non_existent_provider = message.provider("non_existent")
         self.assertIsNone(non_existent_provider)
 
+    async def test_zc_slicing_support(self):
+        """测试 Messages 对象是否支持切片操作"""
+        m1 = SystemMessage("1")
+        m2 = UserMessage("2")
+        m3 = AssistantMessage("3")
+        m4 = UserMessage("4")
+        messages = Messages(m1, m2, m3, m4)
+
+        # 1. Test basic slicing
+        sliced_messages = messages[1:3]
+        self.assertIsInstance(sliced_messages, Messages)
+        self.assertEqual(len(sliced_messages), 2)
+        self.assertIs(sliced_messages[0], m2)
+        self.assertIs(sliced_messages[1], m3)
+
+        # 2. Test slicing with open end
+        sliced_messages_open = messages[2:]
+        self.assertIsInstance(sliced_messages_open, Messages)
+        self.assertEqual(len(sliced_messages_open), 2)
+        self.assertIs(sliced_messages_open[0], m3)
+        self.assertIs(sliced_messages_open[1], m4)
+
+        # 3. Test slicing with open start
+        sliced_messages_start = messages[:2]
+        self.assertIsInstance(sliced_messages_start, Messages)
+        self.assertEqual(len(sliced_messages_start), 2)
+        self.assertIs(sliced_messages_start[0], m1)
+        self.assertIs(sliced_messages_start[1], m2)
+
+        # 4. Test slicing a single element
+        sliced_single = messages[2:3]
+        self.assertIsInstance(sliced_single, Messages)
+        self.assertEqual(len(sliced_single), 1)
+        self.assertIs(sliced_single[0], m3)
+
+    async def test_zd_slice_assignment(self):
+        """测试 Messages 对象的切片赋值功能"""
+        # 1. Setup initial Messages objects
+        m1 = SystemMessage("1")
+        m2 = UserMessage("2")
+        m3 = AssistantMessage("3")
+        m4 = UserMessage("4")
+        messages1 = Messages(m1, m2, m3, m4)
+
+        m5 = SystemMessage("5")
+        m6 = UserMessage("6")
+        messages2 = Messages(m5, m6)
+
+        # 2. Perform slice assignment
+        # This should replace elements from index 1 onwards in messages1
+        # with all elements from messages2
+        messages1[1:] = messages2
+
+        # 3. Verify the result
+        self.assertEqual(len(messages1), 3) # Should be m1, m5, m6
+        self.assertIs(messages1[0], m1)
+        self.assertIs(messages1[1], m5)
+        self.assertIs(messages1[2], m6)
+
+        # 4. Test assigning from a slice, with different roles to prevent merging
+        messages3 = Messages(UserMessage("A"), AssistantMessage("B"), UserMessage("C"))
+        messages4 = Messages(SystemMessage("X"), AssistantMessage("Y"))
+
+        self.assertEqual(len(messages3), 3) # Verify length before assignment
+
+        messages3[1:2] = messages4[1:] # Replace AssistantMessage("B") with AssistantMessage("Y")
+
+        # We need to refresh to access .content property correctly
+        await messages3.refresh()
+
+        self.assertEqual(len(messages3), 3)
+        self.assertEqual(messages3[0].content, "A")
+        self.assertEqual(messages3[1].content, "Y")
+        self.assertEqual(messages3[2].content, "C")
+        self.assertIsInstance(messages3[1], AssistantMessage)
+
+
 # ==============================================================================
 # 6. 演示
 # ==============================================================================

{aient-1.2.6.dist-info → aient-1.2.8.dist-info}/METADATA

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

{aient-1.2.6.dist-info → aient-1.2.8.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=A1ZeZSJwcdl-svYA12uD2qSlU-wfNTVWvpSg8AF8_Gk,26015
+aient/architext/architext/core.py,sha256=83eouPWfZt9G6TGCQguoOxOZM_-HRfXIUwjGzbxPoxw,27164
 aient/architext/test/openai_client.py,sha256=Dqtbmubv6vwF8uBqcayG0kbsiO65of7sgU2-DRBi-UM,4590
-aient/architext/test/test.py,sha256=trVHo2we0W8RN-0QNvP3sJ3yUpe08-34Ae_ZVNZJHdE,56378
+aient/architext/test/test.py,sha256=26cQXXwCcCAVruM-ouM7ng8BQcE-glgmfkGa_EhPDU4,59388
 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.6.dist-info/licenses/LICENSE,sha256=XNdbcWldt0yaNXXWB_Bakoqnxb3OVhUft4MgMA_71ds,1051
-aient-1.2.6.dist-info/METADATA,sha256=aKQPqRvoFl1kd5GXFxOziGdeRhNAZDL8ksFspWdljoo,4841
-aient-1.2.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-aient-1.2.6.dist-info/top_level.txt,sha256=3oXzrP5sAVvyyqabpeq8A2_vfMtY554r4bVE-OHBrZk,6
-aient-1.2.6.dist-info/RECORD,,
+aient-1.2.8.dist-info/licenses/LICENSE,sha256=XNdbcWldt0yaNXXWB_Bakoqnxb3OVhUft4MgMA_71ds,1051
+aient-1.2.8.dist-info/METADATA,sha256=d-gqNwZ00IPMuuqj3sKECwiC1BS1H-M9hUaKi6I74aA,4841
+aient-1.2.8.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+aient-1.2.8.dist-info/top_level.txt,sha256=3oXzrP5sAVvyyqabpeq8A2_vfMtY554r4bVE-OHBrZk,6
+aient-1.2.8.dist-info/RECORD,,