agentic-vision-mcp 0.3.0

MCP server for AgenticVision — universal LLM access to persistent visual memory
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

//! Tool: vision_workspace_query — Query across all vision contexts.

use std::sync::Arc;
use tokio::sync::Mutex;

use serde::Deserialize;
use serde_json::{json, Value};

use crate::session::VisionSessionManager;
use crate::types::{McpError, McpResult, ToolCallResult, ToolDefinition};

#[derive(Debug, Deserialize)]
struct QueryParams {
    workspace_id: String,
    query: String,
    #[serde(default = "default_max")]
    max_per_context: usize,
}

fn default_max() -> usize {
    10
}

pub fn definition() -> ToolDefinition {
    ToolDefinition {
        name: "vision_workspace_query".to_string(),
        description: Some("Search across all vision contexts in a workspace.".to_string()),
        input_schema: json!({
            "type": "object",
            "required": ["workspace_id", "query"],
            "properties": {
                "workspace_id": { "type": "string" },
                "query": { "type": "string" },
                "max_per_context": { "type": "integer", "default": 10 }
            }
        }),
    }
}

pub async fn execute(
    args: Value,
    session: &Arc<Mutex<VisionSessionManager>>,
) -> McpResult<ToolCallResult> {
    let params: QueryParams =
        serde_json::from_value(args).map_err(|e| McpError::InvalidParams(e.to_string()))?;
    let session = session.lock().await;
    let results = session.workspace_manager().query_all(
        &params.workspace_id,
        &params.query,
        params.max_per_context,
    )?;
    let total: usize = results.iter().map(|r| r.matches.len()).sum();
    let items: Vec<Value> = results
        .iter()
        .map(|cr| {
            let matches: Vec<Value> = cr.matches.iter().map(|m| {
                json!({ "observation_id": m.observation_id, "description": m.description, "labels": m.labels, "score": m.score })
            }).collect();
            json!({ "context_id": cr.context_id, "context_role": cr.context_role.label(), "match_count": cr.matches.len(), "matches": matches })
        })
        .collect();
    Ok(ToolCallResult::json(
        &json!({ "workspace_id": params.workspace_id, "query": params.query, "total_matches": total, "results": items }),
    ))
}