curo1305
1cdc532fff
- pytest suite for doc-service: 20+ tests covering category CRUD, document upload/get/delete/patch, ownership isolation, category assignment, AI processing (mock), and live PDF tests (auto-skipped when tests/pdfs/ is empty) - Minimal in-memory PDF builder in conftest so tests run without any fixture files; real PDFs can be dropped into tests/pdfs/ to activate live extraction tests - AI prompt updated to return suggested_categories (2–5 short names) - Frontend: SuggestionChip component in DocumentRow shows AI-suggested categories after processing; "Assign" links to an existing category, "Create & Assign" creates it first, ✕ dismisses locally - Default AI provider changed to LM Studio at http://host.docker.internal:1234/v1 (host.docker.internal resolves to the macOS host from inside Docker Desktop) - tests/pdfs/ directory tracked via .gitkeep; *.pdf excluded by .gitignore Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
120 lines
3.8 KiB
Python
120 lines
3.8 KiB
Python
"""
|
|
Per-service runtime config helpers.
|
|
|
|
Config files live on the shared `app_config` Docker volume at /config/.
|
|
Each service has its own JSON file, e.g. /config/doc_service_config.json.
|
|
|
|
Atomic write pattern: write to .tmp in same dir, then os.replace() so
|
|
doc-service never reads a partial file.
|
|
"""
|
|
import json
|
|
import os
|
|
from pathlib import Path
|
|
from typing import Any
|
|
|
|
from pydantic import BaseModel
|
|
|
|
_CONFIG_DIR = Path(os.environ.get("APP_CONFIG_DIR", "/config"))
|
|
|
|
# ── Config schemas ─────────────────────────────────────────────────────────────
|
|
|
|
|
|
class AnthropicConfig(BaseModel):
|
|
api_key: str = ""
|
|
model: str = "claude-haiku-4-5-20251001"
|
|
|
|
|
|
class OllamaConfig(BaseModel):
|
|
base_url: str = "http://192.168.1.x:11434/v1"
|
|
model: str = "llama3.2"
|
|
api_key: str = "ollama"
|
|
|
|
|
|
class LMStudioConfig(BaseModel):
|
|
# host.docker.internal resolves to the host from inside Docker (macOS/Windows).
|
|
# For local dev outside Docker, use http://localhost:1234/v1 instead.
|
|
base_url: str = "http://host.docker.internal:1234/v1"
|
|
model: str = "local-model"
|
|
api_key: str = "lm-studio"
|
|
|
|
|
|
class AIConfig(BaseModel):
|
|
provider: str = "lmstudio"
|
|
anthropic: AnthropicConfig = AnthropicConfig()
|
|
ollama: OllamaConfig = OllamaConfig()
|
|
lmstudio: LMStudioConfig = LMStudioConfig()
|
|
|
|
|
|
class DocumentsConfig(BaseModel):
|
|
max_pdf_bytes: int = 20 * 1024 * 1024
|
|
|
|
|
|
class DocServiceConfig(BaseModel):
|
|
ai: AIConfig = AIConfig()
|
|
documents: DocumentsConfig = DocumentsConfig()
|
|
|
|
|
|
# ── Masking ────────────────────────────────────────────────────────────────────
|
|
|
|
def _mask_key(key: str) -> str:
|
|
if not key or len(key) <= 8:
|
|
return "••••"
|
|
return key[:7] + "••••"
|
|
|
|
|
|
def _mask_config(data: dict) -> dict:
|
|
"""Return a copy of data with api_key fields masked."""
|
|
import copy
|
|
masked = copy.deepcopy(data)
|
|
ai = masked.get("ai", {})
|
|
for provider in ("anthropic", "ollama", "lmstudio"):
|
|
if provider in ai and "api_key" in ai[provider]:
|
|
ai[provider]["api_key"] = _mask_key(ai[provider]["api_key"])
|
|
return masked
|
|
|
|
|
|
# ── Load / Save ────────────────────────────────────────────────────────────────
|
|
|
|
def _config_path(service: str) -> Path:
|
|
return _CONFIG_DIR / f"{service}_config.json"
|
|
|
|
|
|
def load_service_config(service: str) -> dict:
|
|
path = _config_path(service)
|
|
if not path.exists():
|
|
# Return default config if file doesn't exist yet
|
|
if service == "doc_service":
|
|
return DocServiceConfig().model_dump()
|
|
return {}
|
|
with path.open() as f:
|
|
return json.load(f)
|
|
|
|
|
|
def save_service_config(service: str, data: dict) -> None:
|
|
path = _config_path(service)
|
|
path.parent.mkdir(parents=True, exist_ok=True)
|
|
tmp = path.with_suffix(".tmp")
|
|
tmp.write_text(json.dumps(data, indent=2))
|
|
os.replace(tmp, path)
|
|
|
|
|
|
def load_doc_service_config() -> DocServiceConfig:
|
|
raw = load_service_config("doc_service")
|
|
return DocServiceConfig.model_validate(raw)
|
|
|
|
|
|
def save_doc_service_config(config: DocServiceConfig) -> None:
|
|
save_service_config("doc_service", config.model_dump())
|
|
|
|
|
|
def load_doc_service_config_masked() -> dict:
|
|
raw = load_service_config("doc_service")
|
|
return _mask_config(raw)
|
|
|
|
|
|
def _merge_api_key(new_key: str, existing_key: str) -> str:
|
|
"""If new_key is empty or a masked value, keep the existing key."""
|
|
if not new_key or "••••" in new_key:
|
|
return existing_key
|
|
return new_key
|