feat(environment): introduce disable_omo_env configuration option
- Added a new configuration option `disable_omo_env` to control the injection of the `<omo-env>` block in agent prompts. - Updated relevant functions and tests to support this feature, ensuring that the environment context can be toggled on or off as needed. - Enhanced documentation to reflect the new option and its implications for API cost and cache hit rates.
This commit is contained in:
@@ -1,8 +1,16 @@
|
||||
import type { AgentConfig } from "@opencode-ai/sdk"
|
||||
import { createEnvContext } from "../env-context"
|
||||
|
||||
export function applyEnvironmentContext(config: AgentConfig, directory?: string): AgentConfig {
|
||||
if (!directory || !config.prompt) return config
|
||||
type ApplyEnvironmentContextOptions = {
|
||||
disableOmoEnv?: boolean
|
||||
}
|
||||
|
||||
export function applyEnvironmentContext(
|
||||
config: AgentConfig,
|
||||
directory?: string,
|
||||
options: ApplyEnvironmentContextOptions = {}
|
||||
): AgentConfig {
|
||||
if (options.disableOmoEnv || !directory || !config.prompt) return config
|
||||
const envContext = createEnvContext()
|
||||
return { ...config, prompt: config.prompt + envContext }
|
||||
}
|
||||
|
||||
@@ -23,6 +23,7 @@ export function collectPendingBuiltinAgents(input: {
|
||||
availableModels: Set<string>
|
||||
disabledSkills?: Set<string>
|
||||
useTaskSystem?: boolean
|
||||
disableOmoEnv?: boolean
|
||||
}): { pendingAgentConfigs: Map<string, AgentConfig>; availableAgents: AvailableAgent[] } {
|
||||
const {
|
||||
agentSources,
|
||||
@@ -37,6 +38,7 @@ export function collectPendingBuiltinAgents(input: {
|
||||
uiSelectedModel,
|
||||
availableModels,
|
||||
disabledSkills,
|
||||
disableOmoEnv = false,
|
||||
} = input
|
||||
|
||||
const availableAgents: AvailableAgent[] = []
|
||||
@@ -81,7 +83,7 @@ export function collectPendingBuiltinAgents(input: {
|
||||
}
|
||||
|
||||
if (agentName === "librarian") {
|
||||
config = applyEnvironmentContext(config, directory)
|
||||
config = applyEnvironmentContext(config, directory, { disableOmoEnv })
|
||||
}
|
||||
|
||||
config = applyOverrides(config, override, mergedCategories, directory)
|
||||
|
||||
@@ -4,7 +4,7 @@ import type { CategoryConfig } from "../../config/schema"
|
||||
import type { AvailableAgent, AvailableCategory, AvailableSkill } from "../dynamic-agent-prompt-builder"
|
||||
import { AGENT_MODEL_REQUIREMENTS, isAnyProviderConnected } from "../../shared"
|
||||
import { createHephaestusAgent } from "../hephaestus"
|
||||
import { createEnvContext } from "../env-context"
|
||||
import { applyEnvironmentContext } from "./environment-context"
|
||||
import { applyCategoryOverride, mergeAgentConfig } from "./agent-overrides"
|
||||
import { applyModelResolution, getFirstFallbackModel } from "./model-resolution"
|
||||
|
||||
@@ -20,6 +20,7 @@ export function maybeCreateHephaestusConfig(input: {
|
||||
mergedCategories: Record<string, CategoryConfig>
|
||||
directory?: string
|
||||
useTaskSystem: boolean
|
||||
disableOmoEnv?: boolean
|
||||
}): AgentConfig | undefined {
|
||||
const {
|
||||
disabledAgents,
|
||||
@@ -33,6 +34,7 @@ export function maybeCreateHephaestusConfig(input: {
|
||||
mergedCategories,
|
||||
directory,
|
||||
useTaskSystem,
|
||||
disableOmoEnv = false,
|
||||
} = input
|
||||
|
||||
if (disabledAgents.includes("hephaestus")) return undefined
|
||||
@@ -79,10 +81,7 @@ export function maybeCreateHephaestusConfig(input: {
|
||||
hephaestusConfig = applyCategoryOverride(hephaestusConfig, hepOverrideCategory, mergedCategories)
|
||||
}
|
||||
|
||||
if (directory && hephaestusConfig.prompt) {
|
||||
const envContext = createEnvContext()
|
||||
hephaestusConfig = { ...hephaestusConfig, prompt: hephaestusConfig.prompt + envContext }
|
||||
}
|
||||
hephaestusConfig = applyEnvironmentContext(hephaestusConfig, directory, { disableOmoEnv })
|
||||
|
||||
if (hephaestusOverride) {
|
||||
hephaestusConfig = mergeAgentConfig(hephaestusConfig, hephaestusOverride, directory)
|
||||
|
||||
@@ -22,6 +22,7 @@ export function maybeCreateSisyphusConfig(input: {
|
||||
directory?: string
|
||||
userCategories?: CategoriesConfig
|
||||
useTaskSystem: boolean
|
||||
disableOmoEnv?: boolean
|
||||
}): AgentConfig | undefined {
|
||||
const {
|
||||
disabledAgents,
|
||||
@@ -36,6 +37,7 @@ export function maybeCreateSisyphusConfig(input: {
|
||||
mergedCategories,
|
||||
directory,
|
||||
useTaskSystem,
|
||||
disableOmoEnv = false,
|
||||
} = input
|
||||
|
||||
const sisyphusOverride = agentOverrides["sisyphus"]
|
||||
@@ -78,7 +80,9 @@ export function maybeCreateSisyphusConfig(input: {
|
||||
}
|
||||
|
||||
sisyphusConfig = applyOverrides(sisyphusConfig, sisyphusOverride, mergedCategories, directory)
|
||||
sisyphusConfig = applyEnvironmentContext(sisyphusConfig, directory)
|
||||
sisyphusConfig = applyEnvironmentContext(sisyphusConfig, directory, {
|
||||
disableOmoEnv,
|
||||
})
|
||||
|
||||
return sisyphusConfig
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user