Codex Handbook
ext/memories/src/tools/list.rs 95 lines
use codex_extension_api::JsonToolOutput;use codex_extension_api::ToolCall;use codex_extension_api::ToolExecutor;use codex_extension_api::ToolName;use codex_extension_api::ToolSpec;use codex_otel::MetricsClient;use schemars::JsonSchema;use serde::Deserialize;use serde_json::json;use crate::DEFAULT_LIST_MAX_RESULTS;use crate::LIST_TOOL_NAME;use crate::MAX_LIST_RESULTS;use crate::backend::ListMemoriesRequest;use crate::backend::ListMemoriesResponse;use crate::backend::MemoriesBackend;use crate::metrics::record_tool_call;use crate::metrics::scope_from_optional_path;use crate::metrics::truncated_tag;use super::backend_error_to_function_call;use super::clamp_max_results;use super::memory_function_tool;use super::memory_tool_name;use super::parse_args;#[derive(Deserialize, JsonSchema)]#[serde(deny_unknown_fields)]struct ListArgs {    path: Option<String>,    cursor: Option<String>,    #[schemars(range(min = 1))]    max_results: Option<usize>,}#[derive(Clone)]pub(super) struct ListTool<B> {    pub(super) backend: B,    pub(super) metrics_client: Option<MetricsClient>,}impl<B> ToolExecutor<ToolCall> for ListTool<B>where    B: MemoriesBackend,{    fn tool_name(&self) -> ToolName {        memory_tool_name(LIST_TOOL_NAME)    }    fn spec(&self) -> ToolSpec {        memory_function_tool::<ListArgs, ListMemoriesResponse>(            LIST_TOOL_NAME,            "List immediate files and directories under a path in the Codex memories store.",        )    }    fn handle(&self, call: ToolCall) -> codex_extension_api::ToolExecutorFuture<'_> {        Box::pin(self.handle_call(call))    }}impl<B> ListTool<B>where    B: MemoriesBackend,{    async fn handle_call(        &self,        call: ToolCall,    ) -> Result<Box<dyn codex_extension_api::ToolOutput>, codex_extension_api::FunctionCallError>    {        let backend = self.backend.clone();        let args: ListArgs = parse_args(&call)?;        let scope = scope_from_optional_path(args.path.as_deref(), "root");        let response = backend            .list(ListMemoriesRequest {                path: args.path,                cursor: args.cursor,                max_results: clamp_max_results(                    args.max_results,                    DEFAULT_LIST_MAX_RESULTS,                    MAX_LIST_RESULTS,                ),            })            .await;        record_tool_call(            self.metrics_client.as_ref(),            LIST_TOOL_NAME,            scope,            response.is_ok(),            truncated_tag(response.as_ref().ok().map(|response| response.truncated)),        );        let response = response.map_err(backend_error_to_function_call)?;        Ok(Box::new(JsonToolOutput::new(json!(response))))    }}