MaxKB/apps/models_provider/urls.py
2025-05-07 18:09:45 +08:00

27 lines
1.4 KiB
Python

import os
from django.urls import path
from . import views
app_name = "models_provider"
urlpatterns = [
path('provider', views.Provide.as_view()),
path('provider/model_type_list', views.Provide.ModelTypeList.as_view()),
path('provider/model_list', views.Provide.ModelList.as_view()),
path('provider/model_params_form', views.Provide.ModelParamsForm.as_view()),
path('provider/model_form', views.Provide.ModelForm.as_view()),
path('workspace/<str:workspace_id>/model', views.Model.as_view()),
path('workspace/<str:workspace_id>/model/<str:model_id>/model_params_form', views.Model.ModelParamsForm.as_view()),
path('workspace/<str:workspace_id>/model/<str:model_id>', views.Model.Operate.as_view()),
path('workspace/<str:workspace_id>/model/<str:model_id>/pause_download', views.Model.PauseDownload.as_view()),
path('workspace/<str:workspace_id>/model/<str:model_id>/meta', views.Model.ModelMeta.as_view()),
]
if os.environ.get('SERVER_NAME', 'web') == 'local_model':
urlpatterns += [
path('workspace/<str:workspace_id>/model/<str:model_id>/embed_documents', views.ModelApply.EmbedDocuments.as_view()),
path('workspace/<str:workspace_id>model/<str:model_id>/embed_query', views.ModelApply.EmbedQuery.as_view()),
path('workspace/<str:workspace_id>model/<str:model_id>/compress_documents', views.ModelApply.CompressDocuments.as_view()),
]