Fix pretty printing

This commit is contained in:
2024-11-02 15:22:32 +01:00
parent a58ee40c2c
commit 96d14fc46d
6 changed files with 213 additions and 33 deletions

View File

@@ -3,10 +3,11 @@ from typing import List
import xml.etree.ElementTree as ET
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
from .response_parser import ResponseParser
class BaseAgent(ABC):
"""
@@ -52,4 +53,4 @@ class BaseAgent(ABC):
context = self._metrics.generate_context(context_size)
for entry in memory_context:
context.append(entry)
return ET.tostring(context, encoding="unicode")
return pretty_print_element(context)