mirror of
https://github.com/openai/codex.git
synced 2026-11-13 14:46:14 +00:00
Cache plugin namespace during executor skill discovery (#29831)
## Why Executor skill discovery runs before the remote skills catalog is available. For a remote environment, each `ExecutorFileSystem` operation becomes an exec-server RPC. Previously, every discovered `SKILL.md` independently resolved its plugin namespace by walking its ancestors and probing both supported manifest locations. In the common `plugin/skills/<skill>/SKILL.md` layout, that repeats 8 RPCs per skill even though every skill under the plugin root uses the same namespace. These lookups happen while skills are parsed, so their cost grows linearly with the skill count and adds directly to first-turn latency. A selected capability root can also contain standalone skills, multiple sibling plugins, nested plugins, or symlinked directories. The optimization therefore needs to retain the nearest-ancestor namespace for each skill rather than assuming the selected root represents exactly one plugin. ## What changed - record plugin-root candidates from directory entries already returned during skill discovery - prune candidates that are not ancestors of any discovered `SKILL.md` before reading manifests - resolve each relevant plugin root once, with one fallback lookup per canonical traversal root for symlinked directories - select the nearest cached plugin namespace for each discovered skill - avoid namespace lookup entirely when the root contains no skills No additional directory traversal is required. Namespace work now scales with the number of plugin roots that contain discovered skills, rather than the total number of skills or unrelated sibling plugins. Standalone and nested-plugin names keep their previous behavior. ## Benchmarks I used a temporary counting `ExecutorFileSystem` around the real local filesystem. Each filesystem operation was counted as one remote RPC and given 1 ms of injected latency. Each variant ran three times; times below are medians. ### One plugin with 100 skills | Operation | Before | After | Delta | | --- | ---: | ---: | ---: | | `get_metadata` | 1,002 | 303 | -699 | | `read_file` | 200 | 101 | -99 | | `read_directory` | 102 | 102 | 0 | | **Total filesystem RPCs** | **1,304** | **506** | **-798 (-61.2%)** | | **Median load time** | **2.890 s** | **0.997 s** | **2.90× faster** | The namespace-specific work drops from 800 RPCs to 2 in this layout. ### Multiple plugins under one selected root These runs compare the correct pre-optimization implementation with the final nearest-plugin-root cache. The total plugin skill count stays at 100 while the number of plugin roots changes. | Layout | Before RPCs | After RPCs | Reduction | Before | After | Speedup | | --- | ---: | ---: | ---: | ---: | ---: | ---: | | 2 plugins × 50 skills | 1,312 | 530 | 59.6% | 1,819 ms | 711 ms | 2.56× | | 10 plugins × 10 skills | 1,344 | 578 | 57.0% | 1,850 ms | 778 ms | 2.38× | | 50 plugins × 2 skills | 1,504 | 818 | 45.6% | 2,094 ms | 1,086 ms | 1.93× | | 10 plugins × 10 skills + 10 standalone skills | 1,596 | 630 | 60.5% | 2,209 ms | 860 ms | 2.57× | The remaining cost grows with the number of relevant plugin manifests. Each relevant manifest is read once instead of once per skill, while sibling plugins with no discovered skills are not read. Absolute latency savings depend on the executor's real RPC latency. ## Tests - `just test -p codex-core-skills` (109 passed across the library and integration-test binaries) - one integration test covers standalone, outer-plugin, nested-plugin, and unused sibling-plugin layouts, and asserts the exact set of manifests read
This commit is contained in:
5 files changed
+301
-6
No files matched your search
@@ -1,3 +1,5 @@
|
||||
use std::collections::HashMap;
|
||||
use std::collections::HashSet;
|
||||
use std::io;
|
||||
|
||||
use codex_exec_server::ExecutorFileSystem;
|
||||
@@ -5,7 +7,9 @@ use codex_exec_server::WalkEntryKind;
|
||||
use codex_exec_server::WalkOptions;
|
||||
use codex_protocol::protocol::Product;
|
||||
use codex_utils_path_uri::PathUri;
|
||||
use codex_utils_plugins::plugin_namespace_for_root_uri;
|
||||
use codex_utils_plugins::plugin_namespace_for_skill_uri;
|
||||
use futures::future::join_all;
|
||||
|
||||
use crate::model::SkillDependencies;
|
||||
use crate::model::SkillPolicy;
|
||||
@@ -58,7 +62,11 @@ impl EnvironmentSkillMetadata {
|
||||
}
|
||||
}
|
||||
|
||||
async fn parse(file_system: &dyn ExecutorFileSystem, path: &PathUri) -> Result<Self, String> {
|
||||
async fn parse(
|
||||
file_system: &dyn ExecutorFileSystem,
|
||||
path: &PathUri,
|
||||
plugin_namespace: Option<&str>,
|
||||
) -> Result<Self, String> {
|
||||
let contents = file_system
|
||||
.read_file_text(path, /*sandbox*/ None)
|
||||
.await
|
||||
@@ -69,8 +77,7 @@ impl EnvironmentSkillMetadata {
|
||||
short_description,
|
||||
} = parse_skill_frontmatter_metadata_inner(&contents, || default_skill_name(path))
|
||||
.map_err(|err| err.to_string())?;
|
||||
let name = plugin_namespace_for_skill_uri(file_system, path)
|
||||
.await
|
||||
let name = plugin_namespace
|
||||
.map(|namespace| format!("{namespace}:{base_name}"))
|
||||
.unwrap_or(base_name);
|
||||
validate_len(&name, MAX_QUALIFIED_NAME_LEN, "qualified name")
|
||||
@@ -152,8 +159,65 @@ pub async fn load_environment_skills_from_root(
|
||||
},
|
||||
};
|
||||
outcome.warnings.extend(discovery.warnings);
|
||||
if discovery.skill_files.is_empty() {
|
||||
return outcome;
|
||||
}
|
||||
|
||||
let mut skill_ancestors = HashSet::new();
|
||||
for skill_path in &discovery.skill_files {
|
||||
let mut ancestor = skill_path.parent();
|
||||
while let Some(path) = ancestor {
|
||||
skill_ancestors.insert(path.clone());
|
||||
ancestor = path.parent();
|
||||
}
|
||||
}
|
||||
|
||||
let namespace_roots = discovery.namespace_roots;
|
||||
let namespace_lookups = join_all(namespace_roots.iter().map(|namespace_root| async {
|
||||
(
|
||||
namespace_root.clone(),
|
||||
plugin_namespace_for_skill_uri(file_system, namespace_root).await,
|
||||
)
|
||||
}))
|
||||
.await;
|
||||
let plugin_lookups = join_all(
|
||||
discovery
|
||||
.plugin_roots
|
||||
.iter()
|
||||
.filter(|plugin_root| skill_ancestors.contains(*plugin_root))
|
||||
.filter(|plugin_root| !namespace_roots.contains(*plugin_root))
|
||||
.map(|plugin_root| async {
|
||||
(
|
||||
plugin_root.clone(),
|
||||
plugin_namespace_for_root_uri(file_system, plugin_root).await,
|
||||
)
|
||||
}),
|
||||
)
|
||||
.await;
|
||||
let plugin_namespaces = namespace_lookups
|
||||
.into_iter()
|
||||
.chain(plugin_lookups)
|
||||
.filter_map(|(plugin_root, namespace)| namespace.map(|namespace| (plugin_root, namespace)))
|
||||
.collect::<HashMap<_, _>>();
|
||||
|
||||
for path in discovery.skill_files {
|
||||
match EnvironmentSkillMetadata::parse(file_system, &path).await {
|
||||
let mut ancestor = path.parent();
|
||||
let plugin_namespace = loop {
|
||||
let Some(current) = ancestor else {
|
||||
break None;
|
||||
};
|
||||
if let Some(namespace) = plugin_namespaces.get(¤t) {
|
||||
break Some(namespace.as_str());
|
||||
}
|
||||
ancestor = current.parent();
|
||||
};
|
||||
match EnvironmentSkillMetadata::parse(
|
||||
file_system,
|
||||
&path,
|
||||
/*plugin_namespace*/ plugin_namespace,
|
||||
)
|
||||
.await
|
||||
{
|
||||
Ok(skill) if skill.matches_product_restriction(restriction_product) => {
|
||||
outcome.skills.push(skill);
|
||||
}
|
||||
|
||||
Reference in new issue
Block a user