use memvid_ask_model::{extract_entities, ENTITY_EXTRACTION_PROMPT};
fn main() {
let text = "John Smith met with Microsoft CEO Satya Nadella in Seattle last Tuesday. \
They discussed the partnership with S&P Global and CRISIL.";
println!("Input text: {}\n", text);
println!("Using default prompt:\n{}\n", &ENTITY_EXTRACTION_PROMPT[..200]);
match extract_entities("openai:gpt-4o-mini", text, None, None) {
Ok(response) => {
println!("✓ Extracted {} entities from {} chars using {}:\n",
response.entities.len(), response.text_chars, response.model);
for e in &response.entities {
println!(" • {} ({}) - {:.0}% confidence",
e.name, e.entity_type, e.confidence * 100.0);
}
}
Err(e) => {
eprintln!("✗ Error: {}", e);
std::process::exit(1);
}
}
}