Files
OpenJarvis/tests/agents/test_system_prompt_builder_integration.py
T
2026-03-17 00:13:49 +00:00

29 lines
854 B
Python

from __future__ import annotations
from pathlib import Path
from openjarvis.core.config import MemoryFilesConfig, SystemPromptConfig
def test_base_agent_uses_builder(tmp_path: Path):
soul = tmp_path / "SOUL.md"
soul.write_text("I am Jarvis.")
memory = tmp_path / "MEMORY.md"
memory.write_text("- User likes Python")
from openjarvis.prompt.builder import SystemPromptBuilder
builder = SystemPromptBuilder(
agent_template="You are a helpful assistant.",
memory_files_config=MemoryFilesConfig(
soul_path=str(soul),
memory_path=str(memory),
user_path=str(tmp_path / "USER.md"),
),
system_prompt_config=SystemPromptConfig(),
)
prompt = builder.build()
assert "Jarvis" in prompt
assert "Python" in prompt
assert "helpful assistant" in prompt