mirror of
https://github.com/Tencent/WeKnora.git
synced 2025-11-25 03:15:00 +08:00
fix: Fix init tenant
This commit is contained in:
@@ -1,9 +1,7 @@
|
||||
package handler
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"net/http"
|
||||
"os"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
|
||||
@@ -50,20 +48,6 @@ func (h *TestDataHandler) GetTestData(c *gin.Context) {
|
||||
|
||||
logger.Info(ctx, "Start retrieving test data")
|
||||
|
||||
// Check if we're running in release/production mode
|
||||
if gin.Mode() == gin.ReleaseMode {
|
||||
logger.Warn(ctx, "Attempting to retrieve test data in production mode")
|
||||
c.Error(errors.New("This API is only available in development mode"))
|
||||
return
|
||||
}
|
||||
|
||||
// Check if test data environment variables are configured
|
||||
if os.Getenv("INIT_TEST_TENANT_ID") == "" || os.Getenv("INIT_TEST_KNOWLEDGE_BASE_ID") == "" {
|
||||
logger.Warn(ctx, "Test data environment variables not set")
|
||||
c.Error(errors.New("Test data not enabled"))
|
||||
return
|
||||
}
|
||||
|
||||
tenantID := uint(types.InitDefaultTenantID)
|
||||
logger.Debugf(ctx, "Test tenant ID environment variable: %d", tenantID)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user