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

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')

urlpatterns = router.urls