opencrabs 0.3.47

The autonomous, self-improving AI agent. Single Rust binary. Every channel. Install with: cargo install opencrabs
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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111

//! browser_wait — Wait for a CSS selector to appear or a fixed delay.

use super::manager::BrowserManager;
use crate::brain::tools::error::Result;
use crate::brain::tools::r#trait::{Tool, ToolCapability, ToolExecutionContext, ToolResult};
use async_trait::async_trait;
use serde_json::Value;
use std::sync::Arc;
use std::time::Duration;

pub struct BrowserWaitTool {
    manager: Arc<BrowserManager>,
}

impl BrowserWaitTool {
    pub fn new(manager: Arc<BrowserManager>) -> Self {
        Self { manager }
    }
}

#[async_trait]
impl Tool for BrowserWaitTool {
    fn name(&self) -> &str {
        "browser_wait"
    }

    fn description(&self) -> &str {
        "Wait for a CSS selector to appear on the page, or wait a fixed number of seconds. \
         Polls every 200ms up to the timeout."
    }

    fn input_schema(&self) -> Value {
        serde_json::json!({
            "type": "object",
            "properties": {
                "selector": {
                    "type": "string",
                    "description": "CSS selector to wait for"
                },
                "timeout_secs": {
                    "type": "integer",
                    "description": "Max seconds to wait (default: 10)"
                },
                "delay_secs": {
                    "type": "integer",
                    "description": "Fixed delay in seconds (if no selector, just wait this long)"
                }
            }
        })
    }

    fn capabilities(&self) -> Vec<ToolCapability> {
        vec![]
    }

    fn requires_approval(&self) -> bool {
        false
    }

    async fn execute(&self, input: Value, context: &ToolExecutionContext) -> Result<ToolResult> {
        let selector = input["selector"].as_str();
        let timeout_secs = input["timeout_secs"].as_u64().unwrap_or(10);
        let delay_secs = input["delay_secs"].as_u64();

        // Fixed delay mode
        if let Some(secs) = delay_secs
            && selector.is_none()
        {
            let secs = secs.min(30);
            tokio::time::sleep(Duration::from_secs(secs)).await;
            return Ok(ToolResult::success(format!("Waited {secs} seconds")));
        }

        let sel = match selector {
            Some(s) if !s.is_empty() => s,
            _ => {
                return Ok(ToolResult::error(
                    "'selector' or 'delay_secs' required".into(),
                ));
            }
        };

        let page = match self
            .manager
            .get_or_create_session_page(context.session_id)
            .await
        {
            Ok(p) => p,
            Err(e) => return Ok(ToolResult::error(format!("Browser error: {e}"))),
        };

        let deadline = tokio::time::Instant::now() + Duration::from_secs(timeout_secs);
        loop {
            match page.find_element(sel).await {
                Ok(_) => {
                    return Ok(ToolResult::success(format!(
                        "Element '{sel}' found on page"
                    )));
                }
                Err(_) => {
                    if tokio::time::Instant::now() >= deadline {
                        return Ok(ToolResult::error(format!(
                            "Timeout: '{sel}' not found after {timeout_secs}s"
                        )));
                    }
                    tokio::time::sleep(Duration::from_millis(200)).await;
                }
            }
        }
    }
}