+
+07. role=assistant tool_call_id=call_BVR80Veg57rrDVsaxSwronfh (inline) result_length=4
+ preview: None
+08. role=function tool_call_id=call_BVR80Veg57rrDVsaxSwronfh (inline) result_length=294
+ preview: [browser_snapshot length=225 characters] BEGIN_SNAPSHOT
City Weather
Currently 68°F, partly cl
+09. role=assistant content=The weather dashboard widget has been stored exactly as you provided it. You can reference or retrieve this snapshot any time. Let me know i
+10. role=user content=Confirm that both snapshots have been saved.
+11. role=assistant content=Confirmation: Both snapshots have been successfully saved.
+
+1. NovaPhone smartphone page — contains the launch event details, specs, compari
+
+>>> Step 3: Ask question requiring BOTH cached snapshots
+Assistant response:
+Based on the two snapshots I retrieved:
+
+1. **NovaPhone X Ultra battery capacity:** 5,500 mAh
+
+2. **Current temperature:** 68°F (partly cloudy)
+
+3. **Creative comparison:**
+ Interestingly, the NovaPhone X Ultra's battery capacity (5,500 mAh) is about
+ 80 times larger than the current temperature in degrees Fahrenheit (68°F)!
+
+ While the phone can power through your day with its massive 5,500 mAh battery
+ and 120W wired charging, the weather outside is a pleasant 68°F — perfect
+ conditions for testing that new phone outdoors without worrying about
+ overheating or cold-induced battery drain. The phone's battery is built for
+ performance in any weather!
+
+>>> Step 4: Verify agent can access single snapshot
+Assistant response:
+According to the weather widget snapshot, the sunset time is **7:42 PM**.
+''' # noqa: E501
diff --git a/test/test_chat_agent_tool_cache.py b/test/test_chat_agent_tool_cache.py
new file mode 100644
index 0000000000..02a86a6711
--- /dev/null
+++ b/test/test_chat_agent_tool_cache.py
@@ -0,0 +1,176 @@
+# ========= Copyright 2023-2024 @ CAMEL-AI.org. All Rights Reserved. =========
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ========= Copyright 2023-2024 @ CAMEL-AI.org. All Rights Reserved. =========
+import pytest
+
+from camel.agents import ChatAgent
+from camel.messages import FunctionCallingMessage
+from camel.models.stub_model import StubModel
+from camel.types import ModelType
+
+
+@pytest.mark.parametrize("threshold", [20])
+def test_tool_output_caching(tmp_path, threshold):
+ agent = ChatAgent(
+ system_message="You are a tester.",
+ model=StubModel(model_type=ModelType.STUB),
+ tool_call_cache_threshold=threshold,
+ tool_call_cache_dir=tmp_path,
+ )
+
+ long_result = "A" * (threshold + 10)
+ short_result = "short"
+
+ agent._record_tool_calling(
+ "dummy_tool",
+ args={"value": 1},
+ result=long_result,
+ tool_call_id="call-1",
+ )
+
+ history = [
+ entry
+ for entry in agent._tool_output_history
+ if entry.tool_call_id == "call-1"
+ ]
+ assert history and not history[0].cached
+
+ agent._record_tool_calling(
+ "dummy_tool",
+ args={"value": 2},
+ result=short_result,
+ tool_call_id="call-2",
+ )
+
+ cached_entry = next(
+ entry
+ for entry in agent._tool_output_history
+ if entry.tool_call_id == "call-1"
+ )
+ assert not cached_entry.cached
+
+ flushed = agent._cache_tool_calls()
+ assert flushed == 1
+
+ cached_entry = next(
+ entry
+ for entry in agent._tool_output_history
+ if entry.tool_call_id == "call-1"
+ )
+ assert cached_entry.cached
+ assert cached_entry.cache_id
+
+ cache_file = tmp_path / f"{cached_entry.cache_id}.txt"
+ assert cache_file.exists()
+ assert long_result in cache_file.read_text(encoding="utf-8")
+
+ records = agent.memory.retrieve()
+ cached_message = None
+ for record in records:
+ message = record.memory_record.message
+ if (
+ isinstance(message, FunctionCallingMessage)
+ and getattr(message, "tool_call_id", "") == "call-1"
+ and getattr(message, "result", None)
+ ):
+ cached_message = message
+ break
+
+ assert cached_message is not None
+ assert cached_entry.cache_id in cached_message.result
+ assert agent._cache_lookup_tool_name in cached_message.result
+ assert cached_message.result != long_result
+
+ retrieved = agent.retrieve_cached_tool_output(cached_entry.cache_id)
+ assert long_result in retrieved
+
+
+def test_tool_output_history_cleared_on_reset(tmp_path):
+ agent = ChatAgent(
+ system_message="Cache reset tester.",
+ model=StubModel(model_type=ModelType.STUB),
+ tool_call_cache_threshold=10,
+ tool_call_cache_dir=tmp_path,
+ )
+
+ agent._record_tool_calling(
+ "dummy_tool",
+ args={"value": "a"},
+ result="A" * 20,
+ tool_call_id="call-initial",
+ )
+ assert agent._tool_output_history
+
+ agent.clear_memory()
+ assert agent._tool_output_history == []
+
+ agent._record_tool_calling(
+ "dummy_tool",
+ args={"value": "b"},
+ result="B" * 20,
+ tool_call_id="call-after-clear",
+ )
+ assert len(agent._tool_output_history) == 1
+
+ agent.reset()
+ assert agent._tool_output_history == []
+
+
+def test_retrieve_multiple_cached_outputs(tmp_path):
+ agent = ChatAgent(
+ system_message="Multiple cache retrieval tester.",
+ model=StubModel(model_type=ModelType.STUB),
+ tool_call_cache_threshold=10,
+ tool_call_cache_dir=tmp_path,
+ )
+
+ # Record multiple tool calls
+ agent._record_tool_calling(
+ "tool_1",
+ args={"value": "a"},
+ result="A" * 20,
+ tool_call_id="call-1",
+ )
+ agent._record_tool_calling(
+ "tool_2",
+ args={"value": "b"},
+ result="B" * 30,
+ tool_call_id="call-2",
+ )
+
+ # Cache them
+ cached_count = agent._cache_tool_calls()
+ assert cached_count == 2
+
+ # Get cache IDs
+ cache_ids = [
+ entry.cache_id for entry in agent._tool_output_history if entry.cached
+ ]
+ assert len(cache_ids) == 2
+
+ # Test single retrieval (same tool, single ID)
+ result1 = agent.retrieve_cached_tool_output(cache_ids[0])
+ assert "A" * 20 in result1
+
+ # Test multiple retrieval (same tool, multiple IDs)
+ import json
+
+ result_multiple = agent.retrieve_cached_tool_output(
+ f"{cache_ids[0]}, {cache_ids[1]}"
+ )
+ results_dict = json.loads(result_multiple)
+
+ assert cache_ids[0] in results_dict
+ assert cache_ids[1] in results_dict
+ assert "A" * 20 in results_dict[cache_ids[0]]
+ assert "B" * 30 in results_dict[cache_ids[1]]