aient 1.1.91__py3-none-any.whl → 1.1.92__py3-none-any.whl

aient/architext/test/test_save_load.py

@@ -0,0 +1,93 @@
+import asyncio
+import os
+import sys
+
+# Add the project root to the Python path
+sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
+
+from architext.core import (
+    Messages,
+    SystemMessage,
+    UserMessage,
+    AssistantMessage,
+    ToolCalls,
+    ToolResults,
+    Texts,
+    Tools,
+    Files,
+)
+
+async def main():
+    """
+    Tests the save and load functionality of the Messages class using pickle.
+    """
+    print("--- Test Save/Load (pickle) ---")
+
+    # 1. Create an initial Messages object
+    messages = Messages(
+        SystemMessage(Texts("system_prompt", "You are a helpful assistant.")),
+        UserMessage(Texts("user_input", "What is the weather in Shanghai?")),
+        AssistantMessage(Texts("thought", "I should call a tool for this.")),
+        ToolCalls(tool_calls=[{
+            'id': 'call_1234',
+            'type': 'function',
+            'function': {'name': 'get_weather', 'arguments': '{"location": "Shanghai"}'}
+        }]),
+        ToolResults(tool_call_id="call_1234", content='{"temperature": "25°C"}')
+    )
+
+    # Add a message with Files provider
+    files_provider = Files()
+    files_provider.update("test.txt", "This is a test file.")
+    messages.append(UserMessage(files_provider))
+
+    # Render the original messages
+    original_render = await messages.render_latest()
+    print("Original Messages Render:")
+    print(original_render)
+
+    # 2. Save the messages to a file
+    file_path = "test_messages.pkl"
+    messages.save(file_path)
+    print(f"\nMessages saved to {file_path}")
+
+    assert os.path.exists(file_path), "Save file was not created."
+
+    # 3. Load the messages from the file
+    loaded_messages = Messages.load(file_path)
+    print("\nMessages loaded from file.")
+
+    assert loaded_messages is not None, "Loaded messages should not be None."
+
+    # Render the loaded messages
+    loaded_render = await loaded_messages.render_latest()
+    print("\nLoaded Messages Render:")
+    print(loaded_render)
+
+    # 4. Compare the original and loaded content
+    assert original_render == loaded_render, "Rendered content of original and loaded messages do not match."
+    print("\n✅ Assertion passed: Original and loaded message renders are identical.")
+
+    # 5. Check if the loaded object retains its class structure and methods
+    print(f"\nType of loaded object: {type(loaded_messages)}")
+    assert isinstance(loaded_messages, Messages), "Loaded object is not a Messages instance."
+
+    # Test pop functionality on the loaded object
+    popped_item = loaded_messages.pop(0)
+    assert isinstance(popped_item, SystemMessage), "Popped item is not a SystemMessage."
+    print(f"Popped first message: {popped_item}")
+
+    popped_render = await loaded_messages.render_latest()
+    print("\nRender after popping first message from loaded object:")
+    print(popped_render)
+    assert len(popped_render) == len(original_render) - 1, "Popping a message did not reduce the message count."
+    print("✅ Assertion passed: Pop functionality works on the loaded object.")
+
+    # 6. Clean up the test file
+    os.remove(file_path)
+    print(f"\nCleaned up {file_path}.")
+
+    print("\n--- Test Completed Successfully ---")
+
+if __name__ == "__main__":
+    asyncio.run(main())

aient/models/chatgpt.py

@@ -14,6 +14,7 @@ from ..plugins import PLUGINS, get_tools_result_async, function_call_list, updat
 from ..utils.scripts import safe_get, async_generator_to_sync, parse_function_xml, parse_continuous_json, convert_functions_to_xml, remove_xml_tags_and_content
 from ..core.request import prepare_request_payload
 from ..core.response import fetch_response_stream, fetch_response
+from ..architext.architext import Messages, SystemMessage, UserMessage, AssistantMessage, ToolCalls, ToolResults, Texts, RoleMessage, Images, Files
 
 class APITimeoutError(Exception):
     """Custom exception for API timeout errors."""
