phago-cli 1.0.0

Command-line interface for Phago biological computing framework
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72

//! Query the knowledge graph.

use anyhow::{bail, Result};
use colored::Colorize;
use phago::prelude::*;

use crate::config::current_session_path;

pub fn run(query: &str, max_results: usize, alpha: f64) -> Result<()> {
    let session_path = current_session_path()?;

    if !session_path.exists() {
        bail!(
            "No session found. Run {} first.",
            "phago ingest".cyan()
        );
    }

    // Load session
    let state = load_session(&session_path)?;
    let mut colony = Colony::new();
    restore_into_colony(&mut colony, &state);

    // Run hybrid query
    let config = HybridConfig {
        alpha,
        max_results,
        candidate_multiplier: 3,
    };

    let results = hybrid_query(&colony, query, &config);

    if results.is_empty() {
        println!("{} No results found for: {}", "•".yellow(), query.cyan());
        return Ok(());
    }

    println!(
        "{} Results for {} (alpha={}):",
        "→".blue(),
        query.cyan().bold(),
        alpha
    );
    println!();

    for (i, result) in results.iter().enumerate() {
        let rank = format!("{}.", i + 1);
        let score = format!("{:.3}", result.final_score);

        println!(
            "  {} {} {}",
            rank.blue(),
            result.label.white().bold(),
            format!("({})", score).dimmed()
        );

        // Show score breakdown
        println!(
            "      TF-IDF: {:.3}  Graph: {:.3}",
            result.tfidf_score, result.graph_score
        );
    }

    println!();
    println!(
        "{} {} results",
        "✓".green(),
        results.len().to_string().cyan()
    );

    Ok(())
}