root
37b0b32327
- Implemented QuestionExpansionService for generating and materializing questions based on combinations and AI distillation. - Added question metadata classification logic to infer intent and layer of questions. - Created API handlers for question expansion operations including combination preview, AI distillation, metadata classification, and materialization. - Introduced database migrations to support new question-related fields and constraints in brand_questions and brand_keywords tables. - Added caching mechanism for AI distillation results to improve performance. - Defined JSON schemas for question distillation responses to ensure data integrity.
103 lines
2.6 KiB
Go
103 lines
2.6 KiB
Go
package transport
|
|
|
|
import (
|
|
"strconv"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
|
|
"github.com/geo-platform/tenant-api/internal/bootstrap"
|
|
"github.com/geo-platform/tenant-api/internal/shared/response"
|
|
"github.com/geo-platform/tenant-api/internal/tenant/app"
|
|
)
|
|
|
|
type QuestionExpansionHandler struct {
|
|
svc *app.QuestionExpansionService
|
|
}
|
|
|
|
func NewQuestionExpansionHandler(a *bootstrap.App) *QuestionExpansionHandler {
|
|
return &QuestionExpansionHandler{svc: a.QuestionExpansion}
|
|
}
|
|
|
|
func (h *QuestionExpansionHandler) CombinationPreview(c *gin.Context) {
|
|
brandID, ok := parseBrandIDParam(c)
|
|
if !ok {
|
|
return
|
|
}
|
|
var req app.QuestionCombinationRequest
|
|
if err := c.ShouldBindJSON(&req); err != nil {
|
|
response.Error(c, response.ErrBadRequest(40001, "invalid_params", err.Error()))
|
|
return
|
|
}
|
|
data, err := h.svc.GenerateByCombination(c.Request.Context(), brandID, req)
|
|
if err != nil {
|
|
response.Error(c, err)
|
|
return
|
|
}
|
|
response.Success(c, data)
|
|
}
|
|
|
|
func (h *QuestionExpansionHandler) AIDistill(c *gin.Context) {
|
|
brandID, ok := parseBrandIDParam(c)
|
|
if !ok {
|
|
return
|
|
}
|
|
var req app.QuestionDistillRequest
|
|
if err := c.ShouldBindJSON(&req); err != nil {
|
|
response.Error(c, response.ErrBadRequest(40001, "invalid_params", err.Error()))
|
|
return
|
|
}
|
|
data, err := h.svc.GenerateByAIDistill(c.Request.Context(), brandID, req)
|
|
if err != nil {
|
|
response.Error(c, err)
|
|
return
|
|
}
|
|
response.Success(c, data)
|
|
}
|
|
|
|
func (h *QuestionExpansionHandler) ClassifyMetadata(c *gin.Context) {
|
|
brandID, ok := parseBrandIDParam(c)
|
|
if !ok {
|
|
return
|
|
}
|
|
var req struct {
|
|
Texts []string `json:"texts" binding:"required,min=1"`
|
|
}
|
|
if err := c.ShouldBindJSON(&req); err != nil {
|
|
response.Error(c, response.ErrBadRequest(40001, "invalid_params", err.Error()))
|
|
return
|
|
}
|
|
data, err := h.svc.ClassifyMetadata(c.Request.Context(), brandID, req.Texts)
|
|
if err != nil {
|
|
response.Error(c, err)
|
|
return
|
|
}
|
|
response.Success(c, data)
|
|
}
|
|
|
|
func (h *QuestionExpansionHandler) Materialize(c *gin.Context) {
|
|
brandID, ok := parseBrandIDParam(c)
|
|
if !ok {
|
|
return
|
|
}
|
|
var req app.MaterializeQuestionsRequest
|
|
if err := c.ShouldBindJSON(&req); err != nil {
|
|
response.Error(c, response.ErrBadRequest(40001, "invalid_params", err.Error()))
|
|
return
|
|
}
|
|
data, err := h.svc.MaterializeQuestions(c.Request.Context(), brandID, req)
|
|
if err != nil {
|
|
response.Error(c, err)
|
|
return
|
|
}
|
|
response.Success(c, data)
|
|
}
|
|
|
|
func parseBrandIDParam(c *gin.Context) (int64, bool) {
|
|
brandID, err := strconv.ParseInt(c.Param("id"), 10, 64)
|
|
if err != nil || brandID <= 0 {
|
|
response.Error(c, response.ErrBadRequest(40001, "invalid_id", "brand id must be a number"))
|
|
return 0, false
|
|
}
|
|
return brandID, true
|
|
}
|