Added timeout and limit to script entries

This commit is contained in:
Niels Geens
2024-11-13 17:48:39 +01:00
parent 65478e0b17
commit 834ce12e8f
21 changed files with 262 additions and 141 deletions

View File

@@ -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}")