Added timeout and limit to script entries
This commit is contained in:
@@ -145,8 +145,9 @@ class BaseAgentTest(unittest.TestCase):
|
||||
self.assertEqual(len(list(root)), 0)
|
||||
|
||||
def test_compile_context_with_entries(self):
|
||||
# Add test entry to working memory
|
||||
entry = ReasoningEntry("test reasoning", "test-id", self.test_timestamp)
|
||||
"""Test context compilation with a single entry."""
|
||||
# Add test entry to working memory - note id and timestamp are now first
|
||||
entry = ReasoningEntry("test-id", self.test_timestamp, "test reasoning")
|
||||
self.agent._working_memory.add_entry(entry)
|
||||
|
||||
context = self.agent._compile_context()
|
||||
@@ -166,9 +167,9 @@ class BaseAgentTest(unittest.TestCase):
|
||||
|
||||
def test_multiple_entries(self):
|
||||
"""Test handling multiple entries in working memory."""
|
||||
# Add multiple entries
|
||||
# Add multiple entries - note id and timestamp are now first
|
||||
entries = [
|
||||
ReasoningEntry(f"test {i}", f"id-{i}", self.test_timestamp)
|
||||
ReasoningEntry(f"id-{i}", self.test_timestamp, f"test {i}")
|
||||
for i in range(3)
|
||||
]
|
||||
|
||||
@@ -186,4 +187,4 @@ class BaseAgentTest(unittest.TestCase):
|
||||
self.assertEqual(len(reasoning_elems), 3)
|
||||
for i, elem in enumerate(reasoning_elems):
|
||||
self.assertEqual(elem.get("id"), f"id-{i}")
|
||||
self.assertEqual(elem.text.strip(), f"test {i}")
|
||||
self.assertEqual(elem.text.strip(), f"test {i}")
|
||||
Reference in New Issue
Block a user