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

Added memory editing functionality

Browse Source
This commit is contained in:
geens
2024-11-24 18:00:27 +01:00
parent 2c820f6ead
commit b6c1a95abd
15 changed files with 688 additions and 53 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
sia/entry/entry_factory.py
View File

@@ -54,7 +54,7 @@ class EntryFactory:
if "script" in data:
entry.script = data["script"]
if "timeout" in data:
entry.timeout = float(data["timeout"]) if data["timeout"] else None
entry.timeout = float(data["timeout"]) if data["timeout"] is not None else None
if "limit" in data:
entry.limit = int(data["limit"]) if data["limit"] else None
if "stdout" in data:
@@ -62,7 +62,7 @@ class EntryFactory:
if "stderr" in data:
entry.stderr = data["stderr"]
if "exit_code" in data:
entry.exit_code = int(data["exit_code"]) if data["exit_code"] else None
entry.exit_code = int(data["exit_code"]) if data["exit_code"] is not None else None
if "executed" in data:
entry.executed = bool(data["executed"])
if "timed_out" in data:
@@ -72,7 +72,7 @@ class EntryFactory:
if "script" in data:
entry.script = data["script"]
if "timeout" in data:
entry.timeout = float(data["timeout"]) if data["timeout"] else None
entry.timeout = float(data["timeout"]) if data["timeout"] is not None else None
if "limit" in data:
entry.limit = int(data["limit"]) if data["limit"] else None
if "stdout" in data:
@@ -80,7 +80,7 @@ class EntryFactory:
if "stderr" in data:
entry.stderr = data["stderr"]
if "exit_code" in data:
entry.exit_code = int(data["exit_code"]) if data["exit_code"] else None
entry.exit_code = int(data["exit_code"]) if data["exit_code"] is not None else None
if "executed" in data:
entry.executed = bool(data["executed"])
if "timed_out" in data:
@@ -94,7 +94,7 @@ class EntryFactory:
if "stderr" in data:
entry.stderr = data["stderr"]
if "exit_code" in data:
entry.exit_code = int(data["exit_code"]) if data["exit_code"] else None
entry.exit_code = int(data["exit_code"]) if data["exit_code"] is not None else None
elif isinstance(entry, ParseErrorEntry):
if "content" in data:

8
sia/entry/read_entry.py
View File

@@ -34,6 +34,14 @@ class ReadEntry(Entry):
self.content = self._io_buffer.read()
self.read = True
self.notify_change()
def reset(self) -> None:
"""
Reset the entry state to its initial state.
"""
self.read = False
self.content = ""
self.notify_change()
def generate_context(self) -> ET.Element:
"""

8
sia/entry/repeat_entry.py
View File

@@ -53,9 +53,9 @@ class RepeatEntry(Entry):
self.stdout = process.stdout
self.stderr = process.stderr
self.exit_code = process.returncode
self._timed_out = False
self.timed_out = False
except subprocess.TimeoutExpired as e:
self._timed_out = True
self.timed_out = True
self.notify_change()
def generate_context(self) -> ET.Element:
@@ -69,7 +69,7 @@ class RepeatEntry(Entry):
if self.timeout:
element.set("timeout", str(self.timeout))
element.text = self.script
if self._timed_out:
if self.timed_out:
element.set("timed_out", "true")
elif self.exit_code is not None:
element.set("exit_code", str(self.exit_code))
@@ -101,5 +101,5 @@ class RepeatEntry(Entry):
"stdout": self.stdout,
"stderr": self.stderr,
"exit_code": self.exit_code,
"timed_out": self._timed_out
"timed_out": self.timed_out
}

1
sia/entry/single_entry.py
View File

@@ -45,6 +45,7 @@ class SingleEntry(Entry):
self.stdout = ""
self.stderr = ""
self.exit_code = None
self.notify_change()
def update(self) -> None:
"""

8
sia/entry/write_entry.py
View File

@@ -36,6 +36,13 @@ class WriteEntry(Entry):
self._io_buffer.write(self.content)
self.written = True
self.notify_change()
def reset(self) -> None:
"""
Reset the entry state to its initial state.
"""
self.written = False
self.notify_change()
def generate_context(self) -> ET.Element:
"""
@@ -56,4 +63,5 @@ class WriteEntry(Entry):
"type": "write",
"id": self.id,
"content": self.content,
"written": self.written,
}