aient 1.2.5__tar.gz → 1.2.7__tar.gz

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

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

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

@@ -360,7 +360,17 @@ class Message(ABC):
         if self._parent_messages:
             self._parent_messages._notify_provider_added(item, self)
 
-    def provider(self) -> List[ContextProvider]: return self._items
+    def provider(self, name: Optional[str] = None) -> Optional[Union[ContextProvider, ProviderGroup, List[ContextProvider]]]:
+        if name is None:
+            return self._items
+
+        named_providers = [p for p in self._items if hasattr(p, 'name') and p.name == name]
+
+        if not named_providers:
+            return None
+        if len(named_providers) == 1:
+            return named_providers[0]
+        return ProviderGroup(named_providers)
 
     def __add__(self, other):
         if isinstance(other, str):
@@ -622,6 +632,9 @@ 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 __len__(self) -> int: return len(self._messages)
     def __iter__(self): return iter(self._messages)

{aient-1.2.5 → aient-1.2.7}/aient/architext/test/test.py

@@ -1158,6 +1158,65 @@ Current time: {Texts(lambda: datetime.now().strftime("%Y-%m-%d %H:%M:%S"))}
         with self.assertRaises(IndexError):
             _ = mess[2]
 
+    async def test_zb_message_provider_by_name(self):
+        """测试是否可以通过名称从 Message 对象中获取 provider"""
+        # 1. 创建一个包含命名 provider 的 Message
+        message = UserMessage(
+            Texts("Some instruction", name="instruction"),
+            Tools([{"name": "a_tool"}], name="tools"),
+            Texts("Another instruction", name="instruction")
+        )
+
+        # 2. 测试获取单个 provider
+        tools_provider = message.provider("tools")
+        self.assertIsInstance(tools_provider, Tools)
+        self.assertEqual(tools_provider.name, "tools")
+
+        # 3. 测试获取多个同名 provider
+        instruction_providers = message.provider("instruction")
+        self.assertIsInstance(instruction_providers, ProviderGroup)
+        self.assertEqual(len(instruction_providers), 2)
+        self.assertTrue(all(isinstance(p, Texts) for p in instruction_providers))
+
+        # 4. 测试获取不存在的 provider
+        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)
+
 # ==============================================================================
 # 6. 演示
 # ==============================================================================

{aient-1.2.5 → aient-1.2.7}/aient.egg-info/PKG-INFO

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

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

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