1use 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
101fn 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#[derive(Debug, Clone)]
140pub struct Embedding {
141 pub http: superfluid_daemon::openai::HttpSurface,
143 pub model_naming: superfluid_daemon::openai::ModelNaming,
145 pub model_name: fn(&Path, &std::collections::HashSet<String>) -> String,
149 pub sampling_fallback: Option<superfluid_daemon::openai::SamplingFallback>,
151 pub identity: superfluid_daemon::openai::ServerIdentity,
153 pub session_socket: bool,
156 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
175pub 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 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 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 }
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 superfluid_daemon::runtime_options::set(superfluid_daemon::runtime_options::given_pairs(&runtime_options));
597 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 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 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 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 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 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 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 ®istry {
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 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 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 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 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 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 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 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 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 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(®istry), 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 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(®istry);
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(®istry);
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 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 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
2060pub 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
2092const MODEL_DIR_DEPTH: usize = 8;
2095
2096fn 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 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 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}