aient 1.2.12__tar.gz → 1.2.13__tar.gz

{aient-1.2.12 → aient-1.2.13}/PKG-INFO

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

{aient-1.2.12 → aient-1.2.13}/aient/architext/architext/core.py

@@ -147,6 +147,29 @@ class Texts(ContextProvider):
     async def render(self) -> Optional[str]:
         return self.content
 
+    def __getstate__(self):
+        """Custom state for pickling."""
+        state = self.__dict__.copy()
+        if self._is_dynamic:
+            # For dynamic content, we snapshot its current value for serialization.
+            # The lambda function itself cannot be pickled.
+            try:
+                # Evaluate the lambda and store it as a static string
+                state['_text'] = self.content
+                # Mark it as no longer dynamic in the pickled state
+                state['_is_dynamic'] = False
+            except Exception as e:
+                # If the lambda fails for some reason, store an error message.
+                logging.error(f"Error evaluating dynamic text '{self.name}' during pickling: {e}")
+                state['_text'] = f"[Error: Could not evaluate dynamic content during save: {e}]"
+                state['_is_dynamic'] = False
+        return state
+
+    def __setstate__(self, state):
+        """Custom state for unpickling."""
+        # Just restore the dictionary. The transformation is one-way.
+        self.__dict__.update(state)
+
     def __eq__(self, other):
         if not isinstance(other, Texts):
             return NotImplemented

{aient-1.2.12 → aient-1.2.13}/aient/architext/test/test.py

@@ -1258,6 +1258,50 @@ Current time: {Texts(lambda: datetime.now().strftime("%Y-%m-%d %H:%M:%S"))}
         self.assertEqual(messages3[2].content, "C")
         self.assertIsInstance(messages3[1], AssistantMessage)
 
+    async def test_ze_fstring_lambda_serialization(self):
+        """测试包含 lambda 的 f-string 消息是否可以被序列化和反序列化"""
+        import platform
+        import os
+
+        # 1. 创建一个使用 f-string 和 lambda 的动态消息
+        f_string_message = f"""系统信息: {Texts(lambda: platform.platform())}"""
+        messages_to_save = Messages(SystemMessage(f_string_message))
+
+        # 2. 定义一个临时文件路径
+        test_file_path = "test_lambda_serialization.pkl"
+
+        # 3. 序列化和反序列化
+        try:
+            # 保存
+            messages_to_save.save(test_file_path)
+
+            # 确认文件已创建
+            self.assertTrue(os.path.exists(test_file_path))
+
+            # 加载
+            messages_loaded = Messages.load(test_file_path)
+
+            # 验证加载的对象
+            self.assertIsNotNone(messages_loaded)
+            self.assertIsInstance(messages_loaded, Messages)
+            self.assertEqual(len(messages_loaded), 1)
+
+            # 4. 渲染加载后的消息以验证 lambda 是否仍然有效
+            rendered = await messages_loaded.render_latest()
+
+            self.assertEqual(len(rendered), 1)
+            self.assertIn("系统信息:", rendered[0]['content'])
+            # 验证 platform.platform() 的结果是否在渲染内容中
+            self.assertIn(platform.platform(), rendered[0]['content'])
+
+        except Exception as e:
+            # 如果出现任何异常，测试失败
+            self.fail(f"序列化或反序列化带有 lambda 的 f-string 消息时出错: {e}")
+        finally:
+            # 5. 清理临时文件
+            if os.path.exists(test_file_path):
+                os.remove(test_file_path)
+
 
 # ==============================================================================
 # 6. 演示

{aient-1.2.12 → aient-1.2.13}/aient.egg-info/PKG-INFO

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

{aient-1.2.12 → aient-1.2.13}/pyproject.toml

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