Added the optional Ollama profile advisor
This commit is contained in:
@@ -95,6 +95,20 @@ class MonitorTests(unittest.TestCase):
|
||||
self.assertEqual(status["llm_assessment"]["status"], "ok")
|
||||
self.assertEqual(status["llm_assessment"]["text"], "looks noisy")
|
||||
|
||||
def test_profile_advisor_status_records_error(self):
|
||||
with tempfile.TemporaryDirectory() as tmp:
|
||||
log_path = Path(tmp) / "events.log"
|
||||
log_path.write_text("fgai_stream_id=windows fgai_stream=Windows username=alice eventid=4625 action=failure\n", encoding="utf-8")
|
||||
config_path = Path(tmp) / "config.json"
|
||||
config_path.write_text(json.dumps({"profile_advisor_enabled": True, "profile_advisor_model": "qwen3:8b"}), encoding="utf-8")
|
||||
|
||||
with patch("fgai.monitor.ollama_profile_advice", side_effect=TimeoutError("timeout")):
|
||||
status = build_status(str(log_path), config_path=str(config_path), incident_path=str(Path(tmp) / "incidents.json"))
|
||||
|
||||
advisor = status["capabilities"]["profile_advisor"]
|
||||
self.assertEqual(advisor["status"], "error")
|
||||
self.assertIn("timeout", advisor["error"])
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
|
||||
Reference in New Issue
Block a user