fix advisor and drop down refresh

This commit is contained in:
larssand
2026-07-01 08:56:08 +02:00
parent fcfbe04a58
commit a3cfab1f5a
5 changed files with 66 additions and 5 deletions

23
tests/test_llm.py Normal file
View File

@@ -0,0 +1,23 @@
import unittest
from fgai.llm import _json_object_from_text
class LlmTests(unittest.TestCase):
def test_json_object_from_text_accepts_markdown_wrapped_json(self):
payload = _json_object_from_text(
'```json\n{"profiles":[{"stream_id":"windows","entity_fields":["username"]}]}\n```'
)
self.assertEqual(payload["profiles"][0]["stream_id"], "windows")
def test_json_object_from_text_extracts_object_from_extra_text(self):
payload = _json_object_from_text(
'Here is the profile:\n{"profiles":[{"stream_id":"firewall"}]}\nDone.'
)
self.assertEqual(payload["profiles"][0]["stream_id"], "firewall")
if __name__ == "__main__":
unittest.main()