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

Added output length on truncated scripts

Browse Source
This commit is contained in:
geens
2024-11-23 09:36:34 +01:00
parent d9d7773369
commit bb74fd3621
4 changed files with 9 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
.gitignore vendored
View File

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

3
run.sh
View File

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

4
sia/repeat_entry.py
View File

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

2
sia/single_entry.py
View File

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