mirror of
https://github.com/open-jarvis/OpenJarvis.git
synced 2026-07-28 05:12:26 +00:00
Prathap PandGitHub
c2756964a7
* fix(channels): wire channel→agent handler and fix Telegram send pipeline * format code * add supported tests
126 lines
4.1 KiB
Python
126 lines
4.1 KiB
Python
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
from openjarvis.core.config import MemoryFilesConfig, SystemPromptConfig
|
|
|
|
|
|
@pytest.fixture
|
|
def memory_dir(tmp_path: Path) -> Path:
|
|
soul = tmp_path / "SOUL.md"
|
|
soul.write_text("You are a helpful research assistant.")
|
|
memory = tmp_path / "MEMORY.md"
|
|
memory.write_text("- User prefers concise answers\n- User is a data scientist")
|
|
user = tmp_path / "USER.md"
|
|
user.write_text("- Name: Alice\n- Role: ML Engineer")
|
|
return tmp_path
|
|
|
|
|
|
def test_build_frozen_prefix(memory_dir: Path):
|
|
from openjarvis.prompt.builder import SystemPromptBuilder
|
|
|
|
builder = SystemPromptBuilder(
|
|
agent_template="You are Jarvis.",
|
|
memory_files_config=MemoryFilesConfig(
|
|
soul_path=str(memory_dir / "SOUL.md"),
|
|
memory_path=str(memory_dir / "MEMORY.md"),
|
|
user_path=str(memory_dir / "USER.md"),
|
|
),
|
|
system_prompt_config=SystemPromptConfig(),
|
|
)
|
|
prompt = builder.build()
|
|
assert "Jarvis" in prompt
|
|
assert "helpful research assistant" in prompt
|
|
assert "concise answers" in prompt
|
|
assert "Alice" in prompt
|
|
|
|
|
|
def test_frozen_prefix_stability(memory_dir: Path):
|
|
from openjarvis.prompt.builder import SystemPromptBuilder
|
|
|
|
builder = SystemPromptBuilder(
|
|
agent_template="You are Jarvis.",
|
|
memory_files_config=MemoryFilesConfig(
|
|
soul_path=str(memory_dir / "SOUL.md"),
|
|
memory_path=str(memory_dir / "MEMORY.md"),
|
|
user_path=str(memory_dir / "USER.md"),
|
|
),
|
|
system_prompt_config=SystemPromptConfig(),
|
|
)
|
|
first = builder.build()
|
|
(memory_dir / "MEMORY.md").write_text("- CHANGED CONTENT")
|
|
second = builder.build()
|
|
assert first == second
|
|
|
|
|
|
def test_char_limit_truncation(memory_dir: Path):
|
|
from openjarvis.prompt.builder import SystemPromptBuilder
|
|
|
|
(memory_dir / "SOUL.md").write_text("x" * 10000)
|
|
builder = SystemPromptBuilder(
|
|
agent_template="You are Jarvis.",
|
|
memory_files_config=MemoryFilesConfig(
|
|
soul_path=str(memory_dir / "SOUL.md"),
|
|
memory_path=str(memory_dir / "MEMORY.md"),
|
|
user_path=str(memory_dir / "USER.md"),
|
|
),
|
|
system_prompt_config=SystemPromptConfig(soul_max_chars=100),
|
|
)
|
|
prompt = builder.build()
|
|
assert prompt.count("x") <= 100
|
|
assert "truncated" in prompt.lower()
|
|
|
|
|
|
def test_skill_index_in_prompt(memory_dir: Path):
|
|
from openjarvis.prompt.builder import SystemPromptBuilder
|
|
|
|
skills = [("api_health_check", "Check API health across all endpoints")]
|
|
builder = SystemPromptBuilder(
|
|
agent_template="You are Jarvis.",
|
|
memory_files_config=MemoryFilesConfig(
|
|
soul_path=str(memory_dir / "SOUL.md"),
|
|
memory_path=str(memory_dir / "MEMORY.md"),
|
|
user_path=str(memory_dir / "USER.md"),
|
|
),
|
|
system_prompt_config=SystemPromptConfig(),
|
|
skill_index=skills,
|
|
)
|
|
prompt = builder.build()
|
|
assert "api_health_check" in prompt
|
|
assert "Check API health" in prompt
|
|
|
|
|
|
def test_dynamic_section_appended(memory_dir: Path):
|
|
from openjarvis.prompt.builder import SystemPromptBuilder
|
|
|
|
builder = SystemPromptBuilder(
|
|
agent_template="You are Jarvis.",
|
|
memory_files_config=MemoryFilesConfig(
|
|
soul_path=str(memory_dir / "SOUL.md"),
|
|
memory_path=str(memory_dir / "MEMORY.md"),
|
|
user_path=str(memory_dir / "USER.md"),
|
|
),
|
|
system_prompt_config=SystemPromptConfig(),
|
|
session_context="Platform: CLI | Session: abc123",
|
|
)
|
|
prompt = builder.build()
|
|
assert "Platform: CLI" in prompt
|
|
|
|
|
|
def test_missing_files_handled(tmp_path: Path):
|
|
from openjarvis.prompt.builder import SystemPromptBuilder
|
|
|
|
builder = SystemPromptBuilder(
|
|
agent_template="You are Jarvis.",
|
|
memory_files_config=MemoryFilesConfig(
|
|
soul_path=str(tmp_path / "missing_soul.md"),
|
|
memory_path=str(tmp_path / "missing_memory.md"),
|
|
user_path=str(tmp_path / "missing_user.md"),
|
|
),
|
|
system_prompt_config=SystemPromptConfig(),
|
|
)
|
|
prompt = builder.build()
|
|
assert "Jarvis" in prompt
|