@@ -88,7 +89,6 @@ class chatgpt(BaseLLM):
         print_log: bool = False,
         tools: Optional[Union[list, str, Callable]] = [],
         function_call_max_loop: int = 3,
-        cut_history_by_function_name: str = "",
         cache_messages: list = None,
         logger: logging.Logger = None,
         check_done: bool = False,
@@ -109,8 +109,6 @@ class chatgpt(BaseLLM):
             self.conversation["default"] = cache_messages
         self.function_calls_counter = {}
         self.function_call_max_loop = function_call_max_loop
-        self.cut_history_by_function_name = cut_history_by_function_name
-        self.latest_file_content = {}
         self.check_done = check_done
 
         if logger:
@@ -164,95 +162,48 @@ class chatgpt(BaseLLM):
         if convo_id not in self.conversation:
             self.reset(convo_id=convo_id)
         if function_name == "" and message:
-            self.conversation[convo_id].append({"role": role, "content": message})
+            self.conversation[convo_id].append(RoleMessage(role, message))
         elif function_name != "" and message:
-            # 删除从 cut_history_by_function_name 以后的所有历史记录
-            if function_name == self.cut_history_by_function_name:
-                matching_message = next(filter(lambda x: safe_get(x, "tool_calls", 0, "function", "name", default="") == 'get_next_pdf', self.conversation[convo_id]), None)
-                if matching_message is not None:
-                    self.conversation[convo_id] = self.conversation[convo_id][:self.conversation[convo_id].index(matching_message)]
-
             if not (all(value == False for value in self.plugins.values()) or self.use_plugins == False):
-                self.conversation[convo_id].append({
-                    "role": "assistant",
-                    "tool_calls": [
-                        {
-                            "id": function_call_id,
-                            "type": "function",
-                            "function": {
-                                "name": function_name,
-                                "arguments": function_arguments,
-                            },
-                        }
-                    ],
-                    })
-                self.conversation[convo_id].append({"role": role, "tool_call_id": function_call_id, "content": message})
+                tool_calls = [
+                    {
+                        "id": function_call_id,
+                        "type": "function",
+                        "function": {
+                            "name": function_name,
+                            "arguments": function_arguments,
+                        },
+                    }
+                ]
+                self.conversation[convo_id].append(ToolCalls(tool_calls))
+                self.conversation[convo_id].append(ToolResults(tool_call_id=function_call_id, content=message))
             else:
                 last_user_message = self.conversation[convo_id][-1]["content"]
                 if last_user_message != message:
-                    image_message_list = []
+                    image_message_list = UserMessage()
                     if isinstance(function_arguments, str):
                         functions_list = json.loads(function_arguments)
                     else:
                         functions_list = function_arguments
                     for tool_info in functions_list:
                         if tool_info.get("base64_image"):
-                            image_message_list.append({"type": "text", "text": safe_get(tool_info, "parameter", "image_path", default="") + " image:"})
-                            image_message_list.append({
-                                "type": "image_url",
-                                "image_url": {
-                                    "url": tool_info["base64_image"],
-                                }
-                            })
-                    self.conversation[convo_id].append({"role": "assistant", "content": convert_functions_to_xml(function_arguments)})
+                            image_message_list.extend([
+                                safe_get(tool_info, "parameter", "image_path", default="") + " image:",
+                                Images(tool_info["base64_image"]),
+                            ])
+                    self.conversation[convo_id].append(AssistantMessage(convert_functions_to_xml(function_arguments)))
                     if image_message_list:
-                        self.conversation[convo_id].append({"role": "user", "content": [{"type": "text", "text": message}] + image_message_list})
+                        self.conversation[convo_id].append(UserMessage(message + image_message_list))
                     else:
-                        self.conversation[convo_id].append({"role": "user", "content": message})
+                        self.conversation[convo_id].append(UserMessage(message))
                 else:
-                    self.conversation[convo_id].append({"role": "assistant", "content": "我已经执行过这个工具了，接下来我需要做什么？"})
-
+                    self.conversation[convo_id].append(AssistantMessage("我已经执行过这个工具了，接下来我需要做什么？"))
         else:
             self.logger.error(f"error: add_to_conversation message is None or empty, role: {role}, function_name: {function_name}, message: {message}")
 
