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