diff --git a/internal/admin/service.go b/internal/admin/service.go index 2179814d73..536f0b0c13 100644 --- a/internal/admin/service.go +++ b/internal/admin/service.go @@ -64,6 +64,21 @@ type Service struct { ingestionTaskDAO *dao.IngestionTaskDAO ingestionTaskLogDao *dao.IngestionTaskLogDAO ingestionTaskSvc *servicepkg.IngestionTaskService + + enterpriseUserDAO *dao.EnterpriseUserDAO + enterpriseTaskDAO *dao.EnterpriseTaskDAO + enterpriseFileDAO *dao.EnterpriseFileDAO + enterpriseAPITokenDAO *dao.EnterpriseAPITokenDAO + enterpriseBillingProductDAO *dao.EnterpriseBillingProductDAO + enterpriseBillingSubscriptionDAO *dao.EnterpriseBillingSubscriptionDAO + enterpriseChatDAO *dao.EnterpriseChatDAO + enterpriseDocumentDAO *dao.EnterpriseDocumentDAO + enterpriseMemoryDAO *dao.EnterpriseMemoryDAO + enterpriseSearchDAO *dao.EnterpriseSearchDAO + enterpriseTenantDAO *dao.EnterpriseTenantDAO + enterpriseUserCanvasDAO *dao.EnterpriseUserCanvasDAO + enterpriseUserTenantDAO *dao.EnterpriseUserTenantDAO + enterpriseKnowledgebaseDAO *dao.EnterpriseKnowledgebaseDAO } // NewService create admin service @@ -89,6 +104,21 @@ func NewService() *Service { ingestionTaskDAO: dao.NewIngestionTaskDAO(), ingestionTaskLogDao: dao.NewIngestionTaskLogDAO(), ingestionTaskSvc: servicepkg.NewIngestionTaskService(), + + enterpriseUserDAO: dao.NewEnterpriseUserDAO(), + enterpriseTaskDAO: dao.NewEnterpriseTaskDAO(), + enterpriseFileDAO: dao.NewEnterpriseFileDAO(), + enterpriseAPITokenDAO: dao.NewEnterpriseAPITokenDAO(), + enterpriseBillingProductDAO: dao.NewEnterpriseBillingProductDAO(), + enterpriseBillingSubscriptionDAO: dao.NewEnterpriseBillingSubscriptionDAO(), + enterpriseChatDAO: dao.NewEnterpriseChatDAO(), + enterpriseDocumentDAO: dao.NewEnterpriseDocumentDAO(), + enterpriseMemoryDAO: dao.NewEnterpriseMemoryDAO(), + enterpriseSearchDAO: dao.NewEnterpriseSearchDAO(), + enterpriseTenantDAO: dao.NewEnterpriseTenantDAO(), + enterpriseUserCanvasDAO: dao.NewEnterpriseUserCanvasDAO(), + enterpriseUserTenantDAO: dao.NewEnterpriseUserTenantDAO(), + enterpriseKnowledgebaseDAO: dao.NewEnterpriseKnowledgebaseDAO(), } } diff --git a/internal/dao/api_token_ee.go b/internal/dao/api_token_ee.go index 0e180b9abc..e6ce3cb316 100644 --- a/internal/dao/api_token_ee.go +++ b/internal/dao/api_token_ee.go @@ -15,3 +15,9 @@ // package dao + +type EnterpriseAPITokenDAO struct{} + +func NewEnterpriseAPITokenDAO() *EnterpriseAPITokenDAO { + return &EnterpriseAPITokenDAO{} +} diff --git a/internal/dao/billing_product_ee.go b/internal/dao/billing_product_ee.go index 0e180b9abc..666dd62782 100644 --- a/internal/dao/billing_product_ee.go +++ b/internal/dao/billing_product_ee.go @@ -15,3 +15,9 @@ // package dao + +type EnterpriseBillingProductDAO struct{} + +func NewEnterpriseBillingProductDAO() *EnterpriseBillingProductDAO { + return &EnterpriseBillingProductDAO{} +} diff --git a/internal/dao/billing_subscription_ee.go b/internal/dao/billing_subscription_ee.go index 0e180b9abc..180a85d0c0 100644 --- a/internal/dao/billing_subscription_ee.go +++ b/internal/dao/billing_subscription_ee.go @@ -15,3 +15,9 @@ // package dao + +type EnterpriseBillingSubscriptionDAO struct{} + +func NewEnterpriseBillingSubscriptionDAO() *EnterpriseBillingSubscriptionDAO { + return &EnterpriseBillingSubscriptionDAO{} +} diff --git a/internal/dao/chat_ee.go b/internal/dao/chat_ee.go index 0e180b9abc..34b4f6a80b 100644 --- a/internal/dao/chat_ee.go +++ b/internal/dao/chat_ee.go @@ -15,3 +15,9 @@ // package dao + +type EnterpriseChatDAO struct{} + +func NewEnterpriseChatDAO() *EnterpriseChatDAO { + return &EnterpriseChatDAO{} +} diff --git a/internal/dao/document_ee.go b/internal/dao/document_ee.go index 0e180b9abc..17ab318b96 100644 --- a/internal/dao/document_ee.go +++ b/internal/dao/document_ee.go @@ -15,3 +15,9 @@ // package dao + +type EnterpriseDocumentDAO struct{} + +func NewEnterpriseDocumentDAO() *EnterpriseDocumentDAO { + return &EnterpriseDocumentDAO{} +} diff --git a/internal/dao/file_ee.go b/internal/dao/file_ee.go index 0e180b9abc..9c4e920b62 100644 --- a/internal/dao/file_ee.go +++ b/internal/dao/file_ee.go @@ -15,3 +15,9 @@ // package dao + +type EnterpriseFileDAO struct{} + +func NewEnterpriseFileDAO() *EnterpriseFileDAO { + return &EnterpriseFileDAO{} +} diff --git a/internal/dao/kb_ee.go b/internal/dao/kb_ee.go index 0e180b9abc..8f20e22c65 100644 --- a/internal/dao/kb_ee.go +++ b/internal/dao/kb_ee.go @@ -15,3 +15,9 @@ // package dao + +type EnterpriseKnowledgebaseDAO struct{} + +func NewEnterpriseKnowledgebaseDAO() *EnterpriseKnowledgebaseDAO { + return &EnterpriseKnowledgebaseDAO{} +} diff --git a/internal/dao/memory_ee.go b/internal/dao/memory_ee.go index 0e180b9abc..2d1616406a 100644 --- a/internal/dao/memory_ee.go +++ b/internal/dao/memory_ee.go @@ -15,3 +15,9 @@ // package dao + +type EnterpriseMemoryDAO struct{} + +func NewEnterpriseMemoryDAO() *EnterpriseMemoryDAO { + return &EnterpriseMemoryDAO{} +} diff --git a/internal/dao/search_ee.go b/internal/dao/search_ee.go index 0e180b9abc..51dfa6291b 100644 --- a/internal/dao/search_ee.go +++ b/internal/dao/search_ee.go @@ -15,3 +15,9 @@ // package dao + +type EnterpriseSearchDAO struct{} + +func NewEnterpriseSearchDAO() *EnterpriseSearchDAO { + return &EnterpriseSearchDAO{} +} diff --git a/internal/dao/task_ee.go b/internal/dao/task_ee.go index 0e180b9abc..65789bcf7a 100644 --- a/internal/dao/task_ee.go +++ b/internal/dao/task_ee.go @@ -15,3 +15,9 @@ // package dao + +type EnterpriseTaskDAO struct{} + +func NewEnterpriseTaskDAO() *EnterpriseTaskDAO { + return &EnterpriseTaskDAO{} +} diff --git a/internal/dao/tenant_ee.go b/internal/dao/tenant_ee.go index 0e180b9abc..ff35cc173e 100644 --- a/internal/dao/tenant_ee.go +++ b/internal/dao/tenant_ee.go @@ -15,3 +15,9 @@ // package dao + +type EnterpriseTenantDAO struct{} + +func NewEnterpriseTenantDAO() *EnterpriseTenantDAO { + return &EnterpriseTenantDAO{} +} diff --git a/internal/dao/user_canvas_ee.go b/internal/dao/user_canvas_ee.go index 0e180b9abc..d7b65042b8 100644 --- a/internal/dao/user_canvas_ee.go +++ b/internal/dao/user_canvas_ee.go @@ -15,3 +15,9 @@ // package dao + +type EnterpriseUserCanvasDAO struct{} + +func NewEnterpriseUserCanvasDAO() *EnterpriseUserCanvasDAO { + return &EnterpriseUserCanvasDAO{} +} diff --git a/internal/dao/user_ee.go b/internal/dao/user_ee.go index 0e180b9abc..52c979cdba 100644 --- a/internal/dao/user_ee.go +++ b/internal/dao/user_ee.go @@ -15,3 +15,9 @@ // package dao + +type EnterpriseUserDAO struct{} + +func NewEnterpriseUserDAO() *EnterpriseUserDAO { + return &EnterpriseUserDAO{} +} diff --git a/internal/dao/user_tenant_ee.go b/internal/dao/user_tenant_ee.go index 0e180b9abc..8143af0bce 100644 --- a/internal/dao/user_tenant_ee.go +++ b/internal/dao/user_tenant_ee.go @@ -15,3 +15,9 @@ // package dao + +type EnterpriseUserTenantDAO struct{} + +func NewEnterpriseUserTenantDAO() *EnterpriseUserTenantDAO { + return &EnterpriseUserTenantDAO{} +}