mirror of
https://github.com/Tencent/WeKnora.git
synced 2025-11-25 11:29:31 +08:00
fix: Fix init tenant
This commit is contained in:
@@ -1,9 +1,7 @@
|
|||||||
package handler
|
package handler
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"errors"
|
|
||||||
"net/http"
|
"net/http"
|
||||||
"os"
|
|
||||||
|
|
||||||
"github.com/gin-gonic/gin"
|
"github.com/gin-gonic/gin"
|
||||||
|
|
||||||
@@ -50,20 +48,6 @@ func (h *TestDataHandler) GetTestData(c *gin.Context) {
|
|||||||
|
|
||||||
logger.Info(ctx, "Start retrieving test data")
|
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)
|
tenantID := uint(types.InitDefaultTenantID)
|
||||||
logger.Debugf(ctx, "Test tenant ID environment variable: %d", tenantID)
|
logger.Debugf(ctx, "Test tenant ID environment variable: %d", tenantID)
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user