-        conversation_len = len(self.conversation[convo_id]) - 1
-        message_index = 0
         # if self.print_log:
         #     replaced_text = json.loads(re.sub(r';base64,([A-Za-z0-9+/=]+)', ';base64,***', json.dumps(self.conversation[convo_id])))
         #     self.logger.info(json.dumps(replaced_text, indent=4, ensure_ascii=False))
-        while message_index < conversation_len:
-            if self.conversation[convo_id][message_index]["role"] == self.conversation[convo_id][message_index + 1]["role"]:
-                if self.conversation[convo_id][message_index].get("content") and self.conversation[convo_id][message_index + 1].get("content") \
-                and self.conversation[convo_id][message_index].get("content") != self.conversation[convo_id][message_index + 1].get("content"):
-                    if type(self.conversation[convo_id][message_index + 1]["content"]) == str \
-                    and type(self.conversation[convo_id][message_index]["content"]) == list:
-                        self.conversation[convo_id][message_index + 1]["content"] = [{"type": "text", "text": self.conversation[convo_id][message_index + 1]["content"]}]
-                    if type(self.conversation[convo_id][message_index]["content"]) == str \
-                    and type(self.conversation[convo_id][message_index + 1]["content"]) == list:
-                        self.conversation[convo_id][message_index]["content"] = [{"type": "text", "text": self.conversation[convo_id][message_index]["content"]}]
-                    if type(self.conversation[convo_id][message_index]["content"]) == dict \
-                    and type(self.conversation[convo_id][message_index + 1]["content"]) == str:
-                        self.conversation[convo_id][message_index]["content"] = [self.conversation[convo_id][message_index]["content"]]
-                        self.conversation[convo_id][message_index + 1]["content"] = [{"type": "text", "text": self.conversation[convo_id][message_index + 1]["content"]}]
-                    if type(self.conversation[convo_id][message_index]["content"]) == dict \
-                    and type(self.conversation[convo_id][message_index + 1]["content"]) == list:
-                        self.conversation[convo_id][message_index]["content"] = [self.conversation[convo_id][message_index]["content"]]
-                    if type(self.conversation[convo_id][message_index]["content"]) == dict \
-                    and type(self.conversation[convo_id][message_index + 1]["content"]) == dict:
-                        self.conversation[convo_id][message_index]["content"] = [self.conversation[convo_id][message_index]["content"]]
-                        self.conversation[convo_id][message_index + 1]["content"] = [self.conversation[convo_id][message_index + 1]["content"]]
-                    if type(self.conversation[convo_id][message_index]["content"]) == list \
-                    and type(self.conversation[convo_id][message_index + 1]["content"]) == dict:
-                        self.conversation[convo_id][message_index + 1]["content"] = [self.conversation[convo_id][message_index + 1]["content"]]
-                    if type(self.conversation[convo_id][message_index]["content"]) == str \
-                    and type(self.conversation[convo_id][message_index + 1]["content"]) == str \
-                    and self.conversation[convo_id][message_index].get("content").endswith(self.conversation[convo_id][message_index + 1].get("content")):
-                        self.conversation[convo_id][message_index + 1]["content"] = ""
-                    self.conversation[convo_id][message_index]["content"] += self.conversation[convo_id][message_index + 1]["content"]
-                self.conversation[convo_id].pop(message_index + 1)
-                conversation_len = conversation_len - 1
-            else:
-                message_index = message_index + 1
 
         history_len = len(self.conversation[convo_id])
 
@@ -290,27 +241,6 @@ class chatgpt(BaseLLM):
             else:
                 break
 
-    def get_latest_file_content(self) -> str:
-        """
-        获取最新文件内容
-        """
-        result = ""
-        if self.latest_file_content:
-            for file_path, content in self.latest_file_content.items():
-                result += (
-                    "<file>"
-                    f"<file_path>{file_path}</file_path>"
-                    f"<file_content>{content}</file_content>"
-                    "</file>\n\n"
-                )
-            if result:
-                result = (
-                    "<latest_file_content>"
-                    f"{result}"
-                    "</latest_file_content>"
-                )
-        return result
-
     async def get_post_body(
         self,
         prompt: str,
@@ -321,8 +251,6 @@ class chatgpt(BaseLLM):
         stream: bool = True,
         **kwargs,
     ):
