talon-core 0.4.2

Core retrieval engine for Talon: hybrid search (BM25 + semantic + reranker), indexing, and graph-aware ranking over markdown corpora.
Documentation 
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
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87

//! Shared [`TalonConfig`] literals for unit tests.

use std::path::{Path, PathBuf};

use super::{
    ChatAdapter, ChatAskConfig, ChatExpansionConfig, ChatSection, ChunkerConfig, CredentialsConfig,
    EmbeddingAdapter, EmbeddingConfig, EndpointAuthConfig, InspectConfig, McpConfig, RerankAdapter,
    RerankConfig, RerankScoreScale, ScopesConfig, SearchConfig, TalonConfig,
};

/// Builds a minimal config with default scopes and the given HTTP base URL.
#[must_use]
pub fn minimal(base_url: impl Into<String>) -> TalonConfig {
    minimal_for_paths(
        PathBuf::from("/vault"),
        PathBuf::from("/vault/.talon/index.db"),
        base_url,
        ScopesConfig::default(),
    )
}

/// Builds a minimal config for explicit vault/db paths.
#[must_use]
pub fn minimal_for_paths(
    vault_path: PathBuf,
    db_path: PathBuf,
    base_url: impl Into<String>,
    scopes: ScopesConfig,
) -> TalonConfig {
    let base_url = base_url.into();
    let chat_base = if base_url.ends_with("/v1") {
        base_url.clone()
    } else {
        format!("{base_url}/v1")
    };
    TalonConfig {
        vault_path,
        db_path,
        config_file_path: None,
        include_patterns: Vec::new(),
        ignore_patterns: Vec::new(),
        credentials: CredentialsConfig::default(),
        embedding: EmbeddingConfig {
            base_url: base_url.clone(),
            auth: EndpointAuthConfig::default(),
            adapter: EmbeddingAdapter::Tei,
            model: "embed".to_string(),
            document_model: Some("embed_chunked".to_string()),
            context_tokens: 512,
        },
        rerank: RerankConfig {
            base_url,
            auth: EndpointAuthConfig::default(),
            adapter: RerankAdapter::Minimal,
            model: "rerank".to_string(),
            score_scale: RerankScoreScale::default(),
            truncate: true,
        },
        chat: ChatSection {
            expansion: ChatExpansionConfig {
                base_url: chat_base,
                auth: EndpointAuthConfig::default(),
                adapter: ChatAdapter::default(),
                model: "test".to_string(),
                context_tokens: 32_768,
                max_output_tokens: None,
            },
            ask: ChatAskConfig::default(),
        },
        mcp: McpConfig::default(),
        scopes,
        search: SearchConfig::default(),
        inspect: InspectConfig::default(),
        chunker: ChunkerConfig::default(),
    }
}

/// Builds a minimal config rooted at `vault` with the default sidecar URL.
#[must_use]
pub fn minimal_for_vault(vault: &Path) -> TalonConfig {
    minimal_for_paths(
        vault.to_path_buf(),
        vault.join("idx.sqlite"),
        "http://localhost:1",
        ScopesConfig::default(),
    )
}