AzenKain
a61cf085ce
All checks were successful
Build and Release / release (push) Successful in 1m27s
71 lines
2.1 KiB
Go
71 lines
2.1 KiB
Go
package controllers
|
|
|
|
import (
|
|
"context"
|
|
"history-api/internal/dtos/request"
|
|
"history-api/internal/dtos/response"
|
|
"history-api/internal/services"
|
|
"history-api/pkg/validator"
|
|
"time"
|
|
|
|
"github.com/gofiber/fiber/v3"
|
|
)
|
|
|
|
type ChatbotController struct {
|
|
chatbotService services.ChatbotService
|
|
}
|
|
|
|
func NewChatbotController(chatbotService services.ChatbotService) *ChatbotController {
|
|
return &ChatbotController{
|
|
chatbotService: chatbotService,
|
|
}
|
|
}
|
|
|
|
// Chat godoc
|
|
// @Summary Ask the AI chatbot
|
|
// @Description Ask a history question based on project context or global knowledge using RAG
|
|
// @Tags Chatbot
|
|
// @Accept json
|
|
// @Produce json
|
|
// @Security BearerAuth
|
|
// @Param request body request.ChatbotDto true "Chatbot query"
|
|
// @Success 200 {object} response.CommonResponse{data=string} "Successful response with AI answer"
|
|
// @Failure 400 {object} response.CommonResponse "Invalid request"
|
|
// @Failure 500 {object} response.CommonResponse "Internal server error"
|
|
// @Router /chatbot/chat [post]
|
|
func (cx *ChatbotController) Chat(c fiber.Ctx) error {
|
|
ctx, cancel := context.WithTimeout(context.Background(), 120*time.Second)
|
|
defer cancel()
|
|
|
|
dto := &request.ChatbotDto{}
|
|
if err := validator.ValidateBodyDto(c, dto); err != nil {
|
|
return c.Status(fiber.StatusBadRequest).JSON(response.CommonResponse{
|
|
Status: false,
|
|
Errors: err,
|
|
})
|
|
}
|
|
|
|
claims := c.Locals("user").(*response.JWTClaims)
|
|
|
|
answer, err := cx.chatbotService.Chat(ctx, claims.UId, dto.ProjectID, dto.Question)
|
|
if err != nil {
|
|
// Trả về lỗi 429 (Too Many Requests) nếu hết lượt dùng
|
|
if err.Error() == "you have reached your daily limit of 10 questions. Please come back tomorrow" {
|
|
return c.Status(fiber.StatusTooManyRequests).JSON(response.CommonResponse{
|
|
Status: false,
|
|
Message: err.Error(),
|
|
})
|
|
}
|
|
|
|
return c.Status(fiber.StatusInternalServerError).JSON(response.CommonResponse{
|
|
Status: false,
|
|
Message: err.Error(),
|
|
})
|
|
}
|
|
|
|
return c.Status(fiber.StatusOK).JSON(response.CommonResponse{
|
|
Status: true,
|
|
Data: answer,
|
|
})
|
|
}
|