Jake
5a15c931d0
Thin UTF-8 writer used by the new save-dialog path. Caller owns path safety; the source path is always OS-dialog-provided. Two unit tests: roundtrips a small UTF-8 string with non-ASCII chars and asserts a nonexistent parent path returns an actionable error. Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
115 lines
3.3 KiB
Rust
115 lines
3.3 KiB
Rust
pub mod audio;
|
|
pub mod clipboard;
|
|
pub mod diagnostics;
|
|
pub mod feedback;
|
|
pub mod fs;
|
|
pub mod hardware;
|
|
pub mod hotkey;
|
|
pub mod intentions;
|
|
pub mod live;
|
|
pub mod llm;
|
|
pub mod meeting;
|
|
pub mod models;
|
|
pub mod nudges;
|
|
pub mod paste;
|
|
pub mod power;
|
|
pub mod profiles;
|
|
pub mod rituals;
|
|
pub mod security;
|
|
pub mod tasks;
|
|
pub mod transcription;
|
|
pub mod transcripts;
|
|
pub mod tts;
|
|
pub mod update;
|
|
pub mod windows;
|
|
|
|
/// Build the Whisper `initial_prompt` for a transcription request.
|
|
///
|
|
/// Precedence:
|
|
/// 1. Caller-supplied `request_prompt` (non-empty wins outright — the caller
|
|
/// has already made the decision).
|
|
/// 2. Profile's stored prompt + profile terms (joined: the prompt frames the
|
|
/// task, the vocabulary biases recognition toward domain terms).
|
|
/// 3. Profile prompt alone, or vocabulary alone.
|
|
/// 4. `None` if nothing is set.
|
|
///
|
|
/// Feeding `profile_terms` into `initial_prompt` (the OpenWhispr pattern) lets
|
|
/// whisper.cpp bias its decoder toward the correct spelling of user-specific
|
|
/// vocabulary at decode time, before any LLM cleanup pass.
|
|
pub fn build_initial_prompt(
|
|
request_prompt: &str,
|
|
profile_prompt: &str,
|
|
profile_terms: &[String],
|
|
) -> Option<String> {
|
|
let trimmed_request = request_prompt.trim();
|
|
if !trimmed_request.is_empty() {
|
|
return Some(trimmed_request.to_string());
|
|
}
|
|
|
|
let trimmed_profile = profile_prompt.trim();
|
|
let terms_list = profile_terms
|
|
.iter()
|
|
.map(|term| term.trim())
|
|
.filter(|term| !term.is_empty())
|
|
.collect::<Vec<_>>()
|
|
.join(", ");
|
|
|
|
match (trimmed_profile.is_empty(), terms_list.is_empty()) {
|
|
(true, true) => None,
|
|
(false, true) => Some(trimmed_profile.to_string()),
|
|
(true, false) => Some(format!("Vocabulary: {terms_list}.")),
|
|
(false, false) => Some(format!("{trimmed_profile} Vocabulary: {terms_list}.")),
|
|
}
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use super::build_initial_prompt;
|
|
|
|
#[test]
|
|
fn caller_prompt_overrides_everything() {
|
|
let got = build_initial_prompt(
|
|
"caller wins",
|
|
"profile prompt",
|
|
&["Wren".into(), "CORBEL".into()],
|
|
);
|
|
assert_eq!(got.as_deref(), Some("caller wins"));
|
|
}
|
|
|
|
#[test]
|
|
fn profile_prompt_and_terms_are_joined() {
|
|
let got = build_initial_prompt(
|
|
"",
|
|
"You are a meeting notes assistant.",
|
|
&["Wren".into(), "CORBEL".into()],
|
|
);
|
|
assert_eq!(
|
|
got.as_deref(),
|
|
Some("You are a meeting notes assistant. Vocabulary: Wren, CORBEL."),
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn terms_only_produces_vocabulary_sentence() {
|
|
let got = build_initial_prompt("", "", &["Wren".into(), "CORBEL".into()]);
|
|
assert_eq!(got.as_deref(), Some("Vocabulary: Wren, CORBEL."));
|
|
}
|
|
|
|
#[test]
|
|
fn profile_prompt_alone_is_passed_through() {
|
|
let got = build_initial_prompt("", "Be concise.", &[]);
|
|
assert_eq!(got.as_deref(), Some("Be concise."));
|
|
}
|
|
|
|
#[test]
|
|
fn all_empty_returns_none() {
|
|
assert_eq!(build_initial_prompt("", "", &[]), None);
|
|
}
|
|
|
|
#[test]
|
|
fn whitespace_only_terms_are_skipped() {
|
|
let got = build_initial_prompt("", "", &[" ".into(), "Wren".into(), "".into()]);
|
|
assert_eq!(got.as_deref(), Some("Vocabulary: Wren."));
|
|
}
|
|
}
|