From e80d1820927810708d1575d782f80fb108db2b55 Mon Sep 17 00:00:00 2001
From: Sambhav Dixit <94298612+sambhavnoobcoder@users.noreply.github.com>
Date: Sun, 27 Oct 2024 19:10:37 +0530
Subject: [PATCH] Add test for timestamp in log_step_metadata
- Implemented test_log_step_metadata_timestamp to verify that a timestamp is included in the log result when logging step metadata.
---
tests/agents/test_agent_logging.py | 3 +++
1 file changed, 3 insertions(+)
diff --git a/tests/agents/test_agent_logging.py b/tests/agents/test_agent_logging.py
index cbfb8b80..33e59843 100644
--- a/tests/agents/test_agent_logging.py
+++ b/tests/agents/test_agent_logging.py
@@ -38,3 +38,6 @@ class TestAgentLogging(unittest.TestCase):
log_result = self.agent.log_step_metadata(1, "prompt", "response")
self.assertEqual(log_result['memory_usage']['long_term'], {})
+ def test_log_step_metadata_timestamp(self):
+ log_result = self.agent.log_step_metadata(1, "prompt", "response")
+ self.assertIn('timestamp', log_result)