use edgequake_llm::{ChatMessage, LLMProvider, OpenAIProvider};
#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
let api_key = std::env::var("OPENAI_API_KEY").expect("OPENAI_API_KEY must be set");
let provider = OpenAIProvider::new(&api_key).with_model("gpt-4");
println!("🤖 EdgeQuake LLM - Basic Completion Example\n");
let messages = vec![ChatMessage::user(
"What is the capital of France? Answer in one word.",
)];
println!("Sending request to OpenAI...");
let response = provider.chat(&messages, None).await?;
println!("\n✨ Response: {}", response.content);
println!(
"📊 Tokens: {} prompt + {} completion = {} total",
response.prompt_tokens, response.completion_tokens, response.total_tokens
);
Ok(())
}