aient 1.2.7__tar.gz → 1.2.9__tar.gz

{aient-1.2.7 → aient-1.2.9}/PKG-INFO

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

{aient-1.2.7 → aient-1.2.9}/aient/architext/architext/core.py

@@ -101,7 +101,12 @@ class Texts(ContextProvider):
         if text is None and name is None:
             raise ValueError("Either 'text' or 'name' must be provided.")
 
-        self._text = text
+        # Ensure that non-callable inputs are treated as strings
+        if not callable(text):
+            self._text = str(text) if text is not None else None
+        else:
+            self._text = text
+
         self._is_dynamic = callable(self._text)
 
         if name is None:
@@ -636,5 +641,31 @@ class 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-1.2.7 → aient-1.2.9}/aient/architext/test/test.py

@@ -1217,6 +1217,48 @@ Current time: {Texts(lambda: datetime.now().strftime("%Y-%m-%d %H:%M:%S"))}
         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.7 → aient-1.2.9}/aient.egg-info/PKG-INFO

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

{aient-1.2.7 → aient-1.2.9}/pyproject.toml

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