Lars
fa1c11c33e
Senses grow from confirmations instead of a word list, so a later local pipeline can decide homonyms on a short passage. Default stays semantic. Local detect waits longer, and the reverse-proxy timeout is documented so Dev does not 504 first. Co-authored-by: Cursor <cursoragent@cursor.com>
71 lines
2.3 KiB
Python
71 lines
2.3 KiB
Python
"""Local trusted-zone URL tests. No database."""
|
|
from __future__ import annotations
|
|
|
|
import os
|
|
import sys
|
|
import tempfile
|
|
from pathlib import Path
|
|
|
|
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", "")
|
|
Path(tempfile.gettempdir()).mkdir(parents=True, exist_ok=True)
|
|
|
|
from entity_detect import (
|
|
DETECT_MAX_TOKENS,
|
|
DETECT_MAX_TOKENS_LOCAL,
|
|
DETECT_TIMEOUT,
|
|
DETECT_TIMEOUT_LOCAL,
|
|
detect_chat_limits,
|
|
)
|
|
from providers import ProviderConfig, is_local_url
|
|
|
|
|
|
def expect(ok: bool, message: str) -> None:
|
|
if not ok:
|
|
raise SystemExit(f"FAIL: {message}")
|
|
print(f"OK {message}")
|
|
|
|
|
|
def main() -> None:
|
|
expect(is_local_url("http://127.0.0.1:11434/v1/chat/completions"), "loopback is local")
|
|
expect(is_local_url("http://localhost:11434/v1/chat/completions"), "localhost is local")
|
|
expect(is_local_url("http://192.168.2.144:11434/v1/chat/completions"), "RFC1918 Ollama host is local")
|
|
expect(is_local_url("http://10.0.0.8:11434/v1/chat/completions"), "10/8 is local")
|
|
expect(is_local_url("http://ollama:11434/v1/chat/completions"), "compose hostname ollama is local")
|
|
expect(not is_local_url("https://openrouter.ai/api/v1/chat/completions"), "OpenRouter is not local")
|
|
expect(not is_local_url("https://8.8.8.8/v1/chat/completions"), "public IP is not local")
|
|
remote = ProviderConfig(
|
|
role="detect",
|
|
name="openrouter",
|
|
mode="http",
|
|
url="https://openrouter.ai/api/v1/chat/completions",
|
|
model="openai/gpt-4.1-nano",
|
|
key="x",
|
|
local=False,
|
|
zdr=True,
|
|
no_train=True,
|
|
)
|
|
local = ProviderConfig(
|
|
role="detect",
|
|
name="ollama",
|
|
mode="http",
|
|
url="http://192.168.2.144:11434/v1/chat/completions",
|
|
model="phi3:mini",
|
|
key="",
|
|
local=True,
|
|
zdr=True,
|
|
no_train=True,
|
|
)
|
|
expect(detect_chat_limits(remote) == (DETECT_TIMEOUT, DETECT_MAX_TOKENS), "remote detect keeps 90s/1024")
|
|
expect(
|
|
detect_chat_limits(local) == (DETECT_TIMEOUT_LOCAL, DETECT_MAX_TOKENS_LOCAL),
|
|
"local detect waits 300s with a 256-token cap",
|
|
)
|
|
print("local url tests passed.")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|