Skip to main content

superfluid_daemon/
serve.rs

1//! `superfluid serve`: the server, as a library function so that another
2//! program can run it.
3
4use crate as superfluid_daemon;
5
6use std::os::unix::net::UnixListener;
7use std::path::{Path, PathBuf};
8use std::sync::Arc;
9
10use superfluid_daemon::runtime_pick::{RuntimeId, RuntimePick};
11use superfluid_daemon::{api, Daemon, EngineHost, SessionStore};
12
13fn retired(flag: &str, why: &str) {
14    eprintln!("superfluid: {flag} is retired — {why}");
15}
16
17fn sampling_f32(flag: &str, raw: &str, lo: f32, hi: f32) -> f32 {
18    let v: f32 = raw.parse().unwrap_or_else(|_| {
19        eprintln!("superfluid: {flag} expects a number, got {raw:?}");
20        std::process::exit(2);
21    });
22    if !v.is_finite() || v < lo || v > hi {
23        eprintln!("superfluid: {flag} must be a finite number in [{lo}, {hi}], got {v}");
24        std::process::exit(2);
25    }
26    v
27}
28
29fn usage_error(msg: &str, command: &str) -> ! {
30    eprintln!("superfluid: {msg}");
31    eprintln!("Run 'superfluid {command} --help' for its options.");
32    std::process::exit(2);
33}
34
35fn needs_value(flag: &str) -> ! {
36    usage_error(&format!("{flag} needs a value"), "serve")
37}
38
39fn num<T: std::str::FromStr>(flag: &str, raw: String) -> T {
40    raw.parse().unwrap_or_else(|_| usage_error(&format!("{flag} expects a number, got {raw:?}"), "serve"))
41}
42
43fn unknown_serve_flag(flag: &str) -> ! {
44    let hint = superfluid_daemon::cli::suggest(flag, superfluid_daemon::cli::serve_flag_names())
45        .map(|s| format!(" (did you mean {s}?)"))
46        .unwrap_or_default();
47    usage_error(&format!("serve has no option {flag}{hint}"), "serve")
48}
49
50fn human_bytes(n: u64) -> String {
51    const GIB: f64 = (1u64 << 30) as f64;
52    const MIB: f64 = (1u64 << 20) as f64;
53    if n as f64 >= GIB {
54        format!("{:.1} GiB", n as f64 / GIB)
55    } else {
56        format!("{:.0} MiB", n as f64 / MIB)
57    }
58}
59
60pub(crate) fn settle_park_dir(dir: &std::path::Path, parking: bool, budget_gb: u64) {
61    if parking {
62        let budget = budget_gb.saturating_mul(1 << 30);
63        if budget == 0 {
64            return;
65        }
66        let freed = superfluid_daemon::park::sweep(dir, budget);
67        if freed > 0 {
68            eprintln!(
69                "superfluid: park retention sweep freed {} in {} at startup (budget {budget_gb} GiB)",
70                human_bytes(freed),
71                dir.display()
72            );
73        }
74        return;
75    }
76    let stale: u64 = std::fs::read_dir(dir)
77        .map(|rd| {
78            rd.filter_map(|e| e.ok())
79                .filter(|e| {
80                    let n = e.file_name();
81                    let n = n.to_string_lossy();
82                    n.ends_with(".park") || n.ends_with(".park.tmp")
83                })
84                .filter_map(|e| e.metadata().ok().map(|m| m.len()))
85                .sum()
86        })
87        .unwrap_or(0);
88    if stale > 0 {
89        eprintln!(
90            "superfluid: {} of park artifacts in {} are from an earlier run with --park; \
91             nothing reads or reclaims them while parking is off. Delete them, or pass \
92             --park to put them back under --park-budget-gb.",
93            human_bytes(stale),
94            dir.display()
95        );
96    }
97}
98
99pub const SUN_PATH_LEN: usize = if cfg!(target_os = "linux") { 108 } else { 104 };
100
101/// The session log under `sessions`, or the reason it cannot be served from, and exit.
102fn open_session_store(sessions: &std::path::Path) -> SessionStore {
103    let wal_path = sessions.join("wal.log");
104    match SessionStore::open(&wal_path) {
105        Ok(store) => store,
106        Err(e @ (superfluid_daemon::DaemonError::WalCorrupt | superfluid_daemon::DaemonError::Codec(_))) => {
107            eprintln!(
108                "superfluid: the session log {} is damaged ({e}). A record with more of the log \
109                 after it fails its checksum, or a whole record does not decode, so this was not a \
110                 crash mid-write (a torn last record is dropped on its own); this server won't guess \
111                 which history to keep. Move the file aside to start fresh, or run with \
112                 --sessions-cache if sessions don't need to survive a restart.",
113                wal_path.display()
114            );
115            std::process::exit(1);
116        }
117        Err(e) => {
118            eprintln!("superfluid: could not open the session log {}: {e}", wal_path.display());
119            std::process::exit(1);
120        }
121    }
122}
123
124fn listen_or_exit(flag: &str, addr: &str) -> std::net::TcpListener {
125    std::net::TcpListener::bind(addr).unwrap_or_else(|e| {
126        let hint = match e.kind() {
127            std::io::ErrorKind::AddrInUse => " (another process listens there: stop it or pick another port, e.g. --port 8454)",
128            std::io::ErrorKind::AddrNotAvailable => " (no interface on this machine has that address)",
129            std::io::ErrorKind::PermissionDenied => " (a port under 1024 needs privileges)",
130            _ => "",
131        };
132        eprintln!("superfluid: cannot listen on {flag} {addr}: {e}{hint}");
133        std::process::exit(1);
134    })
135}
136
137/// What a program running [`run`] in place of `superfluid serve` changes.
138/// [`Embedding::default`] is `superfluid serve` itself.
139#[derive(Debug, Clone)]
140pub struct Embedding {
141    /// The APIs served beside the OpenAI-compatible routes.
142    pub http: superfluid_daemon::openai::HttpSurface,
143    /// How a request's `model` picks the model that serves it.
144    pub model_naming: superfluid_daemon::openai::ModelNaming,
145    /// Names registered artifacts (initial, additional and model-dir models).
146    /// Receives already registered names so an embedding can disambiguate
147    /// variants. Explicit HTTP model IDs are not rewritten.
148    pub model_name: fn(&Path, &std::collections::HashSet<String>) -> String,
149    /// Sampling policy for omitted parameters; None is superfluid's default.
150    pub sampling_fallback: Option<superfluid_daemon::openai::SamplingFallback>,
151    /// The name and version the HTTP API reports for the server.
152    pub identity: superfluid_daemon::openai::ServerIdentity,
153    /// Whether the native session API listens on a unix socket (`--socket`).
154    /// Without it the HTTP API is all there is, and `--no-http` is refused.
155    pub session_socket: bool,
156    /// The program that runs a linked runtime's worker, in place of the `superfluid-workerd`
157    /// beside this executable; it passes [`crate::workerd::main`]'s arguments to it.
158    pub workerd: Option<PathBuf>,
159}
160
161impl Default for Embedding {
162    fn default() -> Embedding {
163        Embedding {
164            http: superfluid_daemon::openai::HttpSurface::default(),
165            model_naming: superfluid_daemon::openai::ModelNaming::default(),
166            model_name: superfluid_daemon::registry::distinct_name,
167            sampling_fallback: None,
168            identity: superfluid_daemon::openai::ServerIdentity::default(),
169            session_socket: true,
170            workerd: None,
171        }
172    }
173}
174
175/// Runs `superfluid serve`; `args` is its command line after `serve`.
176pub fn run(args: &[String], embedding: &Embedding) {
177    let mut models: Vec<PathBuf> = Vec::new();
178    let mut pull_flags = superfluid_daemon::pulls::PullFlags::default();
179    let mut runtime_pick = RuntimePick::default();
180    let mut api_key: Option<String> = None;
181    let mut rate_limit: u32 = 0;
182    let mut host: Option<String> = None;
183    let mut port: Option<u16> = None;
184    let mut files_max_bytes: Option<u64> = None;
185    let mut files_expiry: Option<u64> = None;
186    let mut files_sweep: u64 = 300;
187    let mut drain_timeout: u64 = 60;
188    let mut nonstream_keepalive: u64 = 0;
189    let mut default_max_tokens: Option<u32> = None;
190    let mut log_file: Option<PathBuf> = None;
191    let mut model_dir: Option<PathBuf> = None;
192    let mut tool_envelope: Option<superfluid_daemon::codec::ToolEnvelope> = None;
193    let mut idle_timeout: u64 = 0;
194    let mut sessions: Option<PathBuf> = None;
195    let mut socket: Option<PathBuf> = None;
196    let mut http: Option<String> = None;
197    let mut no_http = false;
198    let mut max_context: Option<i32> = None;
199    let mut max_context_auto_flag = false;
200    let mut max_batch: u32 = 8;
201    let mut op_temperature: Option<f32> = None;
202    let mut op_top_p: Option<f32> = None;
203    let mut op_top_k: Option<u32> = None;
204    let mut op_min_p: Option<f32> = None;
205    let mut op_repeat_penalty: Option<f32> = None;
206    let mut op_spec_max_temperature: Option<f32> = None;
207    let mut park = false;
208    let mut sessions_cache = false;
209    let mut park_lossy = false;
210    let mut park_budget_gb: u64 = 20;
211    let mut tool_lease_ms: u64 = 0;
212    let mut prefix_blob_budget_mb: u64 = 0;
213    let mut kv_bits_arg: Option<i32> = None;
214    let mut worker_process = true;
215    let mut os_pressure = true;
216    let mut dialect = String::from("auto");
217    let mut pressure_high: u8 = 0;
218    let mut pressure_low: u8 = 0;
219    let mut pin_budget_pct: u8 = superfluid_daemon::DaemonOptions::default().pin_budget_pct;
220    let mut tick_decode_budget: u32 = 0;
221    let mut tick_target_ms: u32 = 0;
222    let mut prefill_budget: u32 = superfluid_daemon::DaemonOptions::default().prefill_budget;
223    let mut starvation_ticks: u32 = 8;
224    let mut class_lanes_spec: Option<String> = None;
225    let mut http_default_qos = superfluid_daemon::openai::RequestQos::AGENT;
226    let mut http_qos_header = true;
227    let mut http_allow_batch_invariant = false;
228    let mut key_policy_path: Option<PathBuf> = None;
229    let mut qos_flags: Vec<&'static str> = Vec::new();
230    let mut log_dir: Option<PathBuf> = None;
231    let mut log_filter = String::from("info");
232    let mut speculate: Option<String> = None;
233    let mut speculation = superfluid_daemon::SpeculationOptions::default();
234    let mut spec_bitexact: Option<bool> = None;
235    let mut completion_deadline_ms: u64 = 0;
236    let mut completion_bucket = superfluid_daemon::completion_bucket::BucketConfig::default();
237    let mut fim_model: Option<PathBuf> = None;
238    let mut fim_max_context: Option<i32> = None;
239    let mut fim_max_batch: u32 = 2;
240    let mut web: Option<String> = None;
241    let mut web_origins: Vec<String> = Vec::new();
242    let mut otlp_metrics: Option<String> = None;
243    let mut otlp_interval_ms: u64 = 15_000;
244    let mut otlp_endpoint: Option<String> = None;
245    let mut otlp_cfg = superfluid_daemon::otlp::OtlpConfig::default();
246    let mut tui_flag = false;
247    let mut no_tui = false;
248    let mut fleet: Option<String> = None;
249    let mut fleet_auth: Option<PathBuf> = None;
250    let mut fleet_listen: Option<String> = None;
251    let mut fleet_policy = String::from("load-aware");
252    let mut fleet_pool_high: u32 = superfluid_daemon::fleet_manager::DEFAULT_POOL_HIGH_PCT;
253    let mut fleet_conns: usize = superfluid_daemon::fleet_manager::DEFAULT_CONNS_PER_NODE;
254    // A runtime's option, kept as given and resolved once the catalog says which runtimes are here.
255    let mut option_candidates: Vec<(String, Option<String>)> = Vec::new();
256    let args = superfluid_daemon::cli::split_equals(args, superfluid_daemon::cli::serve_takes_value);
257    let mut it = args.iter();
258    while let Some(a) = it.next() {
259        let mut val = || it.next().cloned().unwrap_or_else(|| needs_value(a));
260        // SERVE-FLAGS-BEGIN: every flag `serve` reads is listed in superfluid_daemon::cli (a test checks).
261        match a.as_str() {
262            "--model" => models.push(PathBuf::from(val())),
263            "--offline" => pull_flags.offline = true,
264            flag if flag.starts_with("--pull-") => {
265                let spec = &flag["--pull-".len()..];
266                let (name, value) = match spec.split_once('=') {
267                    Some((n, v)) => (n.to_string(), Some(v.to_string())),
268                    None => (spec.to_string(), it.clone().next().filter(|v| !v.starts_with("--")).cloned()),
269                };
270                if value.is_some() && !spec.contains('=') {
271                    it.next();
272                }
273                if name.is_empty() {
274                    unknown_serve_flag(a);
275                }
276                pull_flags.options.push((name, value));
277            }
278            "--runtime" => {
279                let v = val();
280                if let Err(e) = runtime_pick.add(&v) {
281                    eprintln!("superfluid: --runtime: {e}");
282                    std::process::exit(2);
283                }
284            }
285            "--api-key" => api_key = Some(val()),
286            "--rate-limit" => rate_limit = num(a, val()),
287            "--files-max-bytes" => {
288                files_max_bytes = Some(num(a, val()))
289            }
290            "--files-expiry" => files_expiry = Some(num(a, val())),
291            "--files-sweep" => files_sweep = num(a, val()),
292            "--drain-timeout" => drain_timeout = num(a, val()),
293            "--nonstream-keepalive" => {
294                nonstream_keepalive = num(a, val())
295            }
296            "--max-tokens" => {
297                default_max_tokens = Some(num(a, val()))
298            }
299            "--log-file" => log_file = Some(PathBuf::from(val())),
300            "--model-dir" => model_dir = Some(PathBuf::from(val())),
301            "--tool-call-parser" => {
302                let v = val();
303                if v != "auto" {
304                    tool_envelope = Some(
305                        superfluid_daemon::codec::ToolEnvelope::from_name(&v).unwrap_or_else(|| {
306                            eprintln!(
307                                "superfluid: unknown --tool-call-parser '{v}' (have: auto, {})",
308                                superfluid_daemon::codec::ToolEnvelope::names().join(", ")
309                            );
310                            std::process::exit(2);
311                        }),
312                    );
313                }
314            }
315            "--idle-timeout" => idle_timeout = num(a, val()),
316            "--host" => host = Some(val()),
317            "--port" => port = Some(num(a, val())),
318            "--max-batch-size" => max_batch = num(a, val()),
319            "--request-timeout" => {
320                completion_deadline_ms = num(a, val())
321            }
322            "--verbose" | "-v" => log_filter = "debug".into(),
323            "--help" | "-h" => {
324                print!("{}", superfluid_daemon::cli::serve_help());
325                return;
326            }
327            "--sessions" => sessions = Some(PathBuf::from(val())),
328            "--socket" => socket = Some(PathBuf::from(val())),
329            "--http" => http = Some(val()),
330            "--no-http" => no_http = true,
331            "--max-context" => {
332                let v = val();
333                max_context_auto_flag = v == "auto";
334                max_context = if v == "auto" { None } else { Some(num(a, v)) };
335            }
336            "--max-batch" => max_batch = num(a, val()),
337            "--temperature" => op_temperature = Some(sampling_f32("--temperature", &val(), 0.0, 2.0)),
338            "--top-p" => op_top_p = Some(sampling_f32("--top-p", &val(), 0.0, 1.0)),
339            "--top-k" => op_top_k = Some(num(a, val())),
340            "--min-p" => op_min_p = Some(sampling_f32("--min-p", &val(), 0.0, 1.0)),
341            "--repeat-penalty" => {
342                op_repeat_penalty = Some(sampling_f32("--repeat-penalty", &val(), 0.0, 2.0))
343            }
344            "--park" => park = true,
345            "--no-park" => park = false,
346            "--sessions-cache" => sessions_cache = true,
347            "--park-lossy" => {
348                park = true;
349                park_lossy = true;
350            }
351            "--park-budget-gb" => park_budget_gb = num(a, val()),
352            "--tool-lease-ms" => tool_lease_ms = num(a, val()),
353            "--prefix-blob-budget-mb" => {
354                prefix_blob_budget_mb = num(a, val())
355            }
356            "--kv-bits" => kv_bits_arg = Some(num(a, val())),
357            "--dialect" => dialect = val(),
358            "--worker-process" => worker_process = true,
359            "--no-worker-process" => worker_process = false,
360            "--os-pressure" => os_pressure = true,
361            "--no-os-pressure" => os_pressure = false,
362            "--pressure-high" => pressure_high = num(a, val()),
363            "--pressure-low" => pressure_low = num(a, val()),
364            "--pin-budget-pct" => pin_budget_pct = num(a, val()),
365            "--log-dir" => log_dir = Some(PathBuf::from(val())),
366            "--basert-lib" => std::env::set_var("BASERT_LIB", val()),
367            "--speculate" => {
368                let v = val();
369                speculate = if v == "off" { None } else { Some(v) };
370            }
371            "--spec-draft-tokens" => {
372                let n: u32 = num(a, val());
373                if !(1..=15).contains(&n) {
374                    eprintln!("superfluid: --spec-draft-tokens must be in 1..=15");
375                    std::process::exit(2);
376                }
377                speculation.max_draft = Some(n);
378            }
379            "--spec-adaptive" => speculation.adaptive = Some(true),
380            "--no-spec-adaptive" => speculation.adaptive = Some(false),
381            "--spec-min-yield" => {
382                let v: f64 = num(a, val());
383                if !v.is_finite() || v < 0.0 {
384                    eprintln!("superfluid: --spec-min-yield must be finite and non-negative");
385                    std::process::exit(2);
386                }
387                speculation.min_yield = Some(v);
388            }
389            "--spec-yield-rounds" => {
390                let n: u32 = num(a, val());
391                if n == 0 {
392                    eprintln!("superfluid: --spec-yield-rounds must be at least 1");
393                    std::process::exit(2);
394                }
395                speculation.yield_rounds = Some(n);
396            }
397            "--spec-throughput-gate" => speculation.throughput_gate = Some(true),
398            "--no-spec-throughput-gate" => speculation.throughput_gate = Some(false),
399            "--spec-gate-probe-tokens" => {
400                speculation.gate_probe_tokens = Some(num(a, val()))
401            }
402            "--spec-max-temperature" => {
403                op_spec_max_temperature = Some(sampling_f32("--spec-max-temperature", &val(), 0.0, 2.0))
404            }
405            "--spec-min-speedup" => {
406                let v: f64 = num(a, val());
407                if !v.is_finite() || v < 1.0 {
408                    eprintln!("superfluid: --spec-min-speedup must be finite and at least 1.0");
409                    std::process::exit(2);
410                }
411                speculation.min_speedup = Some(v);
412            }
413            "--spec-gate-reprobe" => {
414                speculation.gate_reprobe = Some(num(a, val()))
415            }
416            "--spec-gate-reprobe-max" => {
417                speculation.gate_reprobe_max = Some(num(a, val()))
418            }
419            "--dspark-confidence" => {
420                let v: f32 = num(a, val());
421                if !v.is_finite() || v <= 0.0 {
422                    eprintln!("superfluid: --dspark-confidence must be finite and positive");
423                    std::process::exit(2);
424                }
425                speculation.dspark_confidence = Some(v);
426            }
427            "--spec-bitexact" => spec_bitexact = Some(true),
428            "--no-spec-bitexact" => spec_bitexact = Some(false),
429            "--log-filter" => log_filter = val(),
430            "--tick-decode-budget" => {
431                tick_decode_budget = num(a, val())
432            }
433            "--tick-target-ms" => tick_target_ms = num(a, val()),
434            "--prefill-budget" => prefill_budget = num(a, val()),
435            "--starvation-ticks" => starvation_ticks = num(a, val()),
436            "--class-lanes" => {
437                class_lanes_spec = Some(val());
438                qos_flags.push("--class-lanes");
439            }
440            "--http-default-qos" => {
441                qos_flags.push("--http-default-qos");
442                let v = val();
443                http_default_qos.class = superfluid_daemon::qos::parse(&v).unwrap_or_else(|| {
444                    eprintln!(
445                        "superfluid: --http-default-qos expects one of {}, got {v:?}",
446                        superfluid_daemon::qos::NAMES.join(", ")
447                    );
448                    std::process::exit(2);
449                });
450            }
451            "--no-http-qos-header" => {
452                http_qos_header = false;
453                qos_flags.push("--no-http-qos-header");
454            }
455            "--http-allow-batch-invariant" => {
456                http_allow_batch_invariant = true;
457                qos_flags.push("--http-allow-batch-invariant");
458            }
459            "--key-policy" => key_policy_path = Some(PathBuf::from(val())),
460            "--completion-deadline-ms" => {
461                completion_deadline_ms = num(a, val())
462            }
463            "--completion-rate" => {
464                completion_bucket.rate_per_sec = num(a, val())
465            }
466            "--completion-burst" => {
467                completion_bucket.burst = num(a, val())
468            }
469            "--fim-model" => fim_model = Some(PathBuf::from(val())),
470            "--fim-max-context" => {
471                fim_max_context = Some(num(a, val()))
472            }
473            "--fim-max-batch" => fim_max_batch = num(a, val()),
474            "--web" => web = Some(val()),
475            "--web-origin" => web_origins.push(val()),
476            "--otlp-metrics" => otlp_metrics = Some(val()),
477            "--tui" => tui_flag = true,
478            "--no-tui" => no_tui = true,
479            "--otlp-interval-ms" => otlp_interval_ms = num(a, val()),
480            "--otlp-endpoint" => otlp_endpoint = Some(val()),
481            "--otlp-header" => {
482                let kv = val();
483                let (k, v) = kv.split_once('=').unwrap_or_else(|| usage_error(&format!("--otlp-header expects k=v, got {kv:?}"), "serve"));
484                otlp_cfg.headers.push((k.trim().to_string(), v.trim().to_string()));
485            }
486            "--otlp-service-name" => otlp_cfg.service_name = val(),
487            "--otlp-filter" => otlp_cfg.filter = val(),
488            "--otlp-queue" => otlp_cfg.queue = num(a, val()),
489            "--otlp-batch-ms" => {
490                let ms = num(a, val());
491                otlp_cfg.interval = std::time::Duration::from_millis(ms);
492            }
493            "--otlp-timeout-ms" => {
494                let ms = num(a, val());
495                otlp_cfg.timeout = std::time::Duration::from_millis(ms);
496            }
497            "--fleet" => fleet = Some(val()),
498            "--fleet-auth" => fleet_auth = Some(PathBuf::from(val())),
499            "--fleet-listen" => fleet_listen = Some(val()),
500            "--fleet-policy" => fleet_policy = val(),
501            "--fleet-pool-high" => fleet_pool_high = num(a, val()),
502            "--fleet-conns-per-node" => {
503                fleet_conns = num(a, val())
504            }
505            "--paged-kv" => retired(a, "paged KV is always on"),
506            "--prefix-cache" => retired(a, "prefix reuse is always on"),
507            "--continuous-batching" => {
508                if let Some(n) = it.clone().next().and_then(|v| v.parse::<u32>().ok()) {
509                    let _ = it.next();
510                    max_batch = n;
511                    retired(a, "always on; its width became --max-batch");
512                } else {
513                    retired(a, "continuous batching is always on");
514                }
515            }
516            "--prefix-cache-file" | "--prefix-cache-save-interval" => {
517                let _ = val();
518                retired(
519                    a,
520                    "superseded by the WAL session log (tokens, always durable) and the \
521                     in-memory prefix cache; KV on disk is --park, which is OFF by default \
522                     and applies to sessions resumed by id through the session API",
523                );
524            }
525            "--files-dir" => {
526                let _ = val();
527                retired(a, "the file store lives under --sessions");
528            }
529            "--media-dir" => {
530                let _ = val();
531                retired(
532                    a,
533                    "superfluid accepts only data: image URLs and never reads a local path, so there is nothing to allow-list",
534                );
535            }
536            "--prefill-chunk" => {
537                let _ = val();
538                retired(a, "an engine-level knob; a runtime takes its own, --mlx-prefill-chunk or --llamacpp-ubatch-size");
539            }
540            "--metallib" | "--gpu-wait-timeout-ms" | "--decode-replay" => {
541                let _ = val();
542                retired(a, "an engine-level knob superfluid does not expose");
543            }
544            "--paged-weights" | "--no-paged-weights-retry" | "--no-baked-decode" => {
545                retired(a, "an engine-level knob superfluid does not expose")
546            }
547            other if !other.starts_with('-') => models.push(PathBuf::from(other)),
548            _ => {
549                let (flag, given) = match a.split_once('=') {
550                    Some((f, v)) => (f, Some(v.to_string())),
551                    None => (a.as_str(), None),
552                };
553                if !superfluid_daemon::runtime_options::looks_like(flag) {
554                    unknown_serve_flag(a);
555                }
556                let value = match given {
557                    Some(v) => Some(v),
558                    None => match it.clone().next() {
559                        Some(v) if !v.starts_with("--") => {
560                            it.next();
561                            Some(v.clone())
562                        }
563                        _ => None,
564                    },
565                };
566                option_candidates.push((flag.to_string(), value));
567            }
568        }
569        // SERVE-FLAGS-END
570    }
571    if models.is_empty() {
572        usage_error(
573            "serve needs a model: superfluid serve <path|org/model[:tag]>, e.g. \
574             superfluid serve unsloth/Qwen3.8-27B-GGUF:UD-Q4_K_M",
575            "serve",
576        );
577    }
578    superfluid_daemon::speculate_resolve::allow_catalog_fetch(!pull_flags.offline);
579    let catalog = Arc::new(match &embedding.workerd {
580        Some(workerd) => superfluid_daemon::runtimes::Catalog::with_workerd(workerd.clone()),
581        None => superfluid_daemon::runtimes::Catalog::from_env(),
582    });
583    let runtime_options: Vec<superfluid_daemon::runtime_options::Given> = {
584        let ids = catalog.ids();
585        option_candidates
586            .iter()
587            .map(|(flag, value)| match superfluid_daemon::runtime_options::resolve(flag, value.as_deref().unwrap_or_default(), &ids) {
588                None => unknown_serve_flag(flag),
589                Some(_) if value.is_none() => needs_value(flag),
590                Some(given) => given,
591            })
592            .collect()
593    };
594    superfluid_daemon::runtime_options::refuse_repeats(&runtime_options).unwrap_or_else(|e| usage_error(&e, "serve"));
595    // Set before the runtime is required, so an option that makes it usable here counts for the first probe.
596    superfluid_daemon::runtime_options::set(superfluid_daemon::runtime_options::given_pairs(&runtime_options));
597    // The llama.cpp tokenizer this process loads finds its library the way the worker does.
598    superfluid_daemon::runtime_options::export_to_daemon();
599    let sessions = Some(sessions.unwrap_or_else(|| superfluid_daemon::cli::default_sessions(catalog.home())));
600    let socket = if embedding.session_socket {
601        Some(socket.unwrap_or_else(|| superfluid_daemon::cli::default_socket(catalog.home(), SUN_PATH_LEN)))
602    } else {
603        if socket.is_some() {
604            usage_error("--socket: this server has no session socket", "serve");
605        }
606        if no_http {
607            usage_error("--no-http leaves nothing to serve: this server has no session socket", "serve");
608        }
609        None
610    };
611    if no_http && (http.is_some() || host.is_some() || port.is_some()) {
612        usage_error("--no-http contradicts --http/--host/--port", "serve");
613    }
614    let fleet_mode = fleet.is_some() || fleet_listen.is_some();
615    let mut pulled_names: std::collections::HashMap<PathBuf, String> = std::collections::HashMap::new();
616    for m in models.iter_mut() {
617        let token = m.to_string_lossy().into_owned();
618        // A fleet head takes a model given by id from a node serving it, weights and all left there.
619        if !superfluid_daemon::pulls::looks_like_id(&token) || fleet_mode {
620            continue;
621        }
622        let runtime = match runtime_pick.explicit(m) {
623            Some(name) => catalog.named(name).ok().or_else(|| RuntimeId::parse(name)).unwrap_or_else(|| {
624                eprintln!("superfluid: --runtime '{name}' is not a runtime id");
625                std::process::exit(2);
626            }),
627            None => RuntimeId::new(superfluid_daemon::cli::runtime_for_id(&token)),
628        };
629        // The runtime first, so one missing exits as it would for a model given by path (4).
630        if let Err(e) = superfluid_daemon::pulls::ensure_installed(&catalog, runtime, pull_flags.offline) {
631            refuse(e.exit_code(), e.to_string());
632        }
633        let path = superfluid_daemon::pulls::pull(&catalog, runtime, &token, &pull_flags).unwrap_or_else(|e| {
634            eprintln!("superfluid: {e}");
635            std::process::exit(1);
636        });
637        if let Err(e) = runtime_pick.add(&format!("{}={}", path.display(), runtime.name())) {
638            eprintln!("superfluid: {e}");
639            std::process::exit(2);
640        }
641        pulled_names.insert(path.clone(), superfluid_daemon::pulls::served_name(&token));
642        *m = path;
643    }
644    if !pull_flags.options.is_empty() && pulled_names.is_empty() {
645        eprintln!("superfluid: --pull-* options apply to models given by id; no --model here is one");
646        std::process::exit(2);
647    }
648    let (Some(model), Some(sessions)) = (models.first().cloned(), sessions) else {
649        unreachable!("a model and sessions are both set above");
650    };
651    if http.is_none() && !no_http {
652        http = Some(format!(
653            "{}:{}",
654            host.unwrap_or_else(|| "127.0.0.1".into()),
655            port.unwrap_or(8453)
656        ));
657    }
658    if let Some(socket) = socket.as_ref().filter(|s| s.as_os_str().len() >= SUN_PATH_LEN) {
659        eprintln!(
660            "superfluid: --socket {} is {} bytes; a unix socket path must be under {SUN_PATH_LEN} bytes here \
661             (pick a shorter one)",
662            socket.display(),
663            socket.as_os_str().len()
664        );
665        std::process::exit(2);
666    }
667    for (flag, addr) in [("--http", &http), ("--web", &web)] {
668        if let Some(addr) = addr {
669            drop(listen_or_exit(flag, addr));
670        }
671    }
672    let key_policy = key_policy_path.as_deref().map(|p| {
673        let table = superfluid_daemon::keypolicy::KeyTable::load(p, api_key.as_deref())
674            .unwrap_or_else(|e| {
675                eprintln!("superfluid: --key-policy {e}");
676                std::process::exit(2);
677            });
678        eprintln!("superfluid: key policy: {} key(s) from {}", table.len(), p.display());
679        Arc::new(table)
680    });
681    if fleet_mode {
682        if otlp_endpoint.is_some() {
683            eprintln!(
684                "superfluid: --otlp-endpoint is not supported in fleet-head mode (the fleet head \
685                 has no scheduler to trace); set it on each node's superfluid instead"
686            );
687            std::process::exit(2);
688        }
689        {
690            use superfluid_daemon::telemetry::{LogConfig, LogSink};
691            let sink = match log_dir.clone() {
692                Some(dir) => LogSink::File(LogConfig { dir, filter: log_filter.clone(), ..Default::default() }),
693                None => LogSink::Stderr(log_filter.clone()),
694            };
695            if let Err(e) = superfluid_daemon::telemetry::init_sinks(sink, None, None) {
696                eprintln!("superfluid: telemetry: {e}");
697            }
698        }
699        std::fs::create_dir_all(&sessions).expect("create sessions dir");
700        let _sessions_lock = lock_sessions_dir(&sessions);
701        if sessions_cache {
702            rotate_session_wal(&sessions);
703            if let Ok(rd) = std::fs::read_dir(sessions.join("models")) {
704                for dir in rd.filter_map(|e| e.ok()).map(|e| e.path()).filter(|d| d.is_dir()) {
705                    rotate_session_wal(&dir);
706                }
707            }
708        }
709        let fleet_addrs = fleet.clone().unwrap_or_default();
710        use superfluid_daemon::fleet_manager::{FleetManager, NodeSpec, PlacementPolicy};
711        let auth: Vec<u8> = match &fleet_auth {
712            Some(pth) => std::fs::read(pth)
713                .map(|t| t.trim_ascii().to_vec())
714                .unwrap_or_else(|e| {
715                    eprintln!("superfluid: cannot read --fleet-auth {}: {e}", pth.display());
716                    std::process::exit(2);
717                }),
718            None => superfluid_daemon::fleet_token::read(catalog.home()).unwrap_or_default(),
719        };
720        if auth.is_empty() {
721            eprintln!("superfluid: no fleet token (`superfluid fleet token` makes one): the links to the nodes are not encrypted, and only nodes on this machine may join");
722        }
723        let specs: Vec<NodeSpec> = fleet_addrs
724            .split(',')
725            .map(str::trim)
726            .filter(|a| !a.is_empty())
727            .map(|a| NodeSpec { addr: a.to_string(), auth: auth.clone() })
728            .collect();
729        if specs.is_empty() && fleet_listen.is_none() {
730            eprintln!("superfluid: --fleet needs at least one host:port, or --fleet-listen for nodes that join");
731            std::process::exit(2);
732        }
733        let policy = match fleet_policy.as_str() {
734            "load-aware" => PlacementPolicy::LoadAware,
735            "least-loaded" => PlacementPolicy::LeastLoaded,
736            "round-robin" => PlacementPolicy::RoundRobin,
737            other => {
738                eprintln!(
739                    "superfluid: unknown --fleet-policy '{other}' \
740                     (load-aware|least-loaded|round-robin)"
741                );
742                std::process::exit(2);
743            }
744        };
745        let Some(addr) = http else {
746            eprintln!("superfluid: --fleet requires --http <addr> (the fleet OpenAI endpoint)");
747            std::process::exit(2);
748        };
749        let lease = superfluid_proto::linkf::Lease { duration_ms: 60_000, renew_by_ms: 40_000 };
750        let n_nodes = specs.len();
751        let manager = Arc::new(
752            FleetManager::with_options(specs, policy, lease, fleet_pool_high, fleet_conns)
753                .with_warm_log(sessions.join("fleet-warm.json")),
754        );
755        // Every model the head serves: one given by path is opened here for its tokenizer and
756        // chat template only; one given by id takes them from a node serving it, below.
757        let mut served: Vec<(String, Option<Box<dyn superfluid_daemon::TextCodec + Send + Sync>>)> = Vec::new();
758        let mut taken: std::collections::HashSet<String> = std::collections::HashSet::new();
759        for m in &models {
760            let token = m.to_string_lossy().into_owned();
761            let by_id = !m.exists() && superfluid_daemon::pulls::looks_like_id(&token);
762            let name = if by_id { superfluid_daemon::pulls::served_name(&token) } else { (embedding.model_name)(m, &taken) };
763            if !taken.insert(name.clone()) {
764                eprintln!("superfluid: model '{name}' given twice, skipping");
765                continue;
766            }
767            if by_id {
768                served.push((name, None));
769                continue;
770            }
771            let rt = runtime_pick.resolve(m, &catalog).and_then(|id| catalog.require(id)).unwrap_or_else(|e| {
772                eprintln!("superfluid: {}: {e}", m.display());
773                std::process::exit(2);
774            });
775            let tok = open_tokenizer(&rt, m).unwrap_or_else(|e| {
776                eprintln!("superfluid: {e}");
777                std::process::exit(2);
778            });
779            manager.expect_tokenizer(&name, superfluid_daemon::fleet::tokenizer_digest(&*tok));
780            let codec = codec_for_dialect(&dialect, tok).map(|(c, _)| c).unwrap_or_else(|e| {
781                eprintln!("superfluid: {}: {e}", m.display());
782                std::process::exit(2);
783            });
784            served.push((name, Some(codec)));
785        }
786        let names: Vec<String> = served.iter().map(|(n, _)| n.clone()).collect();
787        let model_name = names.first().cloned().unwrap_or_else(|| {
788            eprintln!("superfluid: no model to serve");
789            std::process::exit(2);
790        });
791        let mut reached = 0;
792        for node in manager.probe() {
793            match node {
794                Ok(n) => {
795                    reached += 1;
796                    eprintln!(
797                        "superfluid: fleet node '{}' at {}: {}, context window {} tokens, {} lanes",
798                        n.identity,
799                        n.addr,
800                        n.models.join(", "),
801                        n.max_context_tokens,
802                        n.max_lanes
803                    );
804                    if !n.models.iter().any(|m| names.contains(m)) {
805                        eprintln!(
806                            "superfluid: fleet node '{}' serves {}, none of {}, so nothing is placed \
807                             there; start it with --model-name <one of them>",
808                            n.identity,
809                            n.models.join(", "),
810                            names.join(", ")
811                        );
812                    }
813                }
814                Err(e) => eprintln!("superfluid: {e}; retrying it in the background"),
815            }
816        }
817        if max_context.is_none() {
818            for name in &names {
819                match manager.advertised_context(name) {
820                    Some(n) => eprintln!("superfluid: {name}: context window {n} tokens (the largest node's)"),
821                    None => eprintln!("superfluid: {name}: context window unknown until a node serving it answers"),
822                }
823            }
824        }
825        manager.spawn_keepalive();
826        let _announced = fleet_listen.as_deref().map(|listen| {
827            let listener = std::net::TcpListener::bind(listen).unwrap_or_else(|e| {
828                eprintln!("superfluid: --fleet-listen {listen}: {e}");
829                std::process::exit(1);
830            });
831            let port = listener.local_addr().map(|a| a.port()).unwrap_or_default();
832            eprintln!("superfluid: fleet head takes joining nodes on {listen}");
833            let (m, a, all) = (Arc::clone(&manager), auth.clone(), names.clone());
834            std::thread::spawn(move || superfluid_daemon::fleet_join::accept_nodes(listener, m, a, all));
835            // `superfluid top` finds the head's API by this; one on loopback is no use off this machine.
836            let http_port = addr.parse::<std::net::SocketAddr>().ok().filter(|a| !a.ip().is_loopback()).map(|a| a.port());
837            match superfluid_daemon::fleet_mdns::advertise(port, &names, http_port) {
838                Ok(d) => Some(d),
839                Err(e) => {
840                    eprintln!("superfluid: not announced on the local network ({e}); give nodes this head's address");
841                    None
842                }
843            }
844        });
845        if !qos_flags.is_empty() {
846            eprintln!(
847                "superfluid: {} not supported in fleet-head mode (the fleet head has no \
848                 local scheduler and Link F carries no QoS class); set them on each \
849                 node's own `superfluid serve` instead",
850                qos_flags.join(", ")
851            );
852            std::process::exit(2);
853        }
854        if op_spec_max_temperature.is_some() {
855            eprintln!(
856                "superfluid: --spec-max-temperature is not supported in fleet-head mode; \
857                 set it on each node's own `superfluid serve` instead"
858            );
859            std::process::exit(2);
860        }
861        if fim_model.is_some() {
862            eprintln!(
863                "superfluid: --fim-model is not supported in fleet-head mode (the head hosts no \
864                 engine); set it on a node's own `superfluid serve` instead"
865            );
866            std::process::exit(2);
867        }
868        if op_repeat_penalty.is_some() {
869            eprintln!(
870                "superfluid: --repeat-penalty is not supported in fleet-head mode \
871                 (the fleet protocol carries no per-request penalty); \
872                 set it on each node's own `superfluid serve` instead"
873            );
874            std::process::exit(2);
875        }
876        let http_listener = listen_or_exit("--http", &addr);
877        eprintln!(
878            "superfluid: fleet-head openai api on http://{addr} over {n_nodes} node(s), {reached} reachable"
879        );
880        if key_policy.is_some() {
881            eprintln!(
882                "superfluid: --key-policy at the fleet head: keys, rate and concurrency limits \
883                 apply; class and max_class do not (the head has no local scheduler)"
884            );
885        }
886        // A model given by id is served once a node serving it has shared its tokenizer.
887        let served: Vec<(String, Box<dyn superfluid_daemon::TextCodec + Send + Sync>)> = served
888            .into_iter()
889            .map(|(name, codec)| {
890                let codec = codec.unwrap_or_else(|| {
891                    let mut said = false;
892                    let tok = loop {
893                        match manager.remote_tokenizer(&name) {
894                            Ok(t) => break t,
895                            Err(e) => {
896                                if !said {
897                                    eprintln!("superfluid: {name}: waiting for a node serving it to share its tokenizer ({e})");
898                                    said = true;
899                                }
900                                std::thread::sleep(std::time::Duration::from_secs(2));
901                            }
902                        }
903                    };
904                    eprintln!("superfluid: {name}: tokenizer and chat template from a node serving it");
905                    manager.expect_tokenizer(&name, tok.digest());
906                    codec_for_dialect(&dialect, Arc::new(tok)).map(|(c, _)| c).unwrap_or_else(|e| {
907                        eprintln!("superfluid: {name}: {e}");
908                        std::process::exit(2);
909                    })
910                });
911                (name, codec)
912            })
913            .collect();
914        let advertised = manager.advertised_context(&model_name).map(|n| n.min(u32::MAX as u64) as u32);
915        let mut registry = None;
916        for (i, (name, codec)) in served.into_iter().enumerate() {
917            // The first model's sessions where a single server keeps them; each other's in a
918            // directory of its own, as a server with several models keeps them.
919            let dir = if i == 0 {
920                sessions.clone()
921            } else {
922                superfluid_daemon::registry::state_dir(&sessions.join("models"), &name).unwrap_or_else(|e| {
923                    eprintln!("superfluid: {name}: cannot keep its sessions: {e}");
924                    std::process::exit(1);
925                })
926            };
927            let daemon = Daemon::over_fleet(
928                open_session_store(&dir),
929                codec,
930                Arc::clone(&manager),
931                name.clone(),
932                superfluid_daemon::DaemonOptions {
933                    tool_lease_ms,
934                    media_dir: Some(sessions.join("media")),
935                    files_max_bytes,
936                    files_expiry_secs: files_expiry,
937                    completion_deadline_ms,
938                    trace_scope: superfluid_daemon::otlp::TraceScope::model(&name),
939                    ..Default::default()
940                },
941            )
942            .unwrap_or_else(|e| {
943                eprintln!("superfluid: cannot serve '{name}' over the fleet: {e}");
944                std::process::exit(2);
945            });
946            match &registry {
947                None => registry = Some(Arc::new(superfluid_daemon::registry::ModelRegistry::single(name, Arc::new(daemon)))),
948                Some(r) => r.insert_loaded(&name, Arc::new(daemon)),
949            }
950        }
951        let registry = registry.expect("the first model is served");
952        superfluid_daemon::openai::serve_blocking_config(
953            http_listener,
954            registry,
955            superfluid_daemon::openai::ServeConfig {
956                api_key: api_key.clone(),
957                rate_limit_per_minute: rate_limit,
958                drain_timeout_secs: drain_timeout,
959                default_max_tokens,
960                max_context: max_context.map(|n| n.max(512) as u32).or(advertised).unwrap_or(8192),
961                sampling: superfluid_daemon::openai::SamplingOverrides {
962                    temperature: op_temperature,
963                    top_p: op_top_p,
964                    top_k: op_top_k,
965                    min_p: op_min_p,
966                    repeat_penalty: op_repeat_penalty,
967                    spec_max_temperature: None,
968                    fallback: embedding.sampling_fallback,
969                },
970                keys: key_policy.clone(),
971                nonstream_keepalive_secs: nonstream_keepalive,
972                surface: embedding.http,
973                model_naming: embedding.model_naming,
974                identity: embedding.identity.clone(),
975                idempotency_dir: Some(sessions.join("idempotency")),
976                ..Default::default()
977            },
978        )
979        .expect("fleet serve");
980        return;
981    }
982    let picked = runtime_pick.explicit(&model).is_some();
983    fn refuse(code: i32, e: String) -> ! {
984        eprintln!("superfluid: {e}");
985        std::process::exit(code)
986    }
987    let runtime = match runtime_pick.resolve(&model, &catalog) {
988        Err(e) => refuse(2, if picked { e } else { not_there(&model).unwrap_or(e) }),
989        Ok(rt) => match catalog.require(rt) {
990            Ok(rt) => rt,
991            Err(_) => {
992                // The runtime answers without its options and refuses one of them: say which. Asked once, since
993                // asking starts a worker.
994                if let Some(why) = superfluid_daemon::runtime_options::blamed(&catalog, rt, &runtime_options) {
995                    refuse(2, why);
996                }
997                if let Some(why) = not_there(&model) {
998                    refuse(2, why);
999                }
1000                // Not here, or cannot run here: exit 4, so a host can tell it from a flag it got wrong;
1001                // an install tried and failed is 1, a failure along the way (`pulls::Unready`).
1002                superfluid_daemon::pulls::ensure_installed(&catalog, rt, pull_flags.offline)
1003                    .unwrap_or_else(|e| refuse(e.exit_code(), e.to_string()))
1004            }
1005        },
1006    };
1007    if let Err(e) = refuse_statically(&runtime, park_lossy, kv_bits_arg, speculate.as_deref()) {
1008        refuse(2, e);
1009    }
1010    // Each runtime's options are checked against what its worker declares before any model loads.
1011    superfluid_daemon::runtime_options::settle(&runtime_options, &runtime, &catalog).unwrap_or_else(|e| refuse(2, e));
1012    eprintln!("superfluid: runtime {} serves {} ({})", runtime.id.name(), model.display(), describe_worker(&runtime));
1013    if let Some(on) = spec_bitexact {
1014        std::env::set_var("BASERT_SPEC_BITEXACT", if on { "1" } else { "0" });
1015    }
1016    let (speculate_primary, speculate_fallbacks): (Option<String>, Vec<String>) = match speculation_for(
1017        speculate.as_deref(),
1018        &runtime.static_capabilities(),
1019        &model,
1020        runtime.id.name(),
1021    )
1022    .map(|d| superfluid_daemon::speculate_resolve::resolve_all(d, &model))
1023    {
1024        None => (None, Vec::new()),
1025        Some(Ok(mut ranked)) => {
1026            if let Some(r) = ranked.first() {
1027                eprintln!("superfluid: speculation: {r}");
1028            } else {
1029                let dirs = superfluid_daemon::speculate_resolve::auto_search_dirs(&model);
1030                let dirs: Vec<_> = dirs.iter().map(|(d, _)| d.display().to_string()).collect();
1031                eprintln!(
1032                    "superfluid: speculation: auto found no head in {} and no fitting drafter in {} — decoding plainly",
1033                    model.display(),
1034                    dirs.join(", ")
1035                );
1036            }
1037            let primary = (!ranked.is_empty()).then(|| ranked.remove(0));
1038            (primary, ranked)
1039        }
1040        Some(Err(e)) => {
1041            eprintln!("superfluid: {e}");
1042            std::process::exit(2);
1043        }
1044    };
1045    if op_spec_max_temperature.is_some() && speculate.is_none() {
1046        eprintln!("superfluid: --spec-max-temperature ignored: no --speculate strategy is configured");
1047    }
1048    if tui_flag && !cfg!(feature = "tui") {
1049        eprintln!(
1050            "superfluid: --tui: this build carries no TUI (build superfluid-daemon with the `tui` feature, \
1051             which `basert` includes)"
1052        );
1053        std::process::exit(2);
1054    }
1055    let class_lanes = match class_lanes_spec.as_deref() {
1056        None => [0; 4],
1057        Some(spec) => superfluid_daemon::qos::parse_class_lanes(spec, max_batch as usize).unwrap_or_else(|e| {
1058            eprintln!("superfluid: --class-lanes: {e}");
1059            std::process::exit(2);
1060        }),
1061    };
1062    if let Err(e) = completion_bucket.validate() {
1063        eprintln!("superfluid: --completion-rate/--completion-burst: {e}");
1064        std::process::exit(2);
1065    }
1066    if fim_max_batch == 0 {
1067        eprintln!("superfluid: --fim-max-batch must be at least 1");
1068        std::process::exit(2);
1069    }
1070    if sessions_cache && park {
1071        eprintln!(
1072            "superfluid: --sessions-cache discards the session log at each start, which \
1073             --park{} needs to resume parked sessions. Use one or the other.",
1074            if park_lossy { "-lossy" } else { "" }
1075        );
1076        std::process::exit(2);
1077    }
1078    if models.len() > 1 && !worker_process {
1079        eprintln!("superfluid: {} models — enabling --worker-process (each needs its own address space)", models.len());
1080        worker_process = true;
1081    }
1082    if fim_model.is_some() && !worker_process {
1083        eprintln!("superfluid: --fim-model needs its own worker — enabling --worker-process");
1084        worker_process = true;
1085    }
1086    if model_dir.is_some() && !worker_process {
1087        eprintln!("superfluid: --model-dir needs a per-model worker — enabling --worker-process");
1088        worker_process = true;
1089    }
1090    if let Some(path) = &log_file {
1091        use std::os::fd::AsRawFd;
1092        match std::fs::OpenOptions::new().create(true).append(true).open(path) {
1093            Ok(f) => {
1094                // SAFETY: `f` is an open, valid fd for the duration of the
1095                // call; dup2 onto STDERR_FILENO is the documented way to
1096                // redirect a stream. The original stderr is closed by dup2.
1097                unsafe {
1098                    libc::dup2(f.as_raw_fd(), libc::STDERR_FILENO);
1099                }
1100                drop(f);
1101            }
1102            Err(e) => {
1103                eprintln!("superfluid: could not open --log-file {}: {e}", path.display());
1104                std::process::exit(1);
1105            }
1106        }
1107    }
1108    if api_key.is_none() && key_policy.is_none() {
1109        let loopback = |addr: &str| {
1110            addr.starts_with("localhost:") || addr.parse::<std::net::SocketAddr>().is_ok_and(|a| a.ip().is_loopback())
1111        };
1112        if http.as_deref().is_none_or(loopback) && web.as_deref().is_none_or(loopback) {
1113            eprintln!("superfluid: no --api-key: the HTTP API is open to every process on this machine");
1114        } else {
1115            eprintln!(
1116                "superfluid: WARNING --api-key not set. Authentication is OFF — anyone \
1117                 who can reach this port can use the API. Bind loopback or set --api-key."
1118            );
1119        }
1120    }
1121    if park && !fleet_mode {
1122        let budget = if park_budget_gb == 0 {
1123            "NO budget: --park-budget-gb 0 disables the retention sweep, so the \
1124             directory grows without limit"
1125                .to_string()
1126        } else {
1127            format!("budget {park_budget_gb} GiB, oldest evicted first")
1128        };
1129        eprintln!(
1130            "superfluid: WARNING --park is on. Every finishing DURABLE session copies its \
1131             whole KV off the GPU on the tick thread before it retires — seconds and \
1132             gigabytes at agent context, and every other lane waits — then keeps an \
1133             artifact under <sessions>/park ({budget}). A /v1/* request mints a fresh \
1134             session and can never resume it, so its TERMINAL park is skipped; one \
1135             PREEMPTED under load still parks, because it is requeued under the same id \
1136             and adopts that artifact. This pays off only for sessions you resume by id \
1137             through the session API."
1138        );
1139    }
1140
1141    let model_name = pulled_names.get(&model).cloned().unwrap_or_else(|| (embedding.model_name)(&model, &Default::default()));
1142    if prefix_blob_budget_mb > 0 {
1143        std::env::set_var("BASERT_PREFIX_BLOB_BUDGET_MB", prefix_blob_budget_mb.to_string());
1144    }
1145    if std::env::var_os("BASERT_YARN").is_some() {
1146        eprintln!(
1147            "superfluid: WARNING: BASERT_YARN is set but has no effect on this path — the rope \
1148             kernels disable YaRN under paged/batched dispatch, which serving always uses. \
1149             Long requests run UNSCALED past the trained window."
1150        );
1151    }
1152
1153    let tui_on =
1154        cfg!(feature = "tui") && !no_tui && (tui_flag || std::io::IsTerminal::is_terminal(&std::io::stdout()));
1155    let tui_ring = if tui_on {
1156        Some(superfluid_daemon::telemetry::new_log_ring())
1157    } else {
1158        None
1159    };
1160    #[cfg(feature = "tui")]
1161    let stderr_capture = tui_ring.as_ref().and_then(|ring| {
1162        match superfluid_daemon::tui::capture_stderr(ring.clone()) {
1163            Ok(guard) => Some(guard),
1164            Err(e) => {
1165                eprintln!("superfluid: stderr capture unavailable, engine output may overdraw the TUI: {e}");
1166                None
1167            }
1168        }
1169    });
1170    #[cfg(feature = "tui")]
1171    let worker_stderr = stderr_capture.as_ref().map(|c| c.worker_stderr());
1172    #[cfg(not(feature = "tui"))]
1173    let worker_stderr: Option<std::sync::Arc<std::os::fd::OwnedFd>> = None;
1174    let log_dir_eff = log_dir.or(if tui_on { Some(sessions.join("logs")) } else { None });
1175    if let Some(url) = &otlp_metrics {
1176        if let Err(e) = superfluid_daemon::otlp::validate_http_url(url) {
1177            eprintln!("superfluid: --otlp-metrics: {e}");
1178            std::process::exit(2);
1179        }
1180        if otlp_interval_ms == 0 {
1181            eprintln!("superfluid: --otlp-interval-ms must be > 0");
1182            std::process::exit(2);
1183        }
1184    }
1185    if fleet_mode && otlp_endpoint.is_some() {
1186        eprintln!(
1187            "superfluid: --otlp-endpoint is not supported in fleet-head mode (the fleet head \
1188             has no scheduler to trace); set it on each node's superfluid instead"
1189        );
1190        std::process::exit(2);
1191    }
1192    let otlp = otlp_endpoint.map(|endpoint| {
1193        let mut cfg = otlp_cfg;
1194        cfg.endpoint = endpoint;
1195        if let Err(e) = cfg.validate() {
1196            eprintln!("superfluid: --otlp-endpoint: {e}");
1197            std::process::exit(2);
1198        }
1199        eprintln!(
1200            "superfluid: pushing operational traces to {} ({}an EGRESS: ids, lengths, timings and \
1201             codes only, never session content; spans: {})",
1202            cfg.traces_url(),
1203            if cfg.is_loopback() { "" } else { "NON-LOOPBACK, " },
1204            cfg.filter
1205        );
1206        cfg
1207    });
1208    use superfluid_daemon::telemetry::{LogConfig, LogSink};
1209    let sink = match log_dir_eff {
1210        Some(dir) => LogSink::File(LogConfig {
1211            dir,
1212            filter: log_filter,
1213            ..Default::default()
1214        }),
1215        None => LogSink::Stderr(log_filter),
1216    };
1217    if let Err(e) = superfluid_daemon::telemetry::init_sinks(sink, tui_ring.clone(), otlp) {
1218        eprintln!("superfluid: telemetry: {e}");
1219    }
1220    std::fs::create_dir_all(&sessions).expect("create sessions dir");
1221    // A fleet head keeps its sessions too: its handlers commit them as a local server's do.
1222    let _sessions_lock = lock_sessions_dir(&sessions);
1223    if sessions_cache {
1224        rotate_session_wal(&sessions);
1225        if let Ok(rd) = std::fs::read_dir(sessions.join("models")) {
1226            for dir in rd.filter_map(|e| e.ok()).map(|e| e.path()).filter(|d| d.is_dir()) {
1227                rotate_session_wal(&dir);
1228            }
1229        }
1230    }
1231    if !fleet_mode {
1232        settle_park_dir(&sessions.join("park"), park, park_budget_gb);
1233        if let Ok(rd) = std::fs::read_dir(sessions.join("models")) {
1234            let mut dirs: Vec<std::path::PathBuf> = rd
1235                .filter_map(|e| e.ok())
1236                .map(|e| e.path().join("park"))
1237                .filter(|d| d.is_dir())
1238                .collect();
1239            dirs.sort();
1240            for dir in dirs {
1241                settle_park_dir(&dir, park, park_budget_gb);
1242            }
1243        }
1244    }
1245
1246    if let Some(b) = kv_bits_arg {
1247        if ![0, 4, 8, 16, 84].contains(&b) {
1248            eprintln!("--kv-bits {b}: expected 0 (auto), 4 (q4_0), 8 (q8_0), 16 (f16) or 84 (K q8_0 / V q4_0)");
1249            std::process::exit(2);
1250        }
1251    }
1252    let kv_bits = match (park_lossy, kv_bits_arg) {
1253        (true, None) | (true, Some(16)) => 16,
1254        (true, Some(other)) => {
1255            eprintln!("--park-lossy requires F16 KV pools; --kv-bits {other} conflicts");
1256            std::process::exit(2);
1257        }
1258        (false, v) => v.unwrap_or(0),
1259    };
1260
1261    let dir_default = max_context.is_none() && !max_context_auto_flag && model_dir.is_some();
1262    // A fleet head holds no model; its window is the nodes', learned when it reaches them.
1263    let max_context_auto = max_context.is_none() && !dir_default && !fleet_mode;
1264    let sizing_lanes = if fim_model.is_some() { max_batch.max(fim_max_batch) } else { max_batch };
1265    // (the window each lane's share of the device holds, the model's
1266    // trained window)
1267    let mut worker_sized: Option<(u64, u64)> = None;
1268    let unsized_why: Option<(String, String)> = if max_context_auto {
1269        let sizing = runtime.id.base();
1270        let elsewhere = models.iter().chain(fim_model.iter()).find_map(|m| {
1271            let on = if fim_model.as_ref() == Some(m) { catalog.reader_of(m) } else { runtime_pick.resolve(m, &catalog) };
1272            on.ok().filter(|id| id.base() != sizing).map(|id| (m, id))
1273        });
1274        match elsewhere {
1275            Some((m, id)) => Some((
1276                format!(
1277                    "--max-context auto sizes the window for the {} runtime, and {} is served by {}",
1278                    sizing.name(),
1279                    m.display(),
1280                    id.base().name()
1281                ),
1282                format!(
1283                    "not sized for this device: the {} runtime sizes it, and {} is served by {}",
1284                    sizing.name(),
1285                    m.display(),
1286                    id.base().name()
1287                ),
1288            )),
1289            None if !superfluid_daemon::linked::sizes_context(sizing) => {
1290                let set: Vec<&PathBuf> = models.iter().chain(fim_model.iter()).collect();
1291                let sized = set
1292                    .iter()
1293                    .map(|m| superfluid_daemon::runtimes::sizing_of(&runtime, m))
1294                    .collect::<Result<Vec<_>, String>>()
1295                    .and_then(|each| {
1296                        let trained = each.iter().map(|s| s.trained_context).min().unwrap_or(0);
1297                        superfluid_adapter_kit::sizing::window(&each, sizing_lanes).map(|n| (n, trained)).ok_or_else(|| {
1298                            format!("the {} runtime reported no device budget to size a window against", sizing.name())
1299                        })
1300                    });
1301                match sized {
1302                    Ok(n) => {
1303                        worker_sized = Some(n);
1304                        None
1305                    }
1306                    Err(why) => Some((format!("--max-context auto: {why}"), why)),
1307                }
1308            }
1309            None => None,
1310        }
1311    } else {
1312        None
1313    };
1314    let max_context_auto = max_context_auto && unsized_why.is_none();
1315    let ctx = if let Some(n) = max_context {
1316        n
1317    } else if fleet_mode {
1318        8192
1319    } else if dir_default {
1320        eprintln!(
1321            "superfluid: context window 8192 tokens (--model-dir loads models on demand at this window, so it is not \
1322             sized to fill the device with the startup model; --max-context N|auto overrides)"
1323        );
1324        8192
1325    } else if let Some((asked, kept)) = unsized_why {
1326        if max_context_auto_flag {
1327            eprintln!("superfluid: {asked}: pass --max-context <tokens>");
1328            std::process::exit(2);
1329        }
1330        eprintln!("superfluid: context window 8192 tokens ({kept}; --max-context N pins it)");
1331        8192
1332    } else if let Some((per_lane, trained)) = worker_sized {
1333        // One conversation may grow past its share of the pool, up to the trained window.
1334        let whole = per_lane.saturating_mul(u64::from(sizing_lanes.max(1)));
1335        let n = i32::try_from(whole.min(trained.max(per_lane))).unwrap_or(i32::MAX);
1336        eprintln!(
1337            "superfluid: context window {n} tokens (auto: sized for this device, {per_lane} for each of {sizing_lanes} lanes{}{}, \
1338             one conversation up to the trained {trained}; --max-context N pins it)",
1339            if models.len() > 1 { format!(", {} models", models.len()) } else { String::new() },
1340            if fim_model.is_some() { ", FIM satellite included" } else { "" },
1341        );
1342        n
1343    } else {
1344        use superfluid_daemon::speculate_resolve::{resolve_all, sizing_path, SpecLoad};
1345        let ranked: Vec<Vec<String>> = models
1346            .iter()
1347            .map(|m| {
1348                if *m == model {
1349                    speculate_primary.iter().chain(speculate_fallbacks.iter()).cloned().collect()
1350                } else {
1351                    speculate.as_deref().and_then(|d| resolve_all(d, m).ok()).unwrap_or_default()
1352                }
1353            })
1354            .collect();
1355        let sizing_models: Vec<PathBuf> = models.iter().cloned().chain(fim_model.iter().cloned()).collect();
1356        let lanes = if fim_model.is_some() { max_batch.max(fim_max_batch) } else { max_batch } as i32;
1357        let speculator_entry = |i: usize, r: &str| -> Option<(PathBuf, bool, Option<usize>)> {
1358            let (path, load) = sizing_path(r, &models[i])?;
1359            Some((path, load == SpecLoad::EmbeddedHead, (load == SpecLoad::Sidecar).then_some(i)))
1360        };
1361        let speculators_for = |choice: &[Option<&String>]| -> Vec<(PathBuf, bool, Option<usize>)> {
1362            (0..models.len()).filter_map(|i| speculator_entry(i, choice[i].or(ranked[i].first())?)).collect()
1363        };
1364        let sizing = runtime.id.base();
1365        let size = |speculators: &[(PathBuf, bool, Option<usize>)]| {
1366            superfluid_daemon::linked::suggest_max_context(sizing, &sizing_models, speculators, lanes, kv_bits).unwrap_or_else(|e| {
1367                eprintln!("superfluid: {e}");
1368                std::process::exit(2);
1369            })
1370        };
1371        let first_picks: Vec<Option<&String>> = vec![None; models.len()];
1372        let n_spec = speculators_for(&first_picks).len();
1373        const MAX_COMBINATIONS: usize = 16;
1374        let counts: Vec<usize> = ranked.iter().map(|c| c.len().max(1)).collect();
1375        let few_enough = counts
1376            .iter()
1377            .try_fold(1usize, |p, &c| p.checked_mul(c).filter(|p| *p <= MAX_COMBINATIONS))
1378            .is_some();
1379        let (sized, note) = if few_enough {
1380            let mut best: Option<(i32, Vec<Option<&String>>)> = None;
1381            let mut idx = vec![0usize; models.len()];
1382            loop {
1383                let choice: Vec<Option<&String>> =
1384                    idx.iter().enumerate().map(|(i, &k)| (k > 0).then(|| &ranked[i][k])).collect();
1385                if let Some(n) = size(&speculators_for(&choice)) {
1386                    if best.as_ref().is_none_or(|(b, _)| n < *b) {
1387                        best = Some((n, choice));
1388                    }
1389                }
1390                let mut i = 0;
1391                while i < idx.len() {
1392                    idx[i] += 1;
1393                    if idx[i] < counts[i] {
1394                        break;
1395                    }
1396                    idx[i] = 0;
1397                    i += 1;
1398                }
1399                if i == idx.len() {
1400                    break;
1401                }
1402            }
1403            let fallbacks: Vec<&str> =
1404                best.iter().flat_map(|(_, c)| c.iter().flatten().map(|s| s.as_str())).collect();
1405            let note = if fallbacks.is_empty() {
1406                String::new()
1407            } else {
1408                format!(", sized for the fallback {}", fallbacks.join(" + "))
1409            };
1410            (best.map(|(n, _)| n), note)
1411        } else {
1412            let all: Vec<(PathBuf, bool, Option<usize>)> = (0..models.len())
1413                .flat_map(|i| ranked[i].iter().filter_map(move |r| speculator_entry(i, r)))
1414                .collect();
1415            let note = format!(", every one of {} drafter candidates charged at once", all.len());
1416            (size(&all), note)
1417        };
1418        match sized {
1419            Some(n) => {
1420                eprintln!(
1421                    "superfluid: context window {n} tokens (auto: sized for this device, {lanes} lanes{}{}{}; --max-context N pins it)",
1422                    if n_spec == 0 { String::new() } else { format!(", {n_spec} speculator(s)") },
1423                    if fim_model.is_some() { ", FIM satellite included" } else { "" },
1424                    note
1425                );
1426                n
1427            }
1428            None => {
1429                eprintln!(
1430                    "superfluid: context window 8192 tokens (auto could not size this device/model set; --max-context N pins it)"
1431                );
1432                8192
1433            }
1434        }
1435    }
1436    .max(512);
1437    if let Some(n) = max_context {
1438        eprintln!("superfluid: context window {ctx} tokens (--max-context {n})");
1439    }
1440    // A worker process starts loading now, while the tokenizer and chat
1441    // template are built: the two do not depend on each other, and the load
1442    // is most of startup.
1443    let spawns_worker = !runtime.in_process_capable() || (worker_process && runtime.worker.bin.is_file());
1444    let early_host = (!fleet_mode && spawns_worker).then(|| {
1445        let spec = superfluid_daemon::WorkerSpec {
1446            bin: runtime.worker.bin.clone(),
1447            args: runtime.serve_args(&model, ctx, max_batch, kv_bits),
1448            stderr: worker_stderr.clone(),
1449        };
1450        std::thread::spawn(move || EngineHost::spawn_process(spec))
1451    });
1452    let tokenizer = open_tokenizer(&runtime, &model).unwrap_or_else(|e| {
1453        eprintln!("superfluid: {e}");
1454        std::process::exit(2);
1455    });
1456    let codec: Box<dyn superfluid_daemon::TextCodec + Send + Sync> = match codec_for_dialect(&dialect, tokenizer) {
1457        Ok((c, kind)) => {
1458            if dialect == "auto" {
1459                match kind {
1460                    "template" => eprintln!("superfluid: serving via the model's own chat template"),
1461                    "raw" => eprintln!(
1462                        "superfluid: model has no chat template; serving raw (chat routes will refuse)"
1463                    ),
1464                    _ => {}
1465                }
1466            }
1467            c
1468        }
1469        Err(e) => {
1470            eprintln!("superfluid: {e}");
1471            std::process::exit(2);
1472        }
1473    };
1474
1475    let store = open_session_store(&sessions);
1476
1477    let workerd_bin = catalog.workerd().to_path_buf();
1478    let worker_process = if !runtime.in_process_capable() {
1479        if !worker_process {
1480            eprintln!(
1481                "superfluid: --no-worker-process: the {} runtime runs in its own worker ({})",
1482                runtime.id.name(),
1483                runtime.worker.bin.display()
1484            );
1485        }
1486        true
1487    } else if worker_process && !runtime.worker.bin.is_file() {
1488        if fim_model.is_some() {
1489            eprintln!(
1490                "superfluid: --fim-model needs superfluid-workerd (the satellite runs in its own \
1491                 worker), not found at {}",
1492                workerd_bin.display()
1493            );
1494            std::process::exit(1);
1495        }
1496        if models.len() > 1 {
1497            eprintln!(
1498                "superfluid: {} models need superfluid-workerd (one address space each), \
1499                 not found at {}",
1500                models.len(),
1501                workerd_bin.display()
1502            );
1503            std::process::exit(1);
1504        }
1505        eprintln!(
1506            "superfluid: superfluid-workerd not found at {}; running the engine in-process \
1507             (install the bundled workerd or pass --no-worker-process to silence)",
1508            workerd_bin.display()
1509        );
1510        false
1511    } else {
1512        worker_process
1513    };
1514    let model_display = model.display().to_string();
1515    let model_source = model.to_string_lossy().into_owned();
1516    let host = if worker_process {
1517        match early_host {
1518            Some(started) => started.join().expect("worker start thread"),
1519            None => EngineHost::spawn_process(superfluid_daemon::WorkerSpec {
1520                bin: runtime.worker.bin.clone(),
1521                args: runtime.serve_args(&model, ctx, max_batch, kv_bits),
1522                stderr: worker_stderr.clone(),
1523            }),
1524        }
1525        .unwrap_or_else(|e| {
1526            #[cfg(feature = "tui")]
1527            if let Some(c) = stderr_capture.as_ref() {
1528                c.drain(std::time::Duration::from_secs(2));
1529            }
1530            eprintln!(
1531                "superfluid: could not start the engine worker for '{model_display}': {e} \
1532                 (the worker's stderr above carries the engine's reason)"
1533            );
1534            std::process::exit(1);
1535        })
1536    } else {
1537        superfluid_daemon::linked::spawn(runtime.id, model, ctx, max_batch, kv_bits)
1538            .unwrap_or_else(|| Err(format!("this build does not link the {} runtime", runtime.id)))
1539            .unwrap_or_else(|e| {
1540                eprintln!("superfluid: could not load '{model_display}': {e}");
1541                std::process::exit(1);
1542            })
1543    };
1544
1545    let caps = host.capabilities().read().expect("capability record").clone();
1546    if let Err(e) = caps.check_startup_flags(park_lossy, kv_bits_arg, speculate.as_deref()) {
1547        eprintln!("superfluid: {e}");
1548        std::process::exit(2);
1549    }
1550    if prefix_blob_budget_mb > 0 {
1551        if let superfluid_daemon::capabilities::Cap::No(why) = caps.get("ops", "space_promote") {
1552            eprintln!("superfluid: --prefix-blob-budget-mb has no effect on runtime {}: {why}", runtime.id.name());
1553        }
1554    }
1555
1556    let listener = socket.as_ref().map(|socket| {
1557        let _ = std::fs::remove_file(socket);
1558        let listener = UnixListener::bind(socket).unwrap_or_else(|e| {
1559            eprintln!("superfluid: cannot listen on --socket {}: {e}", socket.display());
1560            std::process::exit(1);
1561        });
1562        eprintln!("superfluid: serving on {}", socket.display());
1563        listener
1564    });
1565    let park_dir = park.then(|| sessions.join("park"));
1566
1567    let codec: Box<dyn superfluid_daemon::TextCodec + Send + Sync> = match tool_envelope {
1568        Some(env) => {
1569            if let Some(why) = superfluid_daemon::codec::tool_envelope_override_conflict(&*codec, env) {
1570                eprintln!("superfluid: {why}");
1571                std::process::exit(2);
1572            }
1573            eprintln!("superfluid: tool-call envelope overridden to '{}'", env.name());
1574            Box::new(superfluid_daemon::codec::WithToolEnvelope::new(codec, env))
1575        }
1576        None => codec,
1577    };
1578    let daemon = Arc::new(
1579        Daemon::with_options(
1580            store,
1581            host,
1582            codec,
1583            superfluid_daemon::DaemonOptions {
1584                max_lanes: max_batch as usize,
1585                park_dir,
1586                park_lossy,
1587                park_budget_bytes: park_budget_gb.saturating_mul(1 << 30),
1588                tool_lease_ms,
1589                tick_decode_budget,
1590                tick_target_ms,
1591                runtime_id: Some(runtime.id.name()),
1592                prefill_budget,
1593                agent_starvation_ticks: starvation_ticks,
1594                class_lanes,
1595                pressure_high_pct: if pressure_high > 0 { pressure_high } else { 85 },
1596                pressure_low_pct: if pressure_low > 0 { pressure_low } else { 70 },
1597                pin_budget_pct,
1598                pressure_source: if os_pressure {
1599                    superfluid_daemon::pressure::PressureConfig::Os
1600                } else {
1601                    superfluid_daemon::pressure::PressureConfig::None
1602                },
1603                speculate: speculate_primary.clone(),
1604                speculate_auto: speculate.as_deref().map(str::trim) == Some("auto"),
1605                speculate_fallbacks: speculate_fallbacks.clone(),
1606                speculation: speculation.clone(),
1607                media_dir: Some(sessions.join("media")),
1608                files_max_bytes,
1609                files_expiry_secs: files_expiry,
1610                completion_deadline_ms,
1611                trace_scope: superfluid_daemon::otlp::TraceScope::model(&model_name),
1612                completion_bucket,
1613            },
1614        )
1615        .unwrap_or_else(|e| {
1616            #[cfg(feature = "tui")]
1617            if let Some(c) = stderr_capture.as_ref() {
1618                c.drain(std::time::Duration::from_secs(2));
1619            }
1620            eprintln!("superfluid: cannot serve '{model_display}': {e}");
1621            std::process::exit(2);
1622        }),
1623    );
1624    {
1625        let bits = daemon.sched_stats().kv_bits.load(std::sync::atomic::Ordering::Relaxed);
1626        match superfluid_daemon::scheduler::kv_bits_label(bits) {
1627            Some(label) if kv_bits != 0 && bits != kv_bits as u64 => {
1628                eprintln!("superfluid: kv cache {label} (--kv-bits {kv_bits} overridden by the engine, see above)")
1629            }
1630            Some(label) => eprintln!("superfluid: kv cache {label}"),
1631            None => {}
1632        }
1633    }
1634    let fim_satellite: Option<(String, Arc<Daemon>)> = fim_model.as_ref().map(|path| {
1635        let runtime = catalog.reader_of(path).and_then(|id| catalog.require(id)).unwrap_or_else(|e| {
1636            eprintln!("superfluid: --fim-model {}: {e}", path.display());
1637            std::process::exit(2);
1638        });
1639        let mut fim_ctx = fim_max_context.unwrap_or(ctx.min(8192)).max(512);
1640        if max_context_auto && fim_ctx > ctx {
1641            eprintln!(
1642                "superfluid: --fim-max-context {fim_ctx} is above the window sized for this device; \
1643                 the FIM satellite runs at {ctx}"
1644            );
1645            fim_ctx = ctx;
1646        }
1647        let sat = build_fim_satellite(&FimSatelliteCfg {
1648            model: path.clone(),
1649            runtime,
1650            sessions: sessions.clone(),
1651            ctx: fim_ctx,
1652            max_batch: fim_max_batch,
1653            kv_bits,
1654            pressure_high,
1655            pressure_low,
1656            os_pressure,
1657            completion_deadline_ms,
1658            worker_stderr: worker_stderr.clone(),
1659        })
1660        .unwrap_or_else(|e| {
1661            #[cfg(feature = "tui")]
1662            if let Some(c) = stderr_capture.as_ref() {
1663                c.drain(std::time::Duration::from_secs(2));
1664            }
1665            eprintln!("superfluid: could not start the --fim-model satellite '{}': {e}", path.display());
1666            std::process::exit(1);
1667        });
1668        let name = superfluid_daemon::registry::model_id_for(path);
1669        eprintln!("superfluid: FIM completions served by satellite '{name}'");
1670        (name, sat)
1671    });
1672    if let Some((name, sat)) = &fim_satellite {
1673        if let Err(e) = daemon.attach_fim_satellite(name.clone(), Arc::clone(sat)) {
1674            eprintln!(
1675                "superfluid: --fim-model {}: {e} (its tokenizer has no atomic <|fim_prefix|>/<|fim_suffix|>/<|fim_middle|> tokens)",
1676                fim_model.as_ref().map(|p| p.display().to_string()).unwrap_or_default()
1677            );
1678            std::process::exit(2);
1679        }
1680    }
1681    let state_names: StateNames = Arc::default();
1682    let registry = if worker_process {
1683        let cfg = ModelBuildCfg {
1684            model_name: embedding.model_name,
1685            base_sessions: sessions.clone(),
1686            state_names: Arc::clone(&state_names),
1687            catalog: Arc::clone(&catalog),
1688            pick: runtime_pick.clone(),
1689            dialect: dialect.clone(),
1690            ctx,
1691            max_batch,
1692            kv_bits,
1693            park,
1694            park_lossy,
1695            park_budget_gb,
1696            tool_lease_ms,
1697            tick_decode_budget,
1698            tick_target_ms,
1699            prefill_budget,
1700            starvation_ticks,
1701            class_lanes,
1702            os_pressure,
1703            pressure_high_pct: if pressure_high > 0 { pressure_high } else { 85 },
1704            pressure_low_pct: if pressure_low > 0 { pressure_low } else { 70 },
1705            pin_budget_pct,
1706            speculate: speculate.clone(),
1707            speculation: speculation.clone(),
1708            completion_deadline_ms,
1709            completion_bucket,
1710            fim_satellite: fim_satellite.clone(),
1711            worker_stderr: worker_stderr.clone(),
1712        };
1713        let reload_cfg = cfg.clone();
1714        let loader: superfluid_daemon::registry::ModelLoader =
1715            Box::new(move |id: &str, runtime| build_loaded_daemon(&cfg, std::path::Path::new(id), runtime));
1716        let registry = superfluid_daemon::registry::ModelRegistry::with_initial(model_name.clone(), Arc::clone(&daemon), loader);
1717        registry.remember_source_on(&model_name, &model_source, Some(runtime.id));
1718        let names = Arc::clone(&catalog);
1719        registry.set_runtime_names(Box::new(move |name: &str| {
1720            names.named(name).map_err(|why| format!("{why} ({})", names.here_described()))
1721        }));
1722        registry.set_reloader(Box::new(move |id: &str, runtime, retired: &Daemon| {
1723            reload_daemon(&reload_cfg, std::path::Path::new(id), runtime, retired)
1724        }));
1725        Arc::new(registry)
1726    } else {
1727        Arc::new(superfluid_daemon::registry::ModelRegistry::single(
1728            model_name.clone(),
1729            Arc::clone(&daemon),
1730        ))
1731    };
1732    {
1733        let (registry, name) = (Arc::downgrade(&registry), model_name.clone());
1734        daemon.on_retired(Box::new(move || {
1735            if let Some(registry) = registry.upgrade() {
1736                let _ = registry.resolve(Some(&name));
1737            }
1738        }));
1739    }
1740    let same = |p: &std::path::Path| p.canonicalize().unwrap_or_else(|_| p.to_path_buf());
1741    let mut taken: std::collections::HashSet<String> = std::collections::HashSet::from([model_name.clone()]);
1742    let mut given: std::collections::HashSet<PathBuf> = models.first().map(|m| same(m)).into_iter().collect();
1743    for extra in models.iter().skip(1) {
1744        if !given.insert(same(extra)) {
1745            eprintln!("superfluid: model '{}' already loaded, skipping", (embedding.model_name)(extra, &Default::default()));
1746            continue;
1747        }
1748        let name = match pulled_names.get(extra) {
1749            Some(name) => name.clone(),
1750            None => {
1751                let name = (embedding.model_name)(extra, &taken);
1752                // The worker names the model again with nothing taken, so a
1753                // name the collision chose is recorded for it to find.
1754                if name != (embedding.model_name)(extra, &Default::default()) {
1755                    eprintln!("superfluid: {} is registered as '{name}' (another model took its stem)", extra.display());
1756                    state_names.lock().expect("state names").insert(extra.clone(), name.clone());
1757                }
1758                name
1759            }
1760        };
1761        taken.insert(name.clone());
1762        match registry.load_as(&name, &extra.to_string_lossy()) {
1763            Ok(true) => eprintln!("superfluid: loaded model '{name}'"),
1764            Ok(false) => eprintln!("superfluid: model '{name}' already loaded, skipping"),
1765            Err(e) => {
1766                #[cfg(feature = "tui")]
1767                if let Some(c) = stderr_capture.as_ref() {
1768                    c.drain(std::time::Duration::from_secs(2));
1769                }
1770                eprintln!("superfluid: could not load '{}': {e}", extra.display());
1771                std::process::exit(1);
1772            }
1773        }
1774    }
1775    if let Some(dir) = &model_dir {
1776        let mut n = 0;
1777        let here = catalog.survey();
1778        match artifacts_under(dir, &here, MODEL_DIR_DEPTH) {
1779            Ok(paths) => {
1780                for path in paths {
1781                    let refused = runtime_pick
1782                        .resolve(&path, &catalog)
1783                        .and_then(|rt| catalog.require(rt))
1784                        .and_then(|rt| refuse_statically(&rt, park_lossy, kv_bits_arg, speculate.as_deref()));
1785                    if let Err(e) = refused {
1786                        eprintln!("superfluid: --model-dir: skipping {}: {e}", path.display());
1787                        continue;
1788                    }
1789                    if !given.insert(same(&path)) {
1790                        continue;
1791                    }
1792                    let name = (embedding.model_name)(&path, &taken);
1793                    if registry.is_loaded(&name) {
1794                        taken.insert(name);
1795                        continue;
1796                    }
1797                    if name != (embedding.model_name)(&path, &Default::default()) {
1798                        eprintln!(
1799                            "superfluid: --model-dir: {} is registered as '{name}' (another artifact took its stem)",
1800                            path.display()
1801                        );
1802                        state_names.lock().expect("state names").insert(path.clone(), name.clone());
1803                    }
1804                    registry.register_known(&name, &path);
1805                    taken.insert(name);
1806                    n += 1;
1807                }
1808            }
1809            Err(e) => {
1810                eprintln!("superfluid: --model-dir {}: {e}", dir.display());
1811                std::process::exit(1);
1812            }
1813        }
1814        eprintln!("superfluid: {n} model(s) known from {} (loaded on first use)", dir.display());
1815    }
1816    if idle_timeout > 0 {
1817        registry.track_idle();
1818        let reg = Arc::clone(&registry);
1819        let max_idle = std::time::Duration::from_secs(idle_timeout);
1820        let tick = std::time::Duration::from_secs((idle_timeout / 4).max(1));
1821        std::thread::spawn(move || loop {
1822            std::thread::sleep(tick);
1823            for name in reg.sweep_idle(max_idle) {
1824                eprintln!("superfluid: unloaded idle model '{name}'");
1825            }
1826        });
1827    }
1828    if files_expiry.is_some() && files_sweep > 0 {
1829        let d = Arc::clone(&daemon);
1830        std::thread::spawn(move || loop {
1831            std::thread::sleep(std::time::Duration::from_secs(files_sweep));
1832            let n = d.files().sweep();
1833            if n > 0 {
1834                eprintln!("superfluid: files sweep removed {n} expired file(s)");
1835            }
1836        });
1837    }
1838    let mut http_thread = None;
1839    if let Some(addr) = http {
1840        let http_listener = listen_or_exit("--http", &addr);
1841        eprintln!("superfluid: openai api on http://{addr}{}", embedding.http.also_served());
1842        superfluid_daemon::selfupdate::check_in_background(catalog.home().to_path_buf(), pull_flags.offline);
1843        if nonstream_keepalive > 0 {
1844            eprintln!(
1845                "superfluid: non-streaming keep-alive on: 200 at admission, then a space every \
1846                 {nonstream_keepalive}s until the JSON body (x-superfluid-keepalive: whitespace)"
1847            );
1848        }
1849        let reg = Arc::clone(&registry);
1850        let serve_cfg = superfluid_daemon::openai::ServeConfig {
1851            api_key: api_key.clone(),
1852            rate_limit_per_minute: rate_limit,
1853            drain_timeout_secs: drain_timeout,
1854            default_max_tokens,
1855            max_context: ctx as u32,
1856            sampling: superfluid_daemon::openai::SamplingOverrides {
1857                temperature: op_temperature,
1858                top_p: op_top_p,
1859                top_k: op_top_k,
1860                min_p: op_min_p,
1861                repeat_penalty: op_repeat_penalty,
1862                spec_max_temperature: op_spec_max_temperature,
1863                fallback: embedding.sampling_fallback,
1864            },
1865            qos: superfluid_daemon::openai::HttpQosPolicy {
1866                default: http_default_qos,
1867                honor_headers: http_qos_header,
1868                allow_batch_invariant: http_allow_batch_invariant,
1869            },
1870            keys: key_policy.clone(),
1871            nonstream_keepalive_secs: nonstream_keepalive,
1872            surface: embedding.http,
1873            model_naming: embedding.model_naming,
1874            identity: embedding.identity.clone(),
1875            idempotency_dir: Some(sessions.join("idempotency")),
1876        };
1877        http_thread = Some(std::thread::spawn(move || {
1878            superfluid_daemon::openai::serve_blocking_config(http_listener, reg, serve_cfg)
1879        }));
1880    }
1881    if let Some(addr) = web {
1882        if !addr.starts_with("127.0.0.1:") && !addr.starts_with("[::1]:") && !addr.starts_with("localhost:") {
1883            eprintln!("superfluid: --web must bind loopback (127.0.0.1/::1)");
1884            std::process::exit(2);
1885        }
1886        let token_path = sessions.join("web-token");
1887        let token = superfluid_daemon::apiweb::mint_token(&token_path).expect("mint web token");
1888        let web_listener = listen_or_exit("--web", &addr);
1889        eprintln!("superfluid: api-web on http://{addr} (token: {})", token_path.display());
1890        let d = Arc::clone(&daemon);
1891        std::thread::spawn(move || {
1892            let _ = superfluid_daemon::apiweb::serve_blocking(web_listener, d, token, web_origins);
1893        });
1894    }
1895    let _otlp_pusher = otlp_metrics.map(|url| {
1896        eprintln!(
1897            "superfluid: pushing metrics to {url}/v1/metrics every {otlp_interval_ms}ms (an EGRESS)"
1898        );
1899        superfluid_daemon::export::MetricsPusher::start(
1900            url,
1901            std::time::Duration::from_millis(otlp_interval_ms),
1902            daemon.sched_stats(),
1903        )
1904    });
1905    #[cfg(feature = "tui")]
1906    if let Some(ring) = tui_ring {
1907        if let Some(listener) = listener {
1908            let serve_daemon = Arc::clone(&daemon);
1909            std::thread::spawn(move || {
1910                let _ = api::serve(listener, serve_daemon);
1911            });
1912        }
1913        let node = superfluid_daemon::tui::NodeInfo {
1914            identity: hostname(),
1915            model: model_name.clone(),
1916            engine: if worker_process { "native (worker)".into() } else { "native".into() },
1917            max_lanes: max_batch,
1918        };
1919        superfluid_daemon::tui::run(daemon, ring, node, stderr_capture.as_ref()).expect("tui");
1920        drop(stderr_capture);
1921        return;
1922    }
1923    let Some(listener) = listener else {
1924        // No session socket: the HTTP API is the server, and the process
1925        // lasts as long as it does.
1926        match http_thread.map(std::thread::JoinHandle::join) {
1927            Some(Ok(Ok(()))) => return,
1928            Some(Ok(Err(e))) => eprintln!("superfluid: the HTTP API stopped: {e}"),
1929            Some(Err(_)) => eprintln!("superfluid: the HTTP API stopped: its thread panicked"),
1930            None => unreachable!("without a session socket --no-http is refused"),
1931        }
1932        std::process::exit(1);
1933    };
1934    api::serve(listener, daemon).expect("serve");
1935}
1936
1937#[cfg(feature = "tui")]
1938fn hostname() -> String {
1939    let mut buf = [0u8; 256];
1940    // SAFETY: `buf` is a valid 256-byte writable buffer and `buf.len()` is its
1941    // exact capacity; gethostname writes at most that many bytes and NUL-
1942    // terminates within it. We only read `buf` after checking rc == 0.
1943    let rc = unsafe { libc::gethostname(buf.as_mut_ptr().cast::<libc::c_char>(), buf.len()) };
1944    if rc == 0 {
1945        let end = buf.iter().position(|&b| b == 0).unwrap_or(buf.len());
1946        String::from_utf8_lossy(&buf[..end]).into_owned()
1947    } else {
1948        "local".into()
1949    }
1950}
1951
1952type StateNames = Arc<std::sync::Mutex<std::collections::HashMap<PathBuf, String>>>;
1953
1954#[derive(Clone)]
1955struct ModelBuildCfg {
1956    model_name: fn(&Path, &std::collections::HashSet<String>) -> String,
1957    base_sessions: std::path::PathBuf,
1958    state_names: StateNames,
1959    catalog: Arc<superfluid_daemon::runtimes::Catalog>,
1960    pick: RuntimePick,
1961    dialect: String,
1962    ctx: i32,
1963    max_batch: u32,
1964    kv_bits: i32,
1965    park: bool,
1966    park_lossy: bool,
1967    park_budget_gb: u64,
1968    tool_lease_ms: u64,
1969    tick_decode_budget: u32,
1970    tick_target_ms: u32,
1971    prefill_budget: u32,
1972    starvation_ticks: u32,
1973    class_lanes: [usize; 4],
1974    os_pressure: bool,
1975    pressure_high_pct: u8,
1976    pressure_low_pct: u8,
1977    pin_budget_pct: u8,
1978    speculate: Option<String>,
1979    speculation: superfluid_daemon::SpeculationOptions,
1980    completion_deadline_ms: u64,
1981    completion_bucket: superfluid_daemon::completion_bucket::BucketConfig,
1982    fim_satellite: Option<(String, Arc<Daemon>)>,
1983    worker_stderr: Option<std::sync::Arc<std::os::fd::OwnedFd>>,
1984}
1985
1986struct FimSatelliteCfg {
1987    model: PathBuf,
1988    runtime: superfluid_daemon::runtimes::Runtime,
1989    sessions: PathBuf,
1990    ctx: i32,
1991    max_batch: u32,
1992    kv_bits: i32,
1993    pressure_high: u8,
1994    pressure_low: u8,
1995    os_pressure: bool,
1996    completion_deadline_ms: u64,
1997    worker_stderr: Option<std::sync::Arc<std::os::fd::OwnedFd>>,
1998}
1999
2000fn build_fim_satellite(cfg: &FimSatelliteCfg) -> Result<Arc<Daemon>, superfluid_daemon::DaemonError> {
2001    use superfluid_daemon::DaemonError;
2002    let tok = open_tokenizer(&cfg.runtime, &cfg.model)
2003        .map_err(|e| DaemonError::Generation(format!("satellite tokenizer load failed: {e}")))?;
2004    let codec = superfluid_daemon::BundleCodec::from_tokenizer(tok);
2005    let host = EngineHost::spawn_process(superfluid_daemon::WorkerSpec {
2006        bin: cfg.runtime.worker.bin.clone(),
2007        args: cfg.runtime.serve_args(&cfg.model, cfg.ctx, cfg.max_batch, cfg.kv_bits),
2008        stderr: cfg.worker_stderr.clone(),
2009    })?;
2010    let daemon = Daemon::with_options(
2011        SessionStore::ephemeral(),
2012        host,
2013        Box::new(codec),
2014        superfluid_daemon::DaemonOptions {
2015            max_lanes: cfg.max_batch as usize,
2016            park_dir: None,
2017            pressure_high_pct: if cfg.pressure_high > 0 { cfg.pressure_high } else { 85 },
2018            pressure_low_pct: if cfg.pressure_low > 0 { cfg.pressure_low } else { 70 },
2019            pressure_source: if cfg.os_pressure {
2020                superfluid_daemon::pressure::PressureConfig::Os
2021            } else {
2022                superfluid_daemon::pressure::PressureConfig::None
2023            },
2024            media_dir: Some(cfg.sessions.join("fim").join("media")),
2025            completion_deadline_ms: cfg.completion_deadline_ms,
2026            trace_scope: superfluid_daemon::otlp::TraceScope::model(
2027                &superfluid_daemon::registry::model_id_for(&cfg.model),
2028            ),
2029            ..Default::default()
2030        },
2031    )?;
2032    Ok(Arc::new(daemon))
2033}
2034
2035fn build_codec_for(
2036    dialect: &str,
2037    runtime: &superfluid_daemon::runtimes::Runtime,
2038    model: &std::path::Path,
2039) -> Result<Box<dyn superfluid_daemon::TextCodec + Send + Sync>, superfluid_daemon::DaemonError> {
2040    use superfluid_daemon::DaemonError;
2041    let tok = open_tokenizer(runtime, model).map_err(DaemonError::Generation)?;
2042    let (codec, _) = codec_for_dialect(dialect, tok).map_err(DaemonError::Generation)?;
2043    Ok(codec)
2044}
2045
2046fn refuse_statically(
2047    runtime: &superfluid_daemon::runtimes::Runtime,
2048    park_lossy: bool,
2049    kv_bits: Option<i32>,
2050    speculate: Option<&str>,
2051) -> Result<(), String> {
2052    runtime.static_capabilities().check_startup_flags(park_lossy, kv_bits, speculate)
2053}
2054
2055fn describe_worker(runtime: &superfluid_daemon::runtimes::Runtime) -> String {
2056    let version = runtime.version.as_deref().map(|v| format!("{v}, ")).unwrap_or_default();
2057    format!("{version}{}: {}", runtime.worker.source.name(), runtime.worker.bin.display())
2058}
2059
2060/// The digest of the tokenizer `model` is read with on `runtime`, as a fleet head and its nodes
2061/// compare them; none when no tokenizer opens here.
2062pub fn tokenizer_digest(runtime: &superfluid_daemon::runtimes::Runtime, model: &std::path::Path) -> Option<u64> {
2063    open_tokenizer(runtime, model).ok().map(|t| superfluid_daemon::fleet::tokenizer_digest(&*t))
2064}
2065
2066pub fn open_tokenizer(
2067    runtime: &superfluid_daemon::runtimes::Runtime,
2068    model: &std::path::Path,
2069) -> Result<Arc<dyn superfluid_engine::Tokenizer>, String> {
2070    use superfluid_engine::artifact::TokenizerSource;
2071    if let Some(why) = not_there(model) {
2072        return Err(why);
2073    }
2074    if let Some(lib) = &runtime.tokenizer {
2075        let tok = superfluid_daemon::dylib_tokenizer::DylibTokenizer::open(lib, model)?;
2076        return Ok(Arc::new(tok));
2077    }
2078    if runtime.info.tokenizer == Some(TokenizerSource::HuggingFace) {
2079        let tok = superfluid_tokenizer_hf::HfTokenizer::load(model)
2080            .map_err(|e| format!("tokenizer load failed for {}: {e}", model.display()))?;
2081        return Ok(Arc::new(tok));
2082    }
2083    superfluid_daemon::linked::tokenizer(runtime.id, model).unwrap_or_else(|| {
2084        Err(format!(
2085            "no tokenizer for {} here: the {} runtime ships none beside its worker, and this superfluid does not link it",
2086            model.display(),
2087            runtime.id
2088        ))
2089    })
2090}
2091
2092/// How deep `--model-dir` looks: a hub cache nests a model three levels
2093/// down (`<org>/<repo>/<variant>/model.base`).
2094const MODEL_DIR_DEPTH: usize = 8;
2095
2096/// The artifacts under `dir` that a runtime here reads, in name order. An
2097/// entry that one reads is taken whole (an MLX model is a directory); a
2098/// directory that none reads is looked into.
2099fn artifacts_under(
2100    dir: &std::path::Path,
2101    here: &[superfluid_daemon::runtimes::Found],
2102    depth: usize,
2103) -> std::io::Result<Vec<PathBuf>> {
2104    let mut paths: Vec<PathBuf> = std::fs::read_dir(dir)?.flatten().map(|e| e.path()).collect();
2105    paths.sort();
2106    let mut found = Vec::new();
2107    for path in paths {
2108        if !superfluid_daemon::runtimes::readers_in(here, &path).is_empty() {
2109            found.push(path);
2110        } else if depth > 0 && path.is_dir() {
2111            found.extend(artifacts_under(&path, here, depth - 1).unwrap_or_default());
2112        }
2113    }
2114    Ok(found)
2115}
2116
2117fn not_there(model: &std::path::Path) -> Option<String> {
2118    (!model.exists()).then(|| format!("{}: no such file or directory", model.display()))
2119}
2120
2121fn codec_for_dialect(
2122    dialect: &str,
2123    tok: Arc<dyn superfluid_engine::Tokenizer>,
2124) -> Result<(Box<dyn superfluid_daemon::TextCodec + Send + Sync>, &'static str), String> {
2125    Ok(match dialect {
2126        "chatml" => (
2127            Box::new(
2128                superfluid_daemon::codec::chatml_codec_from_tokenizer(tok)
2129                    .map_err(|e| format!("--dialect chatml refused: {e}"))?,
2130            ),
2131            "chatml",
2132        ),
2133        "atem" => (
2134            Box::new(
2135                superfluid_daemon::atem::AtemCodec::from_tokenizer(tok)
2136                    .map_err(|_| "--dialect atem refused: the tokenizer lacks the ATEM markers".to_string())?,
2137            ),
2138            "atem",
2139        ),
2140        "raw" => (Box::new(superfluid_daemon::BundleCodec::from_tokenizer(tok)), "raw"),
2141        "template" => (
2142            Box::new(
2143                superfluid_daemon::template_codec::TemplateCodec::from_tokenizer(tok)
2144                    .map_err(|_| "--dialect template refused: the model carries no usable chat template".to_string())?,
2145            ),
2146            "template",
2147        ),
2148        "auto" => superfluid_daemon::codec::auto_codec_from_tokenizer(tok),
2149        other => return Err(format!("unknown dialect '{other}' (chatml|atem|raw|template|auto)")),
2150    })
2151}
2152
2153fn speculation_for<'a>(
2154    directive: Option<&'a str>,
2155    caps: &superfluid_daemon::capabilities::Capabilities,
2156    model: &std::path::Path,
2157    runtime: &str,
2158) -> Option<&'a str> {
2159    let d = directive?;
2160    if d.trim() != "auto" {
2161        return Some(d);
2162    }
2163    match caps.declines_auto_speculation() {
2164        Some(why) => {
2165            eprintln!("superfluid: speculation: auto: {} on the {runtime} runtime verifies no drafts ({why}) — decoding plainly", model.display());
2166            None
2167        }
2168        // The other runtimes draft from the prompt alone: there is no
2169        // drafter artifact for `auto` to look for.
2170        None if runtime != "basert" => {
2171            eprintln!("superfluid: speculation: auto: prompt-lookup on the {runtime} runtime");
2172            Some("prompt-lookup")
2173        }
2174        None => Some(d),
2175    }
2176}
2177
2178fn unsupported_runtime(param: Option<&'static str>) -> impl Fn(String) -> superfluid_daemon::DaemonError {
2179    move |m: String| superfluid_daemon::DaemonError::Unsupported(superfluid_daemon::Refusal::new("unsupported_runtime", param, m))
2180}
2181
2182fn runtime_for(
2183    cfg: &ModelBuildCfg,
2184    model_path: &std::path::Path,
2185    requested: Option<RuntimeId>,
2186) -> Result<superfluid_daemon::runtimes::Runtime, superfluid_daemon::DaemonError> {
2187    let runtime = cfg
2188        .pick
2189        .resolve_with(model_path, requested, &cfg.catalog)
2190        .and_then(|rt| cfg.catalog.require(rt))
2191        .map_err(unsupported_runtime(Some(if requested.is_some() { "runtime" } else { "model" })))?;
2192    refuse_statically(&runtime, cfg.park_lossy, Some(cfg.kv_bits), cfg.speculate.as_deref()).map_err(unsupported_runtime(None))?;
2193    Ok(runtime)
2194}
2195
2196fn engine_on(
2197    cfg: &ModelBuildCfg,
2198    runtime: &superfluid_daemon::runtimes::Runtime,
2199    model_path: &std::path::Path,
2200) -> Result<(Box<dyn superfluid_daemon::TextCodec + Send + Sync>, EngineHost), superfluid_daemon::DaemonError> {
2201    use superfluid_daemon::DaemonError;
2202    let codec = build_codec_for(&cfg.dialect, runtime, model_path)?;
2203    let host = EngineHost::spawn_process(superfluid_daemon::WorkerSpec {
2204        bin: runtime.worker.bin.clone(),
2205        args: runtime.serve_args(model_path, cfg.ctx.max(512), cfg.max_batch, cfg.kv_bits),
2206        stderr: cfg.worker_stderr.clone(),
2207    })
2208    .map_err(|_| DaemonError::Generation("engine worker spawn failed".to_string()))?;
2209    host.capabilities()
2210        .read()
2211        .expect("capability record")
2212        .check_startup_flags(cfg.park_lossy, Some(cfg.kv_bits), cfg.speculate.as_deref())
2213        .map_err(unsupported_runtime(None))?;
2214    Ok((codec, host))
2215}
2216
2217fn reload_daemon(
2218    cfg: &ModelBuildCfg,
2219    model_path: &std::path::Path,
2220    requested: Option<RuntimeId>,
2221    retired: &Daemon,
2222) -> Result<(), superfluid_daemon::DaemonError> {
2223    retired.reload_with(|| {
2224        let runtime = runtime_for(cfg, model_path, requested)?;
2225        let (codec, host) = engine_on(cfg, &runtime, model_path)?;
2226        Ok((host, codec))
2227    })
2228}
2229
2230fn build_loaded_daemon(
2231    cfg: &ModelBuildCfg,
2232    model_path: &std::path::Path,
2233    requested: Option<RuntimeId>,
2234) -> Result<Arc<Daemon>, superfluid_daemon::DaemonError> {
2235    use superfluid_daemon::DaemonError;
2236    let id: String = cfg
2237        .state_names
2238        .lock()
2239        .expect("state names")
2240        .get(model_path)
2241        .cloned()
2242        .unwrap_or_else(|| (cfg.model_name)(model_path, &Default::default()));
2243    let mdir = superfluid_daemon::registry::state_dir(&cfg.base_sessions.join("models"), &id)?;
2244    let runtime = runtime_for(cfg, model_path, requested)?;
2245    settle_park_dir(&mdir.join("park"), cfg.park, cfg.park_budget_gb);
2246    let store = SessionStore::open(&mdir.join("wal.log"))?;
2247    let (codec, host) = engine_on(cfg, &runtime, model_path)?;
2248    let park_dir = cfg.park.then(|| mdir.join("park"));
2249    let caps = host.capabilities().read().expect("capability record").clone();
2250    let mut speculate_ranked = match speculation_for(cfg.speculate.as_deref(), &caps, model_path, runtime.id.name()) {
2251        Some(d) => superfluid_daemon::speculate_resolve::resolve_all(d, model_path).map_err(DaemonError::Generation)?,
2252        None => Vec::new(),
2253    };
2254    let speculate_first = (!speculate_ranked.is_empty()).then(|| speculate_ranked.remove(0));
2255    let daemon = Daemon::with_options(
2256        store,
2257        host,
2258        codec,
2259        superfluid_daemon::DaemonOptions {
2260            max_lanes: cfg.max_batch as usize,
2261            park_dir,
2262            park_lossy: cfg.park_lossy,
2263            park_budget_bytes: cfg.park_budget_gb.saturating_mul(1 << 30),
2264            tool_lease_ms: cfg.tool_lease_ms,
2265            tick_decode_budget: cfg.tick_decode_budget,
2266            tick_target_ms: cfg.tick_target_ms,
2267            prefill_budget: cfg.prefill_budget,
2268            agent_starvation_ticks: cfg.starvation_ticks,
2269            class_lanes: cfg.class_lanes,
2270            pressure_high_pct: cfg.pressure_high_pct,
2271            pressure_low_pct: cfg.pressure_low_pct,
2272            pin_budget_pct: cfg.pin_budget_pct,
2273            pressure_source: if cfg.os_pressure {
2274                superfluid_daemon::pressure::PressureConfig::Os
2275            } else {
2276                superfluid_daemon::pressure::PressureConfig::None
2277            },
2278            speculate: speculate_first,
2279            speculate_auto: cfg.speculate.as_deref().map(str::trim) == Some("auto"),
2280            speculate_fallbacks: speculate_ranked,
2281            speculation: cfg.speculation.clone(),
2282            media_dir: Some(mdir.join("media")),
2283            completion_deadline_ms: cfg.completion_deadline_ms,
2284            trace_scope: superfluid_daemon::otlp::TraceScope::model(&id),
2285            completion_bucket: cfg.completion_bucket,
2286            runtime_id: Some(runtime.id.name()),
2287            ..Default::default()
2288        },
2289    )?;
2290    if let Some((name, sat)) = &cfg.fim_satellite {
2291        daemon.attach_fim_satellite(name.clone(), Arc::clone(sat))?;
2292    }
2293    Ok(Arc::new(daemon))
2294}
2295
2296fn lock_sessions_dir(sessions: &std::path::Path) -> std::fs::File {
2297    use std::os::fd::AsRawFd;
2298    let path = sessions.join("serve.lock");
2299    let file = match std::fs::OpenOptions::new().read(true).write(true).create(true).truncate(false).open(&path) {
2300        Ok(file) => file,
2301        Err(e) => {
2302            eprintln!("superfluid: could not open {}: {e}", path.display());
2303            std::process::exit(1);
2304        }
2305    };
2306    loop {
2307        // SAFETY: `file` is an open descriptor for the duration of the call.
2308        let rc = unsafe { libc::flock(file.as_raw_fd(), libc::LOCK_EX | libc::LOCK_NB) };
2309        if rc == 0 {
2310            return file;
2311        }
2312        let err = std::io::Error::last_os_error();
2313        match err.kind() {
2314            std::io::ErrorKind::Interrupted => continue,
2315            std::io::ErrorKind::WouldBlock => {
2316                eprintln!(
2317                    "superfluid: another server is already using the sessions directory {}. \
2318                     Stop it, or give this one its own --sessions directory.",
2319                    sessions.display()
2320                );
2321                std::process::exit(1);
2322            }
2323            _ => {
2324                eprintln!("superfluid: could not lock {}: {err}", path.display());
2325                std::process::exit(1);
2326            }
2327        }
2328    }
2329}
2330
2331fn rotate_session_wal(dir: &std::path::Path) {
2332    use superfluid_daemon::store_id::store_id_path;
2333    let wal = dir.join("wal.log");
2334    let prev = dir.join("wal.log.prev");
2335    let keep = std::fs::metadata(&wal).is_ok_and(|m| m.len() > 0);
2336    let aside = |from: &std::path::Path, to: Option<&std::path::Path>| -> bool {
2337        if !from.exists() {
2338            return false;
2339        }
2340        if to.is_some_and(|to| std::fs::rename(from, to).is_ok()) {
2341            return true;
2342        }
2343        if let Err(e) = std::fs::remove_file(from) {
2344            eprintln!(
2345                "superfluid: --sessions-cache could not set aside or remove {} ({e}); \
2346                 remove it, or run without --sessions-cache.",
2347                from.display()
2348            );
2349            std::process::exit(1);
2350        }
2351        false
2352    };
2353    if keep {
2354        let _ = std::fs::remove_file(store_id_path(&prev));
2355    }
2356    if keep && aside(&wal, Some(&prev)) {
2357        aside(&store_id_path(&wal), Some(&store_id_path(&prev)));
2358    } else {
2359        aside(&wal, None);
2360        aside(&store_id_path(&wal), None);
2361    }
2362}
2363#[cfg(test)]
2364mod tests {
2365    use super::*;
2366    use superfluid_daemon::runtimes::{Found, Info};
2367    use superfluid_engine::artifact::Format;
2368
2369    fn reads(id: &'static str, format: Format) -> Found {
2370        let info = Info { aliases: Vec::new(), formats: vec![format], tokenizer: None, named: None };
2371        Found { id: RuntimeId::new(id), worker: None, info: Some(info), usable: Err("not started".into()) }
2372    }
2373
2374    #[test]
2375    fn model_dir_looks_into_a_hub_cache_and_takes_a_directory_model_whole() {
2376        let root = std::env::temp_dir().join(format!("superfluid-model-dir-{}", std::process::id()));
2377        let _ = std::fs::remove_dir_all(&root);
2378        let file = |rel: &str| {
2379            let p = root.join(rel);
2380            std::fs::create_dir_all(p.parent().unwrap()).unwrap();
2381            std::fs::write(&p, b"BASE....").unwrap();
2382            p
2383        };
2384        let flat = file("Flat-Q4.base");
2385        let q4 = file("org/Model/default-q4/model.base");
2386        file("org/Model/default-q4/hub.json");
2387        let q8 = file("org/Model/default-q8/model.base");
2388        file("notes.txt");
2389        file("mlx-model/config.json");
2390        file("mlx-model/weights.safetensors");
2391        file("mlx-model/stray.base");
2392        let here = [
2393            reads("basert", Format::file("base", "a .base bundle", &["base"], None)),
2394            reads("mlx", Format::directory("mlx", "an MLX model", &["config.json"], &["safetensors"])),
2395        ];
2396        let found = artifacts_under(&root, &here, MODEL_DIR_DEPTH).unwrap();
2397        assert_eq!(found, vec![flat, root.join("mlx-model"), q4, q8]);
2398        assert!(artifacts_under(&root, &here, 0).unwrap().len() == 2, "depth 0 reads only the top level");
2399    }
2400}