fix: Fix init tenant

This commit is contained in:
wizardchen
2025-08-10 22:06:15 +08:00
committed by lyingbug
parent df65b8bcd1
commit 4baf6b7b8b

View File

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