use agent_core::agent::{Agent, AgentConfig};
use agent_provider::OpenAIProvider;

let provider = OpenAIProvider::new("https://api.openai.com/v1", &api_key);
let agent = Agent::new(AgentConfig {
    provider: Box::new(provider),
    model: Model::openai("gpt-4o"),
    system_prompt: "You are a helpful assistant.".into(),
    tools: vec![],  // add tools here
    ..Default::default()
});

let result = agent.query("Hello!", &cancel, |event| {
    // handle streaming events
}).await?;

use agent_tools::tool_set::ToolSet;
use agent_sandbox::local::LocalSandbox;

let sandbox = Arc::new(LocalSandbox::new("/tmp/workspace"));
let tool_set = ToolSet::cli(sandbox)
    .with_web_fetch(Arc::new(DefaultFetcher))
    .build();

let agent = Agent::new(AgentConfig {
    tools: tool_set.tool_definitions(),
    ..config
});

async fn chat(State(agent): State<Agent>) -> Sse<impl Stream<Item = Event>> {
    let (tx, rx) = mpsc::channel(32);

    tokio::spawn(async move {
        agent.query(&input, &cancel, move |event| {
            if let AgentEvent::MessageUpdate { event: AssistantMessageEvent::TextDelta { delta, .. }, .. } = event {
                tx.blocking_send(Event::default().data(delta)).ok();
            }
        }).await;
    });

    Sse::new(ReceiverStream::new(rx))
}

use agent_session::{SessionManager, SessionBoundAgent};

let manager = SessionManager::new(FileSystemStorage::new("./sessions"));
let mut bound = SessionBoundAgent::create(manager, agent_config, session_config).await?;

// Conversations auto-persist across prompts
bound.prompt("Fix the login bug", &cancel).await?;

// Resume later
let mut bound = SessionBoundAgent::resume(manager, session_id, agent_config).await?;
bound.prompt("What was I working on?", &cancel).await?;