-        self.conversation[convo_id][0] = {"role": "system","content": self.system_prompt + "\n\n" + self.get_latest_file_content()}
-
         # 构造 provider 信息
         provider = {
             "provider": "openai",
@@ -336,10 +264,10 @@ class chatgpt(BaseLLM):
         # 构造请求数据
         request_data = {
             "model": model or self.engine,
-            "messages": copy.deepcopy(self.conversation[convo_id]) if pass_history else [
-                {"role": "system","content": self.system_prompt + "\n\n" + self.get_latest_file_content()},
-                {"role": role, "content": prompt}
-            ],
+            "messages": self.conversation[convo_id].render_latest() if pass_history else Messages(
+                SystemMessage(self.system_prompt, self.conversation[convo_id].provider("files")),
+                UserMessage(prompt)
+            ),
             "stream": stream,
             "temperature": kwargs.get("temperature", self.temperature)
         }
@@ -655,7 +583,7 @@ class chatgpt(BaseLLM):
                         else:
                             yield chunk
                 if tool_name == "read_file" and "<tool_error>" not in tool_response:
-                    self.latest_file_content[tool_info['parameter']["file_path"]] = tool_response
+                    self.conversation[convo_id].provider("files").update(tool_info['parameter']["file_path"], tool_response)
                     all_responses.append(f"[{tool_name}({tool_args}) Result]:\n\nRead file successfully! The file content has been updated in the tag <latest_file_content>.")
                 elif tool_name == "write_to_file" and "<tool_error>" not in tool_response:
                     all_responses.append(f"[{tool_name} Result]:\n\n{tool_response}")
@@ -998,9 +926,8 @@ class chatgpt(BaseLLM):
         Reset the conversation
         """
         self.system_prompt = system_prompt or self.system_prompt
-        self.latest_file_content = {}
-        self.conversation[convo_id] = [
-            {"role": "system", "content": self.system_prompt},
-        ]
+        self.conversation[convo_id] = Messages(
+            SystemMessage(Texts("system_prompt", self.system_prompt), self.conversation[convo_id].provider("files")),
+        )
         self.tokens_usage[convo_id] = 0
         self.current_tokens[convo_id] = 0

{aient-1.1.91.dist-info → aient-1.1.92.dist-info}/METADATA

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

{aient-1.1.91.dist-info → aient-1.1.92.dist-info}/RECORD

@@ -1,7 +1,9 @@
 aient/__init__.py,sha256=SRfF7oDVlOOAi6nGKiJIUK6B_arqYLO9iSMp-2IZZps,21
-aient/architext/test.py,sha256=n2hUwHOPhOdpHRLCPuqeR-r4hGK43SF1WKaLdivdmC8,10021
 aient/architext/architext/__init__.py,sha256=79Ih1151rfcqZdr7F8HSZSTs_iT2SKd1xCkehMsXeXs,19
-aient/architext/architext/core.py,sha256=vww5nxNyoGXmWwmNEfLMGfv7k4GhCDZSiRqKzyqYv_8,10070
+aient/architext/architext/core.py,sha256=i8UGEwoAS7Cwat3wB-jU8EC23gSwAXjFiwEc8QknE-k,17875
+aient/architext/test/openai_client.py,sha256=Dqtbmubv6vwF8uBqcayG0kbsiO65of7sgU2-DRBi-UM,4590
+aient/architext/test/test.py,sha256=CeFyloj-RJRk8hr6QDLjnRETG-KylD2bo3MVq5GVKPA,40597
+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
 aient/core/models.py,sha256=KMlCRLjtq1wQHZTJGqnbWhPS2cHq6eLdnk7peKDrzR8,7490
@@ -15,7 +17,7 @@ aient/core/test/test_payload.py,sha256=8jBiJY1uidm1jzL-EiK0s6UGmW9XkdsuuKFGrwFhF
 aient/models/__init__.py,sha256=ZTiZgbfBPTjIPSKURE7t6hlFBVLRS9lluGbmqc1WjxQ,43
 aient/models/audio.py,sha256=kRd-8-WXzv4vwvsTGwnstK-WR8--vr9CdfCZzu8y9LA,1934
 aient/models/base.py,sha256=-nnihYnx-vHZMqeVO9ljjt3k4FcD3n-iMk4tT-10nRQ,7232
-aient/models/chatgpt.py,sha256=2RaObZmliqJlGveOSWbwgpscjPWk7R1RmxwbEAH0xXo,47315
+aient/models/chatgpt.py,sha256=EqvIcB6R_JGmvD5kEfR1ZRQYLfyGpmy4PvfRx8hMIKE,42019
 aient/plugins/__init__.py,sha256=p3KO6Aa3Lupos4i2SjzLQw1hzQTigOAfEHngsldrsyk,986
 aient/plugins/arXiv.py,sha256=yHjb6PS3GUWazpOYRMKMzghKJlxnZ5TX8z9F6UtUVow,1461
 aient/plugins/config.py,sha256=TGgZ5SnNKZ8MmdznrZ-TEq7s2ulhAAwTSKH89bci3dA,7079
@@ -33,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.1.91.dist-info/licenses/LICENSE,sha256=XNdbcWldt0yaNXXWB_Bakoqnxb3OVhUft4MgMA_71ds,1051
-aient-1.1.91.dist-info/METADATA,sha256=SByLoF4YtLoR1QxQ1qmVim3FbMHQIgD-b1OcxliO4bw,4842
-aient-1.1.91.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-aient-1.1.91.dist-info/top_level.txt,sha256=3oXzrP5sAVvyyqabpeq8A2_vfMtY554r4bVE-OHBrZk,6
-aient-1.1.91.dist-info/RECORD,,
+aient-1.1.92.dist-info/licenses/LICENSE,sha256=XNdbcWldt0yaNXXWB_Bakoqnxb3OVhUft4MgMA_71ds,1051
+aient-1.1.92.dist-info/METADATA,sha256=EYclEjlBbW-N9rjlfZPA-1HlyQST_DmK3qqBNabzC3g,4842
+aient-1.1.92.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+aient-1.1.92.dist-info/top_level.txt,sha256=3oXzrP5sAVvyyqabpeq8A2_vfMtY554r4bVE-OHBrZk,6
+aient-1.1.92.dist-info/RECORD,,

aient/architext/test.py

@@ -1,226 +0,0 @@
-import unittest
-from unittest.mock import AsyncMock
-from architext import *
-
-# ==============================================================================
-# 单元测试部分
-# ==============================================================================
-class TestContextManagement(unittest.IsolatedAsyncioTestCase):
-
-    def setUp(self):
-        """在每个测试前设置环境"""
-        self.system_prompt_provider = Texts("system_prompt", "你是一个AI助手。")
-        self.tools_provider = Tools(tools_json=[{"name": "read_file"}])
-        self.files_provider = Files()
-
-    async def test_a_initial_construction_and_render(self):
-        """测试优雅的初始化和首次渲染"""
-        messages = Messages(
-            SystemMessage(self.system_prompt_provider, self.tools_provider),
-            UserMessage(self.files_provider, Texts("user_input", "这是我的初始问题。"))
-        )
-
-        self.assertEqual(len(messages), 2)
-        rendered = await messages.render_latest()
-
-        self.assertEqual(len(rendered), 2)
-        self.assertIn("<tools>", rendered[0]['content'])
-        self.assertNotIn("<files>", rendered[1]['content'])
-
-    async def test_b_provider_passthrough_and_refresh(self):
-        """测试通过 mock 验证缓存和刷新逻辑"""
-        # 我们真正关心的是 _fetch_content 是否被不必要地调用
-        # 所以我们 mock 底层的它，而不是 refresh 方法
-        original_fetch_content = self.files_provider._fetch_content
-        self.files_provider._fetch_content = AsyncMock(side_effect=original_fetch_content)
-
-        messages = Messages(UserMessage(self.files_provider))
-
-        # 1. 首次刷新
-        self.files_provider.update("path1", "content1")
-        await messages.refresh()
-        # _fetch_content 应该被调用了 1 次
-        self.assertEqual(self.files_provider._fetch_content.call_count, 1)
-
-        # 2. 再次刷新，内容未变，不应再次调用 _fetch_content
-        await messages.refresh()
-        # 调用次数应该仍然是 1，证明缓存生效
-        self.assertEqual(self.files_provider._fetch_content.call_count, 1)
-
-        # 3. 更新文件内容，这会标记 provider 为 stale
-        self.files_provider.update("path2", "content2")
-
-        # 4. 再次刷新，现在应该会重新调用 _fetch_content
-        await messages.refresh()
-        rendered = messages.render()
-        # 调用次数应该变为 2
-        self.assertEqual(self.files_provider._fetch_content.call_count, 2)
-        # 并且渲染结果包含了新内容
-        self.assertIn("content2", rendered[0]['content'])
-
-    async def test_c_global_pop_and_indexed_insert(self):
-        """测试全局pop和通过索引insert的功能"""
-        messages = Messages(
-            SystemMessage(self.system_prompt_provider, self.tools_provider),
-            UserMessage(self.files_provider)
-        )
-
-        # 验证初始状态
-        initial_rendered = await messages.render_latest()
-        self.assertTrue(any("<tools>" in msg['content'] for msg in initial_rendered if msg['role'] == 'system'))
-
-        # 全局弹出 'tools' Provider
-        popped_tools_provider = messages.pop("tools")
-        self.assertIs(popped_tools_provider, self.tools_provider)
-
-        # 验证 pop 后的状态
-        rendered_after_pop = messages.render()
-        self.assertFalse(any("<tools>" in msg['content'] for msg in rendered_after_pop if msg['role'] == 'system'))
-
-        # 通过索引将弹出的provider插入到UserMessage的开头
-        messages[1].insert(0, popped_tools_provider)
-
-        # 验证 insert 后的状态
-        rendered_after_insert = messages.render()
-        user_message_content = next(msg['content'] for msg in rendered_after_insert if msg['role'] == 'user')
-        self.assertTrue(user_message_content.startswith("<tools>"))
-
-    async def test_d_multimodal_rendering(self):
-        """测试多模态（文本+图片）渲染"""
-        # Create a dummy image file for the test
-        dummy_image_path = "test_dummy_image.png"
-        with open(dummy_image_path, "w") as f:
-            f.write("dummy content")
-
-        messages = Messages(
-            UserMessage(
-                Texts("prompt", "Describe the image."),
-                Images(dummy_image_path) # Test with optional name
-            )
-        )
-
-        rendered = await messages.render_latest()
-        self.assertEqual(len(rendered), 1)
-
-        content = rendered[0]['content']
-        self.assertIsInstance(content, list)
-        self.assertEqual(len(content), 2)
-
-        # Check text part
-        self.assertEqual(content[0]['type'], 'text')
-        self.assertEqual(content[0]['text'], 'Describe the image.')
-
-        # Check image part
-        self.assertEqual(content[1]['type'], 'image_url')
-        self.assertIn('data:image/png;base64,', content[1]['image_url']['url'])
-
-        # Clean up the dummy file
-        import os
-        os.remove(dummy_image_path)
-
-    async def test_e_multimodal_type_switching(self):
-        """测试多模态消息在pop图片后是否能正确回退到字符串渲染"""
-        dummy_image_path = "test_dummy_image_2.png"
-        with open(dummy_image_path, "w") as f:
-            f.write("dummy content")
-
-        messages = Messages(
-            UserMessage(
-                Texts("prefix", "Look at this:"),
-                Images(dummy_image_path, name="image"), # Explicit name for popping
-                Texts("suffix", "Any thoughts?")
-            )
-        )
-
-        # 1. Initial multimodal render
-        rendered_multi = await messages.render_latest()
-        content_multi = rendered_multi[0]['content']
-        self.assertIsInstance(content_multi, list)
-        self.assertEqual(len(content_multi), 3) # prefix, image, suffix
-
-        # 2. Pop the image
-        popped_image = messages.pop("image")
-        self.assertIsNotNone(popped_image)
-
-        # 3. Render again, should fall back to string content
-        rendered_str = messages.render() # No refresh needed
-        content_str = rendered_str[0]['content']
-        self.assertIsInstance(content_str, str)
-        self.assertEqual(content_str, "Look at this:\n\nAny thoughts?")
-
-        # Clean up
-        import os
-        os.remove(dummy_image_path)
-
-    def test_f_message_merging(self):
-        """测试初始化和追加时自动合并消息的功能"""
-        # 1. Test merging during initialization
-        messages = Messages(
-            UserMessage(Texts("part1", "Hello,")),
-            UserMessage(Texts("part2", "world!")),
-            SystemMessage(Texts("system", "System prompt.")),
-            UserMessage(Texts("part3", "How are you?"))
-        )
-        # Should be merged into: User, System, User
-        self.assertEqual(len(messages), 3)
-        self.assertEqual(len(messages[0]._items), 2) # First UserMessage has 2 items
-        self.assertEqual(messages[0]._items[1].name, "part2")
-        self.assertEqual(messages[1].role, "system")
-        self.assertEqual(messages[2].role, "user")
-
-        # 2. Test merging during append
-        messages.append(UserMessage(Texts("part4", "I am fine.")))
-        self.assertEqual(len(messages), 3) # Still 3 messages
-        self.assertEqual(len(messages[2]._items), 2) # Last UserMessage now has 2 items
-        self.assertEqual(messages[2]._items[1].name, "part4")
-
-        # 3. Test appending a different role
-        messages.append(SystemMessage(Texts("system2", "Another prompt.")))
-        self.assertEqual(len(messages), 4) # Should not merge
-        self.assertEqual(messages[3].role, "system")
-
-    async def test_g_state_inconsistency_on_direct_message_modification(self):
-        """
-        测试当直接在 Message 对象上执行 pop 操作时，
-        顶层 Messages 对象的 _providers_index 是否会产生不一致。
-        """
-        messages = Messages(
-            SystemMessage(self.system_prompt_provider, self.tools_provider),
-            UserMessage(self.files_provider)
-        )
-
-        # 0. 先刷新一次，确保所有 provider 的 cache 都已填充
-        await messages.refresh()
-
-        # 1. 初始状态：'tools' 提供者应该在索引中
-        self.assertIsNotNone(messages.provider("tools"), "初始状态下 'tools' 提供者应该能被找到")
-        self.assertIs(messages.provider("tools"), self.tools_provider)
-
-        # 2. 直接在子消息对象上执行 pop 操作
-        system_message = messages[0]
-        popped_provider = system_message.pop("tools")
-
-        # 验证是否真的从 Message 对象中弹出了
-        self.assertIs(popped_provider, self.tools_provider, "应该从 SystemMessage 中成功弹出 provider")
-        self.assertNotIn(self.tools_provider, system_message.providers(), "provider 不应再存在于 SystemMessage 的 providers 列表中")
-
-        # 3. 核心问题：检查顶层 Messages 的索引
-        # 在理想情况下，直接修改子消息应该同步更新顶层索引。
-        # 因此，我们断言 provider 现在应该是找不到的。这个测试现在应该会失败。
-        provider_after_pop = messages.provider("tools")
-        self.assertIsNone(provider_after_pop, "BUG: 直接从子消息中 pop 后，顶层索引未同步，仍然可以找到 provider")
-
-        # 4. 进一步验证：渲染结果和索引内容不一致
-        # 渲染结果应该不再包含 tools 内容，因为 Message 对象本身是正确的
-        rendered_messages = messages.render()
-        self.assertGreater(len(rendered_messages), 0, "渲染后的消息列表不应为空")
-        rendered_content = rendered_messages[0]['content']
-        self.assertNotIn("<tools>", rendered_content, "渲染结果中不应再包含 'tools' 的内容，证明数据源已更新")
-
-
-if __name__ == '__main__':
-    # 为了在普通脚本环境中运行，添加这两行
-    suite = unittest.TestSuite()
-    suite.addTest(unittest.makeSuite(TestContextManagement))
-    runner = unittest.TextTestRunner()
-    runner.run(suite)