fix: 创建 agent 后刷新 loadMarkdownFilesForSubdir 缓存

新建 agent 后 clearAgentDefinitionsCache 漏清底层
loadMarkdownFilesForSubdir 的 memoize 缓存,导致新
agent 不会立即出现在列表中,需要重启才能生效。

Upstream: eb833da3
Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
This commit is contained in:
James Feng 2026-06-04 13:54:47 +08:00
parent cbe32ea58a
commit 44bfc189e0
2 changed files with 42 additions and 44 deletions

View File

@ -394,6 +394,7 @@ export const getAgentDefinitionsWithOverrides = memoize(
export function clearAgentDefinitionsCache(): void { export function clearAgentDefinitionsCache(): void {
getAgentDefinitionsWithOverrides.cache.clear?.() getAgentDefinitionsWithOverrides.cache.clear?.()
loadMarkdownFilesForSubdir.cache?.clear?.()
clearPluginAgentCache() clearPluginAgentCache()
} }

View File

@ -1,37 +1,36 @@
import chalk from 'chalk' import chalk from 'chalk';
import React, { type ReactNode, useCallback, useState } from 'react' import React, { type ReactNode, useCallback, useState } from 'react';
import { import {
type AnalyticsMetadata_I_VERIFIED_THIS_IS_NOT_CODE_OR_FILEPATHS, type AnalyticsMetadata_I_VERIFIED_THIS_IS_NOT_CODE_OR_FILEPATHS,
logEvent, logEvent,
} from 'src/services/analytics/index.js' } from 'src/services/analytics/index.js';
import { useSetAppState } from 'src/state/AppState.js' import { useSetAppState } from 'src/state/AppState.js';
import type { Tools } from '../../../../Tool.js' import type { Tools } from '../../../../Tool.js';
import type { AgentDefinition } from '@claude-code-best/builtin-tools/tools/AgentTool/loadAgentsDir.js' import type { AgentDefinition } from '@claude-code-best/builtin-tools/tools/AgentTool/loadAgentsDir.js';
import { getActiveAgentsFromList } from '@claude-code-best/builtin-tools/tools/AgentTool/loadAgentsDir.js' import {
import { editFileInEditor } from '../../../../utils/promptEditor.js' clearAgentDefinitionsCache,
import { useWizard } from '../../../wizard/index.js' getActiveAgentsFromList,
import { getNewAgentFilePath, saveAgentToFile } from '../../agentFileUtils.js' } from '@claude-code-best/builtin-tools/tools/AgentTool/loadAgentsDir.js';
import type { AgentWizardData } from '../types.js' import { editFileInEditor } from '../../../../utils/promptEditor.js';
import { ConfirmStep } from './ConfirmStep.js' import { useWizard } from '../../../wizard/index.js';
import { getNewAgentFilePath, saveAgentToFile } from '../../agentFileUtils.js';
import type { AgentWizardData } from '../types.js';
import { ConfirmStep } from './ConfirmStep.js';
type Props = { type Props = {
tools: Tools tools: Tools;
existingAgents: AgentDefinition[] existingAgents: AgentDefinition[];
onComplete: (message: string) => void onComplete: (message: string) => void;
} };
export function ConfirmStepWrapper({ export function ConfirmStepWrapper({ tools, existingAgents, onComplete }: Props): ReactNode {
tools, const { wizardData } = useWizard<AgentWizardData>();
existingAgents, const [saveError, setSaveError] = useState<string | null>(null);
onComplete, const setAppState = useSetAppState();
}: Props): ReactNode {
const { wizardData } = useWizard<AgentWizardData>()
const [saveError, setSaveError] = useState<string | null>(null)
const setAppState = useSetAppState()
const saveAgent = useCallback( const saveAgent = useCallback(
async (openInEditor: boolean): Promise<void> => { async (openInEditor: boolean): Promise<void> => {
if (!wizardData?.finalAgent) return if (!wizardData?.finalAgent) return;
try { try {
await saveAgentToFile( await saveAgentToFile(
@ -44,14 +43,12 @@ export function ConfirmStepWrapper({
wizardData.finalAgent.color, wizardData.finalAgent.color,
wizardData.finalAgent.model, wizardData.finalAgent.model,
wizardData.finalAgent.memory, wizardData.finalAgent.memory,
) );
setAppState(state => { setAppState(state => {
if (!wizardData.finalAgent) return state if (!wizardData.finalAgent) return state;
const allAgents = state.agentDefinitions.allAgents.concat( const allAgents = state.agentDefinitions.allAgents.concat(wizardData.finalAgent);
wizardData.finalAgent,
)
return { return {
...state, ...state,
agentDefinitions: { agentDefinitions: {
@ -59,15 +56,17 @@ export function ConfirmStepWrapper({
activeAgents: getActiveAgentsFromList(allAgents), activeAgents: getActiveAgentsFromList(allAgents),
allAgents, allAgents,
}, },
} };
}) });
clearAgentDefinitionsCache();
if (openInEditor) { if (openInEditor) {
const filePath = getNewAgentFilePath({ const filePath = getNewAgentFilePath({
source: wizardData.location!, source: wizardData.location!,
agentType: wizardData.finalAgent.agentType, agentType: wizardData.finalAgent.agentType,
}) });
await editFileInEditor(filePath) await editFileInEditor(filePath);
} }
logEvent('tengu_agent_created', { logEvent('tengu_agent_created', {
@ -80,25 +79,23 @@ export function ConfirmStepWrapper({
has_memory: !!wizardData.finalAgent.memory, has_memory: !!wizardData.finalAgent.memory,
memory_scope: wizardData.finalAgent.memory ?? 'none', memory_scope: wizardData.finalAgent.memory ?? 'none',
...(openInEditor ? { opened_in_editor: true } : {}), ...(openInEditor ? { opened_in_editor: true } : {}),
} as AnalyticsMetadata_I_VERIFIED_THIS_IS_NOT_CODE_OR_FILEPATHS) } as AnalyticsMetadata_I_VERIFIED_THIS_IS_NOT_CODE_OR_FILEPATHS);
const message = openInEditor const message = openInEditor
? `Created agent: ${chalk.bold(wizardData.finalAgent.agentType)} and opened in editor. ` + ? `Created agent: ${chalk.bold(wizardData.finalAgent.agentType)} and opened in editor. ` +
`If you made edits, restart to load the latest version.` `If you made edits, restart to load the latest version.`
: `Created agent: ${chalk.bold(wizardData.finalAgent.agentType)}` : `Created agent: ${chalk.bold(wizardData.finalAgent.agentType)}`;
onComplete(message) onComplete(message);
} catch (err) { } catch (err) {
setSaveError( setSaveError(err instanceof Error ? err.message : 'Failed to save agent');
err instanceof Error ? err.message : 'Failed to save agent',
)
} }
}, },
[wizardData, onComplete, setAppState], [wizardData, onComplete, setAppState],
) );
const handleSave = useCallback(() => saveAgent(false), [saveAgent]) const handleSave = useCallback(() => saveAgent(false), [saveAgent]);
const handleSaveAndEdit = useCallback(() => saveAgent(true), [saveAgent]) const handleSaveAndEdit = useCallback(() => saveAgent(true), [saveAgent]);
return ( return (
<ConfirmStep <ConfirmStep
@ -108,5 +105,5 @@ export function ConfirmStepWrapper({
onSaveAndEdit={handleSaveAndEdit} onSaveAndEdit={handleSaveAndEdit}
error={saveError} error={saveError}
/> />
) );
} }