193829b7ff
- Add COMPLETIONS_API_KEY to config.py (env var + auto-generated fallback) - Fix perplexity auto-search: upstream sends logprobs=true, parse_llama_stream_chunk extracts per-token logprobs, all_logprobs populated during streaming - Fix all /api/models endpoints to target LLAMA_SERVER_BASE (port 8081) not OLLAMA_BASE - Fix RAG embedding endpoint URL from port 11434 (Ollama) to 8081 (llama-server) - Correct misleading error messages: 'inference server' not 'Ollama' - Remove raw_results leak from SSE event stream in /api/search - Fix weather query extractor: pattern-match instead of unconditional suffix append - Escape FTS5 operator keywords (AND/OR/NOT/NEAR) in memory search - Move auth.py BODY_LIMIT_DEFAULT_BYTES imports to module level - Change RAG injection log level from warning to info - Fix all 8 test files after modular refactor (rewire imports from correct modules) - Update AGENTS.md and README.md to reflect v1.8.0 changes
78 lines
2.4 KiB
Python
78 lines
2.4 KiB
Python
import os
|
|
from pathlib import Path
|
|
|
|
import httpx
|
|
from fastapi.testclient import TestClient
|
|
|
|
import app
|
|
import config
|
|
import db
|
|
import routers.memories
|
|
from security import SESSIONS, PIN_ATTEMPTS, RATE_EVENTS
|
|
|
|
|
|
def make_client(tmp_path: Path) -> TestClient:
|
|
os.environ["JARVISCHAT_ADMIN_PIN"] = "1234"
|
|
db.DB_PATH = tmp_path / "jarvischat-errors.db"
|
|
SESSIONS.clear()
|
|
PIN_ATTEMPTS.clear()
|
|
RATE_EVENTS.clear()
|
|
db.init_db()
|
|
return TestClient(app.app, raise_server_exceptions=False)
|
|
|
|
|
|
def test_unhandled_api_exception_returns_friendly_error_with_incident_key(
|
|
tmp_path: Path, monkeypatch
|
|
):
|
|
with make_client(tmp_path) as client:
|
|
sid = client.post("/api/auth/guest", headers={"Origin": "http://testserver"}).json()[
|
|
"session_id"
|
|
]
|
|
headers = {"X-Session-ID": sid}
|
|
|
|
def boom(_topic=None):
|
|
raise RuntimeError("super secret db internals")
|
|
|
|
monkeypatch.setattr(routers.memories, "get_all_memories", boom)
|
|
|
|
resp = client.get("/api/memories", headers=headers)
|
|
assert resp.status_code == 500
|
|
payload = resp.json()
|
|
assert payload.get("error_key", "").startswith("INC-")
|
|
assert "support lookup" in payload.get("detail", "").lower()
|
|
assert "super secret db internals" not in resp.text
|
|
|
|
|
|
def test_chat_stream_error_hides_internal_exception_and_emits_incident_key(
|
|
tmp_path: Path, monkeypatch
|
|
):
|
|
with make_client(tmp_path) as client:
|
|
sid = client.post("/api/auth/guest", headers={"Origin": "http://testserver"}).json()[
|
|
"session_id"
|
|
]
|
|
headers = {"X-Session-ID": sid, "Origin": "http://testserver"}
|
|
|
|
class BrokenStreamContext:
|
|
async def __aenter__(self):
|
|
raise RuntimeError("ultra secret model transport failure")
|
|
|
|
async def __aexit__(self, exc_type, exc, tb):
|
|
return False
|
|
|
|
def broken_stream(*args, **kwargs):
|
|
return BrokenStreamContext()
|
|
|
|
monkeypatch.setattr(httpx.AsyncClient, "stream", broken_stream)
|
|
|
|
resp = client.post(
|
|
"/api/chat",
|
|
json={"message": "hello", "model": config.DEFAULT_MODEL},
|
|
headers=headers,
|
|
)
|
|
|
|
assert resp.status_code == 200
|
|
body = resp.text
|
|
assert "ultra secret model transport failure" not in body
|
|
assert "error_key" in body
|
|
assert "support lookup" in body.lower()
|