AzenKain
600246426b
All checks were successful
Build and Release / release (push) Successful in 1m30s
119 lines
3.6 KiB
Go
119 lines
3.6 KiB
Go
package controllers
|
|
|
|
import (
|
|
"context"
|
|
"history-api/internal/dtos/request"
|
|
"history-api/internal/dtos/response"
|
|
"history-api/internal/models"
|
|
"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 {
|
|
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,
|
|
})
|
|
}
|
|
|
|
// GetHistory godoc
|
|
// @Summary Get chatbot history
|
|
// @Description Get chatbot history for the current user
|
|
// @Tags Chatbot
|
|
// @Accept json
|
|
// @Produce json
|
|
// @Security BearerAuth
|
|
// @Param cursor query string false "Cursor ID for pagination"
|
|
// @Param limit query int false "Limit number of items returned, default 10"
|
|
// @Success 200 {object} response.CommonResponse{data=response.GetChatbotHistoryResponse} "Successful response"
|
|
// @Failure 400 {object} response.CommonResponse "Invalid request"
|
|
// @Failure 500 {object} response.CommonResponse "Internal server error"
|
|
// @Router /chatbot/history [get]
|
|
func (cx *ChatbotController) GetHistory(c fiber.Ctx) error {
|
|
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
|
|
defer cancel()
|
|
|
|
dto := &request.GetChatbotHistoryDto{}
|
|
if err := validator.ValidateQueryDto(c, dto); err != nil {
|
|
return c.Status(fiber.StatusBadRequest).JSON(response.CommonResponse{
|
|
Status: false,
|
|
Errors: err,
|
|
})
|
|
}
|
|
|
|
uid := c.Locals("uid").(string)
|
|
histories, err := cx.chatbotService.GetHistory(ctx, uid, dto)
|
|
if err != nil {
|
|
return c.Status(fiber.StatusInternalServerError).JSON(response.CommonResponse{
|
|
Status: false,
|
|
Message: err.Error(),
|
|
})
|
|
}
|
|
|
|
var preCursor string
|
|
if len(histories) > 0 {
|
|
preCursor = histories[0].ID
|
|
}
|
|
|
|
return c.Status(fiber.StatusOK).JSON(response.CommonResponse{
|
|
Status: true,
|
|
Data: response.GetChatbotHistoryResponse{
|
|
Items: models.ChatbotHistoryEntitiesToResponse(histories),
|
|
PreCursor: preCursor,
|
|
},
|
|
})
|
|
}
|