75 lines
2.6 KiB
Python
75 lines
2.6 KiB
Python
from abc import ABC, abstractmethod
|
|
from typing import List
|
|
import xml.etree.ElementTree as ET
|
|
import time
|
|
|
|
from .llm_engine import LlmEngine
|
|
from .response_parser import ResponseParser
|
|
from .system_metrics import SystemMetrics
|
|
from .util import pretty_print_element
|
|
from .working_memory import WorkingMemory
|
|
from .xml_validator import XMLValidator
|
|
|
|
class BaseAgent(ABC):
|
|
"""
|
|
Abstract base class for SIA agents.
|
|
|
|
Provides core functionality for maintaining working memory, system metrics,
|
|
and coordinating components for LLM inference.
|
|
"""
|
|
|
|
def __init__(self,
|
|
action_schema: str,
|
|
working_memory: WorkingMemory,
|
|
system_metrics: SystemMetrics,
|
|
llm: LlmEngine,
|
|
validator: XMLValidator,
|
|
parser: ResponseParser):
|
|
"""
|
|
Initialize agent with required components.
|
|
"""
|
|
self._working_memory = working_memory
|
|
self._metrics = system_metrics
|
|
self._llm = llm
|
|
self._validator = validator
|
|
self._parser = parser
|
|
self._action_schema = action_schema
|
|
|
|
def __del__(self):
|
|
"""Clean up resources on deletion."""
|
|
if hasattr(self, '_metrics'):
|
|
self._metrics.stop()
|
|
|
|
def _compile_context(self) -> str:
|
|
"""
|
|
Compile the current context for LLM inference.
|
|
Includes system metrics and working memory entries.
|
|
|
|
Returns:
|
|
str: Complete context as XML string
|
|
"""
|
|
# Get memory context and calculate size
|
|
memory_context = self._working_memory.generate_context()
|
|
context_size = len(memory_context) / 100
|
|
|
|
# Get system metrics
|
|
metrics_data = self._metrics.get_metrics()
|
|
|
|
# Create context element with metrics
|
|
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("context", str(round(context_size * 100)))
|
|
context.set("stdin", str(self._parser.io_buffer.buffer_length()))
|
|
|
|
# Add memory entries
|
|
for entry in memory_context:
|
|
context.append(entry)
|
|
|
|
return pretty_print_element(context)
|