microvms-core 1.2.0

The MicroVMs control-plane and in-VM client: every trap closure, the cost engine, the sizing model
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

// SPDX-License-Identifier: Apache-2.0
//! Prompt an already provisioned VM. The external reference worker owns its lifecycle.
//! Run: cargo run -p microvms-core --example background_prompt
//! Required environment: REVIEW_VM_ID, REVIEW_ENDPOINT, REVIEW_AGENT_TOKEN, REVIEW_AGENT.
//! Optional: REVIEW_REGION (default us-east-1). Never print the private agent token.
use std::time::Duration;

use microvms_core::agents::{self, Agent, AgentPermissionMode, AgentSpec, PromptOptions};
use microvms_core::prelude::*;
use microvms_core::{Region, session::Session};

#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
    let region: Region = std::env::var("REVIEW_REGION")
        .unwrap_or_else(|_| "us-east-1".into())
        .parse()?;
    let agent: Agent = std::env::var("REVIEW_AGENT")?.parse()?;
    let session = Session::attach(
        region,
        &std::env::var("REVIEW_VM_ID")?,
        &std::env::var("REVIEW_ENDPOINT")?,
        &std::env::var("REVIEW_AGENT_TOKEN")?,
        None,
        Some(Duration::from_secs(10)),
    )
    .await?;
    let handle = agents::prompt(
        &session,
        &AgentSpec::new(agent),
        "Review /workspace/project for code quality and write /workspace/REVIEW.md.",
        &PromptOptions {
            exec_id: Some("review-task-1".into()),
            timeout: Some(Duration::from_secs(1200)),
            permission_mode: AgentPermissionMode::Unrestricted,
            reap_group_on_exit: true,
        },
    )
    .await?;
    println!("{}", handle.exec_id());
    Ok(())
}