Merge pull request #166 from MODSetter/dev

fix: onboarding bug
This commit is contained in:
Rohan Verma 2025-06-09 17:17:35 -07:00 committed by GitHub
commit fa54de1f41
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 16 additions and 4 deletions

View file

@ -17,7 +17,7 @@ const TOTAL_STEPS = 3;
const OnboardPage = () => {
const router = useRouter();
const { llmConfigs, loading: configsLoading } = useLLMConfigs();
const { llmConfigs, loading: configsLoading, refreshConfigs } = useLLMConfigs();
const { preferences, loading: preferencesLoading, isOnboardingComplete, refreshPreferences } = useLLMPreferences();
const [currentStep, setCurrentStep] = useState(1);
const [hasUserProgressed, setHasUserProgressed] = useState(false);
@ -173,7 +173,7 @@ const OnboardPage = () => {
exit={{ opacity: 0, x: -20 }}
transition={{ duration: 0.3 }}
>
{currentStep === 1 && <AddProviderStep />}
{currentStep === 1 && <AddProviderStep onConfigCreated={refreshConfigs} onConfigDeleted={refreshConfigs} />}
{currentStep === 2 && <AssignRolesStep onPreferencesUpdated={refreshPreferences} />}
{currentStep === 3 && <CompletionStep />}
</motion.div>

View file

@ -30,7 +30,12 @@ const LLM_PROVIDERS = [
{ value: 'CUSTOM', label: 'Custom Provider', example: 'your-custom-model' },
];
export function AddProviderStep() {
interface AddProviderStepProps {
onConfigCreated?: () => void;
onConfigDeleted?: () => void;
}
export function AddProviderStep({ onConfigCreated, onConfigDeleted }: AddProviderStepProps) {
const { llmConfigs, createLLMConfig, deleteLLMConfig } = useLLMConfigs();
const [isAddingNew, setIsAddingNew] = useState(false);
const [formData, setFormData] = useState<CreateLLMConfig>({
@ -70,6 +75,8 @@ export function AddProviderStep() {
litellm_params: {}
});
setIsAddingNew(false);
// Notify parent component that a config was created
onConfigCreated?.();
}
};
@ -114,7 +121,12 @@ export function AddProviderStep() {
<Button
variant="ghost"
size="sm"
onClick={() => deleteLLMConfig(config.id)}
onClick={async () => {
const success = await deleteLLMConfig(config.id);
if (success) {
onConfigDeleted?.();
}
}}
className="text-destructive hover:text-destructive"
>
<Trash2 className="w-4 h-4" />