server.py

Go to the documentation of this file.

# Copyright (c) 2025-present Aquin Labs Private Limited. All Rights Reserved.
# This file is part of the Aquin Engine. Unauthorized copying, modification,
# distribution, or use of this file, via any medium, is strictly prohibited.
# Proprietary and confidential. See LICENSE for terms.
 
"""
Route index ingested from inspection-backend/server.py.
NOT a runnable server — kept as a grep reference for which HTTP route calls which function.
Never import FastAPI, never run uvicorn from the CLI package.
 
Route → function mapping:
  GET  /health                          → (inline) returns status dict
  GET  /sae-layers                      → model_loader.get_available_sae_layers
  GET  /model/load                      → causal_trace.load_model + feature_analysis.load_sae  (streaming)
  GET  /workspace/status                → tab_queue.TabQueue.status()
 
  POST /chat/inspect                    → causal_trace.run_chat
  POST /chat/inspect/stream             → causal_trace.stream_chat  (SSE)
  POST /trace                           → causal_trace.run_trace
  POST /logit-lens                      → causal_trace.run_logit_lens
  POST /attribution                     → causal_trace.run_prompt_attribution
  POST /tokenize                        → feature_analysis.load_sae + causal_trace.load_model  (SSE, hook-based)
  POST /tokenize-multi                     → feature_analysis.load_sae + causal_trace.load_model  (SSE, combined hook)
 
  POST /features                        → feature_analysis.run_feature_analysis_unlabeled
  POST /label-feature                   → feature_analysis.get_causal_label
  POST /label-features/stream           → feature_analysis.run_feature_analysis_unlabeled + get_causal_label  (SSE)
  POST /interp-score                    → interp_score.run_interp_score
  POST /feature-logits                  → (inline) sae.W_dec @ model.unembed.W_U
  POST /feature-neighbors               → (inline) cosine similarity over sae.W_dec
 
  POST /model/weight-trojans            → weight_trojans.run_weight_trojan_analysis
  POST /evals/consistency               → evals.consistency_eval
  POST /evals/suppression               → evals.suppression_eval
  POST /evals/boundary                  → evals.boundary_eval
 
  POST /model/activation-stability      → (inline) _run_activation_stability
  POST /model/attention-routing         → (inline) _run_attention_routing
  POST /model/weight-rank               → (inline) _run_weight_rank
  POST /model/perturbation-sensitivity  → (inline) _run_perturbation_sensitivity
  POST /model/umap                      → (inline) UMAP over sae.W_dec
 
  POST /train/simulate                  → train_simulate.run_training_simulation  (SSE)
  POST /train/sae                       → train_sae  (SSE via subprocess or direct call)
"""