add ollama bagcround run and cache
This commit is contained in:
@@ -1,8 +1,9 @@
|
||||
import tempfile
|
||||
import unittest
|
||||
from pathlib import Path
|
||||
from unittest.mock import patch
|
||||
|
||||
from fgai.monitor import build_status, write_status
|
||||
from fgai.monitor import add_llm_assessment, build_status, write_status
|
||||
|
||||
|
||||
class MonitorTests(unittest.TestCase):
|
||||
@@ -33,6 +34,24 @@ class MonitorTests(unittest.TestCase):
|
||||
|
||||
self.assertTrue(output.exists())
|
||||
|
||||
def test_add_llm_assessment_records_error_without_ollama(self):
|
||||
status = {"summary": {}, "anomalies": []}
|
||||
|
||||
with patch("fgai.monitor.ollama_dashboard_assessment", side_effect=TimeoutError("timeout")):
|
||||
add_llm_assessment(status, previous="old text")
|
||||
|
||||
self.assertEqual(status["llm_assessment"]["status"], "error")
|
||||
self.assertEqual(status["llm_assessment"]["text"], "old text")
|
||||
|
||||
def test_add_llm_assessment_records_text(self):
|
||||
status = {"summary": {}, "anomalies": []}
|
||||
|
||||
with patch("fgai.monitor.ollama_dashboard_assessment", return_value="looks noisy"):
|
||||
add_llm_assessment(status)
|
||||
|
||||
self.assertEqual(status["llm_assessment"]["status"], "ok")
|
||||
self.assertEqual(status["llm_assessment"]["text"], "looks noisy")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
|
||||
Reference in New Issue
Block a user