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

Server side implementation for interactive entry edits

Browse Source
This commit is contained in:
geens
2024-11-23 15:34:33 +01:00
parent bb74fd3621
commit dbc0068a82
28 changed files with 749 additions and 553 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
sia/base_agent.py
View File

@@ -34,11 +34,6 @@ class BaseAgent(ABC):
self._metrics = metrics
self._validator = validator
self._parser = parser
def __del__(self):
"""Clean up resources on deletion."""
if hasattr(self, '_metrics'):
self._metrics.stop()
@property
def system_prompt(self) -> str:
@@ -59,14 +54,12 @@ class BaseAgent(ABC):
# Create context element
context = ET.Element("context")
context.set("time", metrics_data["timestamp"])
context.set("cpu", str(metrics_data["cpu"]))
context.set("gpu", str(metrics_data["gpu"]))
context.set("memory_used", str(metrics_data["memory_used"]))
context.set("memory_total", str(metrics_data["memory_total"]))
context.set("disk_used", str(metrics_data["disk_used"]))
context.set("disk_total", str(metrics_data["disk_total"]))
context.set("stdin", str(self._parser.io_buffer.buffer_length()))
context.set("context", "100")
context.set("context", "100%")
for entry in memory_context:
context.append(entry)
@@ -79,6 +72,6 @@ class BaseAgent(ABC):
context_usage = (float(token_count) / float(token_limit)) * 100.0
# Update context usage metric
context.set("context", str(round(context_usage, 2)))
context.set("context", f"{str(round(context_usage, 2))}%")
return pretty_print_element(context)