Added output length on truncated scripts

This commit is contained in:
2024-11-23 09:36:34 +01:00
parent d9d7773369
commit bb74fd3621
4 changed files with 9 additions and 3 deletions

3
.gitignore vendored
View File

@@ -1,4 +1,5 @@
.env
pdf/
model/
claude.txt
claude.txt
__pycache__/

3
run.sh
View File

@@ -13,5 +13,6 @@ docker run \
-p 8080:8080 \
-v /$(pwd)/model/:/root/model/ \
-v /$(pwd)/.env:/root/.env \
-v /$(pwd)/iterations/:/root/sia/iterations \
-v /$(pwd)/iterations/:/root/sia/iterations/ \
-v /$(pwd)/.git/:/root/sia/.git/ \
sia "$@"

View File

@@ -43,7 +43,7 @@ class RepeatEntry(Entry):
super().__init__(id, timestamp)
self._script = script
self._timeout = timeout
self._limit = limit or self.default_limit
self._limit = limit
self._stdout = ""
self._stderr = ""
self._exit_code: Optional[int] = None
@@ -110,10 +110,12 @@ class RepeatEntry(Entry):
element.set("limit", str(self._limit))
if len(self._stdout) > (self._limit or self.default_limit):
element.set("stdout_truncated", "true")
element.set("stdout_length", str(len(self._stdout)))
stdout_elem = ET.SubElement(element, "stdout")
stdout_elem.text = self._stdout[:(self._limit or self.default_limit)]
if len(self._stderr) > (self._limit or self.default_limit):
element.set("stderr_truncated", "true")
element.set("stderr_length", str(len(self._stderr)))
stderr_elem = ET.SubElement(element, "stderr")
stderr_elem.text = self._stderr[:(self._limit or self.default_limit)]

View File

@@ -113,10 +113,12 @@ class SingleEntry(Entry):
element.set("limit", str(self._limit))
if len(self._stdout) > (self._limit or self.default_limit):
element.set("stdout_truncated", "true")
element.set("stdout_length", str(len(self._stdout)))
stdout_elem = ET.SubElement(element, "stdout")
stdout_elem.text = self._stdout[:(self._limit or self.default_limit)]
if len(self._stderr) > (self._limit or self.default_limit):
element.set("stderr_truncated", "true")
element.set("stderr_length", str(len(self._stderr)))
stderr_elem = ET.SubElement(element, "stderr")
stderr_elem.text = self._stderr[:(self._limit or self.default_limit)]
return element