55 lines
2.0 KiB
Python
55 lines
2.0 KiB
Python
|
"""GLiNER client: local URL only, fail-soft, no request text in logs."""
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import os
|
||
|
|
import sys
|
||
|
|
from pathlib import Path
|
||
|
|
from unittest.mock import patch
|
||
|
|
|
||
|
|
import httpx
|
||
|
|
|
||
|
|
ROOT = Path(__file__).resolve().parents[1]
|
||
|
|
sys.path.insert(0, str(ROOT))
|
||
|
|
|
||
|
|
os.environ.setdefault("KANSHO_PROVIDER_KEY", "")
|
||
|
|
os.environ.setdefault("KANSHO_DETECT_PROVIDER_KEY", "")
|
||
|
|
|
||
|
|
from gliner_client import DEFAULT_URL, extract_entities, health, normalize_base_url
|
||
|
|
|
||
|
|
|
||
|
|
def expect(ok: bool, message: str) -> None:
|
||
|
|
if not ok:
|
||
|
|
raise SystemExit(f"FAIL: {message}")
|
||
|
|
print(f"OK {message}")
|
||
|
|
|
||
|
|
|
||
|
|
class _Resp:
|
||
|
|
def __init__(self, status_code: int, payload):
|
||
|
|
self.status_code = status_code
|
||
|
|
self._payload = payload
|
||
|
|
|
||
|
|
def json(self):
|
||
|
|
return self._payload
|
||
|
|
|
||
|
|
|
||
|
|
def main() -> None:
|
||
|
|
expect(normalize_base_url("http://192.168.2.144:8090/") == DEFAULT_URL, "default GLiNER URL is normalized")
|
||
|
|
expect(extract_entities("Anna", url="https://openrouter.ai/api/v1") is None, "OpenRouter URL is rejected")
|
||
|
|
expect(health("https://openrouter.ai").get("reason") == "not_local", "health rejects non-local URL")
|
||
|
|
|
||
|
|
with patch("gliner_client.httpx.post", return_value=_Resp(200, {"spans": [{"text": "Anna", "start": 0, "end": 4, "label": "person", "score": 0.9}]})):
|
||
|
|
spans = extract_entities("Anna kam.", url="http://192.168.2.144:8090")
|
||
|
|
expect(spans == [{"text": "Anna", "start": 0, "end": 4, "label": "person", "score": 0.9}], "extract returns spans")
|
||
|
|
|
||
|
|
with patch("gliner_client.httpx.post", side_effect=httpx.ConnectError("down")):
|
||
|
|
expect(extract_entities("Anna kam.", url="http://192.168.2.144:8090") is None, "extract is fail-soft on errors")
|
||
|
|
|
||
|
|
with patch("gliner_client.httpx.get", return_value=_Resp(200, {"status": "ok", "model": "urchade/gliner_multi-v2.1", "offline": True})):
|
||
|
|
ping = health("http://192.168.2.144:8090")
|
||
|
|
expect(ping.get("ok") is True and ping.get("offline") is True, "health reports ready offline model")
|
||
|
|
print("gliner client: OK")
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
main()
|