from rest_framework.routers import DefaultRouter

from apps.orgs.viewsets import OrganizationViewSet
from apps.users.viewsets import UserViewSet
from apps.mlstore.viewsets import AgentViewSet, AgentRunViewSet
from apps.onboarding.viewsets import OnboardingFlowViewSet, OnboardingPageViewSet, OnboardingFieldViewSet, OnboardingSessionViewSet

router = DefaultRouter()
router.register(r'user', UserViewSet, basename='user')
router.register(r'organization', OrganizationViewSet, basename='organization')
router.register(r'agent', AgentViewSet, basename='agent')
router.register(r'agent-run', AgentRunViewSet, basename='agent-run')
router.register(r'onboarding/flow', OnboardingFlowViewSet, basename='onboarding-flow')
router.register(r'onboarding/page', OnboardingPageViewSet, basename='onboarding-page')
router.register(r'onboarding/field', OnboardingFieldViewSet, basename='onboarding-field')
router.register(r'onboarding/session', OnboardingSessionViewSet, basename='onboarding-session')

urlpatterns = router.urls