agent-sdk-toolkit 0.1.0-alpha.4

//! Resource-reader helpers layered over explicit URI resolvers and core content refs.
//! Use these modules when a host wants toolkit tools to read approved resources.
//! Resolver implementations own any backing-store or network side effects. This file
//! contains the resolver portion of that contract.
//!
use std::sync::Arc;

use agent_sdk_core::{
    AgentError, PolicyRef, ResourceReadRequest, ResourceResolution, ResourceResolver,
    ResourceScheme, RetentionClass, domain::ContentRef,
};

#[derive(Clone)]
/// Resource in memory resource resolver request or result value.
/// Creating the value does not fetch content; resource executors document resolver and content-store effects.
pub struct InMemoryResourceResolver {
    scheme: ResourceScheme,
    content_ref: ContentRef,
    source: agent_sdk_core::SourceRef,
    policy_ref: PolicyRef,
}

impl InMemoryResourceResolver {
    /// Creates a new resources::resolver value with explicit
    /// caller-provided inputs. This constructor is data-only and
    /// performs no I/O or external side effects.
    pub fn new(
        scheme: &str,
        content_ref: ContentRef,
        source: agent_sdk_core::SourceRef,
        policy_ref: PolicyRef,
    ) -> Arc<Self> {
        Arc::new(Self {
            scheme: ResourceScheme::new(scheme),
            content_ref,
            source,
            policy_ref,
        })
    }
}

impl ResourceResolver for InMemoryResourceResolver {
    fn scheme(&self) -> &ResourceScheme {
        &self.scheme
    }

    fn resolve(&self, request: &ResourceReadRequest) -> Result<ResourceResolution, AgentError> {
        Ok(ResourceResolution {
            uri: request.uri.clone(),
            scheme: self.scheme.clone(),
            content_ref: self.content_ref.clone(),
            source: self.source.clone(),
            policy_refs: vec![self.policy_ref.clone()],
            byte_len: 0,
            truncated: false,
            parser_version: "toolkit.in_memory_resource.v1".to_string(),
            privacy: agent_sdk_core::PrivacyClass::ContentRefsOnly,
            retention: RetentionClass::RunScoped,
            redacted_summary: "in-memory resource resolved to content ref".to_string(),
        })
    }
}