-
Notifications
You must be signed in to change notification settings - Fork 0
Developer Architecture
This document explains how cAIc is structured, the external services it integrates with, and the key architectural decisions.
cAIc is a single-process FastAPI service with a Jinja2 frontend and SQLite persistence. It connects to an external llama-server for inference and optionally to SearXNG (web search), Qdrant (vector search), and RabbitMQ (AMQP cluster messaging).
| File | Role |
|---|---|
app.py |
FastAPI app, middleware, router registration, lifespan |
config.py |
Constants, env vars, rate/payload limits, built-in skills registry |
db.py |
SQLite schema, connection factory, settings helpers, upload_context CRUD |
auth.py |
PIN-based guest/admin sessions, auth routes |
security.py |
Rate limiting, origin checks, IP allowlist, audit/incident logging |
memory.py |
FTS5 memory CRUD, remember/forget command parsing |
search.py |
SearXNG integration, perplexity scoring, refusal detection |
rag.py |
Qdrant vector search, system prompt assembly, chunk_text() helper |
eviction.py |
Score-based RAG eviction engine |
gpu.py |
AMD GPU stats via rocm-smi |
hardware.py |
Hardware self-assessment — CPU, RAM, VRAM, service health probes |
amqp.py |
aio-pika connection manager for RabbitMQ |
cluster.py |
Cluster node registry, event log, coordinator election, ping/pong |
triage.py |
Phi-4-mini query classification + select_node() for cluster routing |
routers/ |
One module per endpoint group |
| Service | Required | Port | Purpose |
|---|---|---|---|
| llama-server | Yes | 8081 | LLM inference (OpenAI-compat) |
| SearXNG | No | 8888 | Privacy-respecting web search |
| Qdrant | No | 6333 | Vector database for RAG |
| Ollama | No | 11434 | Embeddings for RAG |
| RabbitMQ | No | 5672 | AMQP broker for cluster messaging |
| rocm-smi | No | — | AMD GPU stats (host-level) |
| Variable | Default | Service |
|---|---|---|
LLAMA_SERVER_BASE |
http://192.168.50.108:8081 |
llama-server on coordinator |
OLLAMA_BASE |
http://localhost:11434 |
Legacy — embeddings |
SEARXNG_BASE |
http://localhost:8888 |
SearXNG |
QDRANT_URL |
http://192.168.50.108:6333 |
Qdrant on coordinator |
CAIC_AMQP_URL |
amqp://caic:password@localhost:5672/caic |
RabbitMQ |
- Validate session, role, origin, rate, and payload limits in middleware
- Intercept "remember that..." / "forget about..." commands → process_remember_command()
- Persist user message and conversation metadata
- Build system prompt: profile + FTS5 memory + Qdrant RAG results + preset + active skills
- Stream from llama-server with
logprobs: truefor perplexity scoring - If perplexity > 15.0 OR refusal patterns match → re-query with SearXNG results
- Persist final assistant message and emit terminal SSE event
- Persist search-as-message into conversation
- Emit
searchingSSE event - Pull web results from SearXNG
- Summarize via llama-server SSE stream
- Persist summary and emit
doneevent
- Bearer token auth (same key as completions API)
- Chunk text via shared
chunk_text()helper (512-token chunks, 128-token overlap) - Embed via Ollama
/api/embeddings - Upsert to Qdrant collection
caic_rag - Trigger
maybe_evict()if collection exceeds high-water mark
- Admin required, multipart file upload
- Validate MIME type + size against config limits
- PDF text extraction via pypdf; plain text for all other types
- Three modes:
context(SQLite with 1hr expiry),ingest(RAG/Qdrant),both - Trigger
maybe_evict()if ingest mode
Key tables:
-
conversations— headers, timestamps, attachment_count -
messages— ordered chat history per conversation -
profile— singleton row for injected profile prompt -
settings— runtime toggles and selected defaults -
system_presets— named reusable system prompts -
skills— per-skill enabled state and timestamp -
memories(FTS5 virtual table) — full-text searchable user memory facts -
upload_context— auto-expiring document storage for context injection
Design notes:
- Startup is idempotent: tables created if missing, defaults seeded only when absent
- No connection pool: each request opens and closes a short-lived SQLite connection
-
init_db()called in FastAPI lifespan
- Guest session by default (POST /api/auth/guest)
- Admin unlock via 4-digit PIN (POST /api/auth/login)
- Admin required for PUT/DELETE/PATCH + all POST except allowlist
- /api/ingest is exempt from session auth — self-authenticates via Bearer token
- Session heartbeat/timeout (90s default) and explicit logout
- Admin PIN hashed with PBKDF2-HMAC-SHA256 + salt
- Failed PIN attempts tracked per client IP (max 5, 300s lockout)
- Default PIN allowed only if CAIC_ALLOW_DEFAULT_PIN=true
- Origin checks on all /api/ requests
- Rate limiting per endpoint category and identity (IP/session)
- Payload size limits per route class (64KB default, 128KB chat, 20MB upload)
- Settings key allowlist (5 keys)
- IP allowlist/CIDR gate with trusted proxy forwarding mode
- Search result URLs sanitized to http/https only
- Client-safe error envelopes with incident key correlation
- Full stack traces logged server-side only
- Structured audit events for auth actions, admin ops, guardrail denials
- Incident logs with event type, key, path/method, and runtime metadata
- Qdrant collection
caic_ragon coordinator:6333 - Embeddings via Ollama on worker:11434 (
/api/embeddings) - Shared
chunk_text(text, chunk_size=512, overlap=128)helper in rag.py - Upload and ingest endpoints share the same chunk+embed+upsert pipeline
When RAG_MAX_VECTORS is exceeded, eviction fires with hysteresis:
- High-water mark: 80% of max → trigger eviction
- Low-water mark: 20% of max → stop eviction
- Batch size: 1000 vectors per cycle
- Score formula:
score = (access_weight * retrieval_count) + (age_weight * hours_since_ingested) - Lower score evicted first (least useful)
- Excluded sources:
upload,profile(pinned) - Grace period: 1 hour before any vector is eligible
- Thread-safe via
asyncio.Lock
GET /api/rag/stats (admin required) returns:
- vector_count, max_vectors, high_water_pct, low_water_pct, percent_full
- pinned_sources list, grace_hours
- at_risk_count, pinned_count, avg_retrieval_count
- eviction_counts_last_{1,5,30}m
POST /api/rag/flush (admin required) — deletes all non-pinned vectors.
cAIc uses a broker-mediated cluster design:
- A single RabbitMQ broker acts as the central nervous system
- Coordinator nodes run the FastAPI app, host the HTTP API/UI, and publish commands to the broker
- Worker nodes connect as AMQP clients only — they consume commands and publish status events
- Communication is asynchronous and persistent via TCP connection on startup
What it is NOT:
- Not a service mesh — workers do not run identical software stacks
- Not autonomous failover — if the coordinator dies, a replacement must be manually promoted
- Not a peer-to-peer cluster — all orchestration flows through the coordinator
| Aspect | Coordinator | Worker |
|---|---|---|
| Role | HTTP API/UI, orchestrates inference, owns cluster state | Runs inference models |
| Python | Required — runs FastAPI app | Required — runs node agent |
| RabbitMQ server | Required — hosts the broker | Not required — AMQP client only |
| FastAPI / uvicorn | Required | Not needed |
| SQLite | Required — owns caic.db | Not needed |
| Qdrant | Optional — vector DB for RAG | Not needed |
| SearXNG | Optional — web search | Not needed |
| llama-server | Optional — can share its own GPU | Required — this is why the worker exists |
| Ollama | Optional — embeddings | Not needed |
| Exchange | Type | Purpose |
|---|---|---|
jc.admin |
topic | Lifecycle commands: register, deregister, ping, pong, model swap |
jc.system |
topic | Events: model_ready, model_failed, heartbeat, coord_query |
All exchanges, queues, and bindings are declared by amqp.py at startup.
All streaming endpoints yield data: {json}\n\n:
-
{token, conversation_id}— streaming token -
{searching: true}— web search triggered -
{search_results: N}— N results found -
{done: true, perplexity, tokens_per_sec, searched?}— terminal -
{error: "...", error_key: "..."}— error with incident key
- pytest with
tmp_path+ monkeypatched httpx.AsyncClient - No live external services required
- Test factories reset globals per test
For substantive changes:
- Implement code change
- Add/adjust tests proving behavior and guardrail intent
- Update this wiki and README in the same change set
- Validate with full test run before commit
On startup, assess_hardware() probes:
- RAM total/available (psutil)
- VRAM total/free (rocm-smi, best-effort)
- llama-server reachability + model list
- Qdrant reachability + collection list
- SearXNG reachability
Writes hardware_state.json to working directory (configurable via CAIC_HW_STATE_PATH).