File size: 930 Bytes
ebe934f | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 | from pathlib import Path
KNOWLEDGE_ROOT = Path(__file__).parent.parent / "knowledge"
EMBEDDER_MODEL = "all-MiniLM-L6-v2"
DOMAIN_CLIENTS: dict[str, list[str]] = {
"retail": ["novamart", "shelfwise"],
"pharma": ["clinixone", "pharmalink"],
}
CLIENT_DOMAIN: dict[str, str] = {
client: domain
for domain, clients in DOMAIN_CLIENTS.items()
for client in clients
}
DISPLAY_NAMES: dict[str, str] = {
"novamart": "NovaMart",
"shelfwise": "ShelfWise",
"clinixone": "ClinixOne",
"pharmalink": "PharmaLink",
}
def term_catalog_path(domain: str) -> Path:
return KNOWLEDGE_ROOT / domain / "term-catalog.yaml"
def features_path(domain: str) -> Path:
return KNOWLEDGE_ROOT / domain / "features.yaml"
def domain_for(client: str) -> str:
if client not in CLIENT_DOMAIN:
raise ValueError(f"Unknown client: {client!r}. Valid: {list(CLIENT_DOMAIN)}")
return CLIENT_DOMAIN[client]
|