feat(cli): add response guardrails and grounded followup re-anchoring
This commit is contained in:
@@ -4,7 +4,7 @@ from unittest.mock import MagicMock, patch
|
||||
|
||||
from tai.ai_client import DEFAULT_AI_HOST, DEFAULT_MODEL, AIClient, AIConfig
|
||||
from tai.collectors import CollectedItem, CollectionReport
|
||||
from tai.prompt_builder import build_system_prompt, build_user_message
|
||||
from tai.prompt_builder import build_followup_message, build_system_prompt, build_user_message
|
||||
from tai.ssh_client import SSHCommandResult
|
||||
|
||||
# ---------------------------------------------------------------------------
|
||||
@@ -218,3 +218,16 @@ def test_build_user_message_handles_no_output() -> None:
|
||||
report = _make_report([("empty", "cat /nonexistent", 1, "", "")])
|
||||
msg = build_user_message("test", report)
|
||||
assert "no output" in msg
|
||||
|
||||
|
||||
def test_build_followup_message_includes_question_context() -> None:
|
||||
report = _make_report([("kernel", "uname -a", 0, "Linux web01", "")])
|
||||
msg = build_followup_message(
|
||||
"nginx is failing",
|
||||
report,
|
||||
"what should I check next?",
|
||||
["is nginx running?", "show me logs"],
|
||||
)
|
||||
assert "Current follow-up question" in msg
|
||||
assert "what should I check next?" in msg
|
||||
assert "Recent user follow-up questions" in msg
|
||||
|
||||
Reference in New Issue
Block a user