llm/SIA
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fixed stdin buffer length in context

Browse Source
This commit is contained in:
Niels Geens
2024-11-14 13:36:20 +01:00
parent 834ce12e8f
commit 325870a81a
6 changed files with 27 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
test/base_agent_test.py
View File

@@ -66,7 +66,7 @@ class BaseAgentTest(unittest.TestCase):
self.parser = ResponseParser(self.io_buffer)
# Mock metrics context generator
def generate_context(context_size):
def generate_context(context_size, stdin_buffer_length):
metrics_elem = ET.Element("context")
metrics_elem.set("time", "2024-10-31T12:00:00Z")
metrics_elem.set("cpu", "10")
@@ -76,7 +76,7 @@ class BaseAgentTest(unittest.TestCase):
metrics_elem.set("disk_used", "5000")
metrics_elem.set("disk_total", "10000")
metrics_elem.set("context", str(int(context_size * 100)))
metrics_elem.set("stdin", "0")
metrics_elem.set("stdin", stdin_buffer_length)
return metrics_elem
self.mock_metrics.generate_context.side_effect = generate_context

8
test/system_metrics_test.py
View File

@@ -96,7 +96,7 @@ class SystemMetricsTest(unittest.TestCase):
def test_initial_context(self):
"""Test initial context generation"""
context = self.metrics.generate_context(0.5)
context = self.metrics.generate_context(0.5, 10)
# Verify it's an XML element
self.assertIsInstance(context, ET.Element)
@@ -125,13 +125,13 @@ class SystemMetricsTest(unittest.TestCase):
])
# Validate stdin buffer size
self.assertEqual(context.get("stdin"), "0")
self.assertEqual(context.get("stdin"), 10)
def test_multiple_context_generations(self):
"""Test multiple context generations have different timestamps"""
context1 = self.metrics.generate_context(0.5)
context1 = self.metrics.generate_context(0.5, 10)
time.sleep(1)
context2 = self.metrics.generate_context(0.5)
context2 = self.metrics.generate_context(0.5, 10)
self.assertNotEqual(
context1.get("time"),

4
test/web_agent_test.py
View File

@@ -46,7 +46,7 @@ class WebAgentTest(unittest.TestCase):
self.parser = ResponseParser(self.io_buffer)
# Mock metrics context generator
def generate_context(context_size):
def generate_context(context_size, stdin_buffer_length):
metrics_elem = ET.Element("context")
metrics_elem.set("time", "2024-10-31T12:00:00Z")
metrics_elem.set("cpu", "10")
@@ -56,7 +56,7 @@ class WebAgentTest(unittest.TestCase):
metrics_elem.set("disk_used", "5000")
metrics_elem.set("disk_total", "10000")
metrics_elem.set("context", str(int(context_size * 100)))
metrics_elem.set("stdin", "0")
metrics_elem.set("stdin", stdin_buffer_length)
return metrics_elem
self.mock_metrics.generate_context.side_effect = generate_context