versun/claude-code
1
0
Fork 0
mirror of https://github.com/claude-code-best/claude-code.git synced 2026-06-23 00:35:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

style: 完成所有文件的lint

Browse Source
This commit is contained in:
claude-code-best
2026-05-01 21:39:30 +08:00
parent d136872cc9
commit 6182015005
1333 changed files with 68255 additions and 77882 deletions
Download Patch File Download Diff File Expand all files Collapse all files

147
src/components/agents/new-agent-creation/wizard-steps/GenerateStep.tsx
View File

@@ -1,56 +1,55 @@
import { APIUserAbortError } from '@anthropic-ai/sdk'
import React, { type ReactNode, useCallback, useRef, useState } from 'react'
import { useMainLoopModel } from '../../../../hooks/useMainLoopModel.js'
import { Box, Byline, Text } from '@anthropic/ink'
import { useKeybinding } from '../../../../keybindings/useKeybinding.js'
import { createAbortController } from '../../../../utils/abortController.js'
import { editPromptInEditor } from '../../../../utils/promptEditor.js'
import { ConfigurableShortcutHint } from '../../../ConfigurableShortcutHint.js'
import { Spinner } from '../../../Spinner.js'
import TextInput from '../../../TextInput.js'
import { useWizard } from '../../../wizard/index.js'
import { WizardDialogLayout } from '../../../wizard/WizardDialogLayout.js'
import { generateAgent } from '../../generateAgent.js'
import type { AgentWizardData } from '../types.js'
import { APIUserAbortError } from '@anthropic-ai/sdk';
import React, { type ReactNode, useCallback, useRef, useState } from 'react';
import { useMainLoopModel } from '../../../../hooks/useMainLoopModel.js';
import { Box, Byline, Text } from '@anthropic/ink';
import { useKeybinding } from '../../../../keybindings/useKeybinding.js';
import { createAbortController } from '../../../../utils/abortController.js';
import { editPromptInEditor } from '../../../../utils/promptEditor.js';
import { ConfigurableShortcutHint } from '../../../ConfigurableShortcutHint.js';
import { Spinner } from '../../../Spinner.js';
import TextInput from '../../../TextInput.js';
import { useWizard } from '../../../wizard/index.js';
import { WizardDialogLayout } from '../../../wizard/WizardDialogLayout.js';
import { generateAgent } from '../../generateAgent.js';
import type { AgentWizardData } from '../types.js';
export function GenerateStep(): ReactNode {
const { updateWizardData, goBack, goToStep, wizardData } =
useWizard<AgentWizardData>()
const [prompt, setPrompt] = useState(wizardData.generationPrompt || '')
const [isGenerating, setIsGenerating] = useState(false)
const [error, setError] = useState<string | null>(null)
const [cursorOffset, setCursorOffset] = useState(prompt.length)
const model = useMainLoopModel()
const abortControllerRef = useRef<AbortController | null>(null)
const { updateWizardData, goBack, goToStep, wizardData } = useWizard<AgentWizardData>();
const [prompt, setPrompt] = useState(wizardData.generationPrompt || '');
const [isGenerating, setIsGenerating] = useState(false);
const [error, setError] = useState<string | null>(null);
const [cursorOffset, setCursorOffset] = useState(prompt.length);
const model = useMainLoopModel();
const abortControllerRef = useRef<AbortController | null>(null);
// Cancel generation when escape pressed during generation
const handleCancelGeneration = useCallback(() => {
if (abortControllerRef.current) {
abortControllerRef.current.abort()
abortControllerRef.current = null
setIsGenerating(false)
setError('Generation cancelled')
abortControllerRef.current.abort();
abortControllerRef.current = null;
setIsGenerating(false);
setError('Generation cancelled');
}
}, [])
}, []);
// Use Settings context so 'n' key doesn't cancel (allows typing 'n' in prompt input)
useKeybinding('confirm:no', handleCancelGeneration, {
context: 'Settings',
isActive: isGenerating,
})
});
const handleExternalEditor = useCallback(async () => {
const result = await editPromptInEditor(prompt)
const result = await editPromptInEditor(prompt);
if (result.content !== null) {
setPrompt(result.content)
setCursorOffset(result.content.length)
setPrompt(result.content);
setCursorOffset(result.content.length);
}
}, [prompt])
}, [prompt]);
useKeybinding('chat:externalEditor', handleExternalEditor, {
context: 'Chat',
isActive: !isGenerating,
})
});
// Go back when escape pressed while not generating
const handleGoBack = useCallback(() => {
@@ -61,43 +60,38 @@ export function GenerateStep(): ReactNode {
whenToUse: '',
generatedAgent: undefined,
wasGenerated: false,
})
setPrompt('')
setError(null)
goBack()
}, [updateWizardData, goBack])
});
setPrompt('');
setError(null);
goBack();
}, [updateWizardData, goBack]);
// Use Settings context so 'n' key doesn't cancel (allows typing 'n' in prompt input)
useKeybinding('confirm:no', handleGoBack, {
context: 'Settings',
isActive: !isGenerating,
})
});
const handleGenerate = async (): Promise<void> => {
const trimmedPrompt = prompt.trim()
const trimmedPrompt = prompt.trim();
if (!trimmedPrompt) {
setError('Please describe what the agent should do')
return
setError('Please describe what the agent should do');
return;
}
setError(null)
setIsGenerating(true)
setError(null);
setIsGenerating(true);
updateWizardData({
generationPrompt: trimmedPrompt,
isGenerating: true,
})
});
// Create abort controller for this generation
const controller = createAbortController()
abortControllerRef.current = controller
const controller = createAbortController();
abortControllerRef.current = controller;
try {
const generated = await generateAgent(
trimmedPrompt,
model,
[],
controller.signal,
)
const generated = await generateAgent(trimmedPrompt, model, [], controller.signal);
updateWizardData({
agentType: generated.identifier,
@@ -106,41 +100,32 @@ export function GenerateStep(): ReactNode {
generatedAgent: generated,
isGenerating: false,
wasGenerated: true,
})
});
// Skip directly to ToolsStep (index 6) - matching original flow
goToStep(6)
goToStep(6);
} catch (err) {
// Don't show error if it was cancelled (already set in escape handler)
if (err instanceof APIUserAbortError) {
// User cancelled - no error to show
} else if (
err instanceof Error &&
!err.message.includes('No assistant message found')
) {
setError(err.message || 'Failed to generate agent')
} else if (err instanceof Error && !err.message.includes('No assistant message found')) {
setError(err.message || 'Failed to generate agent');
}
updateWizardData({ isGenerating: false })
updateWizardData({ isGenerating: false });
} finally {
setIsGenerating(false)
abortControllerRef.current = null
setIsGenerating(false);
abortControllerRef.current = null;
}
}
};
const subtitle =
'Describe what this agent should do and when it should be used (be comprehensive for best results)'
const subtitle = 'Describe what this agent should do and when it should be used (be comprehensive for best results)';
if (isGenerating) {
return (
<WizardDialogLayout
subtitle={subtitle}
footerText={
<ConfigurableShortcutHint
action="confirm:no"
context="Settings"
fallback="Esc"
description="cancel"
/>
<ConfigurableShortcutHint action="confirm:no" context="Settings" fallback="Esc" description="cancel" />
}
>
<Box flexDirection="row" alignItems="center">
@@ -148,7 +133,7 @@ export function GenerateStep(): ReactNode {
<Text color="suggestion"> Generating agent from description...</Text>
</Box>
</WizardDialogLayout>
)
);
}
return (
@@ -156,24 +141,14 @@ export function GenerateStep(): ReactNode {
subtitle={subtitle}
footerText={
<Byline>
<ConfigurableShortcutHint
action="confirm:yes"
context="Confirmation"
fallback="Enter"
description="submit"
/>
<ConfigurableShortcutHint action="confirm:yes" context="Confirmation" fallback="Enter" description="submit" />
<ConfigurableShortcutHint
action="chat:externalEditor"
context="Chat"
fallback="ctrl+g"
description="open in editor"
/>
<ConfigurableShortcutHint
action="confirm:no"
context="Settings"
fallback="Esc"
description="go back"
/>
<ConfigurableShortcutHint action="confirm:no" context="Settings" fallback="Esc" description="go back" />
</Byline>
}
>
@@ -196,5 +171,5 @@ export function GenerateStep(): ReactNode {
/>
</Box>
</WizardDialogLayout>
)
);
}