c48788d4b6
Geofences could be drawn but not removed. VesselAPI Hormuz dots vanished
on restart and DVR skipped between the 5 daily polls. Sentinel-1 re-hit
STAC on every pan and often painted a neighbouring swath. Executive
briefs truncated; ticker stayed empty unless something was critical.
- Layer-panel list + polygon popup DELETE /api/geofences/{id}
- Persist VesselAPI polls to vessels (UTC-day purge, DVR as-of, boot hydrate)
- Cache Sentinel-1 by 2° cell; pick covering scene; clip Leaflet tiles
- Retry truncated LLM JSON; ticker falls back to medium/low; 3-min HUD poll
124 lines
3.9 KiB
Python
124 lines
3.9 KiB
Python
from unittest.mock import MagicMock
|
|
|
|
import httpx
|
|
from nous_client import chat
|
|
|
|
DEFAULT_UA = "osint-dashboard-news-summarizer"
|
|
BASE = "https://inference-api.nousresearch.com/v1"
|
|
|
|
|
|
def _ok_response(content="hello"):
|
|
resp = MagicMock()
|
|
resp.status_code = 200
|
|
resp.json.return_value = {"choices": [{"message": {"content": content}}]}
|
|
return resp
|
|
|
|
|
|
def _install_fake(monkeypatch, post_impl):
|
|
captured = {}
|
|
|
|
class FakeClient:
|
|
def __init__(self, timeout=None, **kwargs):
|
|
captured["timeout"] = timeout
|
|
|
|
def __enter__(self):
|
|
return self
|
|
|
|
def __exit__(self, *exc):
|
|
return False
|
|
|
|
def post(self, url, *, headers=None, json=None, **kwargs):
|
|
captured["url"] = url
|
|
captured["headers"] = headers
|
|
captured["json"] = json
|
|
return post_impl(url, headers, json)
|
|
|
|
monkeypatch.setattr(httpx, "Client", FakeClient)
|
|
return captured
|
|
|
|
|
|
def test_posts_chat_completions_with_auth_body_and_returns_content(monkeypatch):
|
|
captured = _install_fake(monkeypatch, lambda *a: _ok_response("the-content"))
|
|
out = chat(
|
|
"summarize this",
|
|
api_key="secret-key",
|
|
model="hermes-3",
|
|
base_url=BASE,
|
|
)
|
|
assert out == "the-content"
|
|
assert captured["url"] == f"{BASE}/chat/completions"
|
|
assert captured["headers"]["Authorization"] == "Bearer secret-key"
|
|
assert captured["headers"]["User-Agent"] == DEFAULT_UA
|
|
assert captured["json"]["model"] == "hermes-3"
|
|
assert captured["json"]["messages"] == [{"role": "user", "content": "summarize this"}]
|
|
assert captured["json"]["temperature"] == 0.2
|
|
assert captured["json"]["max_tokens"] == 4096
|
|
assert "response_format" not in captured["json"]
|
|
|
|
|
|
def test_user_agent_equals_osint_user_agent_env(monkeypatch):
|
|
monkeypatch.setenv("OSINT_USER_AGENT", "custom-ua/2.0")
|
|
captured = _install_fake(monkeypatch, lambda *a: _ok_response("ok"))
|
|
chat("p", api_key="k", model="m", base_url=BASE)
|
|
assert captured["headers"]["User-Agent"] == "custom-ua/2.0"
|
|
|
|
|
|
def test_json_mode_sets_response_format(monkeypatch):
|
|
captured = _install_fake(monkeypatch, lambda *a: _ok_response("{}"))
|
|
chat("p", api_key="k", model="m", base_url=BASE, json_mode=True)
|
|
assert captured["json"]["response_format"] == {"type": "json_object"}
|
|
assert captured["json"]["max_tokens"] >= 8192
|
|
roles = [m["role"] for m in captured["json"]["messages"]]
|
|
assert "system" in roles
|
|
assert "user" in roles
|
|
|
|
|
|
def test_retries_once_when_finish_reason_is_length(monkeypatch):
|
|
calls = {"n": 0}
|
|
|
|
def post_impl(*a):
|
|
calls["n"] += 1
|
|
if calls["n"] == 1:
|
|
resp = MagicMock()
|
|
resp.status_code = 200
|
|
resp.json.return_value = {
|
|
"choices": [{
|
|
"message": {"content": "{\"summary_en\": \"cut off"},
|
|
"finish_reason": "length",
|
|
}]
|
|
}
|
|
return resp
|
|
return _ok_response('{"summary_en": "complete brief."}')
|
|
|
|
_install_fake(monkeypatch, post_impl)
|
|
out = chat("p", api_key="k", model="m", base_url=BASE, json_mode=True)
|
|
assert calls["n"] == 2
|
|
assert "complete brief" in out
|
|
|
|
|
|
def test_401_returns_empty_string(monkeypatch):
|
|
def post_impl(*a):
|
|
resp = MagicMock()
|
|
resp.status_code = 401
|
|
return resp
|
|
|
|
_install_fake(monkeypatch, post_impl)
|
|
assert chat("p", api_key="bad", model="m", base_url=BASE) == ""
|
|
|
|
|
|
def test_5xx_returns_empty_string(monkeypatch):
|
|
def post_impl(*a):
|
|
resp = MagicMock()
|
|
resp.status_code = 503
|
|
return resp
|
|
|
|
_install_fake(monkeypatch, post_impl)
|
|
assert chat("p", api_key="k", model="m", base_url=BASE) == ""
|
|
|
|
|
|
def test_timeout_returns_empty_string(monkeypatch):
|
|
def post_impl(*a):
|
|
raise httpx.TimeoutException("timed out")
|
|
|
|
_install_fake(monkeypatch, post_impl)
|
|
assert chat("p", api_key="k", model="m", base_url=BASE) == ""
|