rogee
aeddedf2a3
Restructure the monorepo into clear top-level directories: - backend/: Go module root (cmd, internal, pkg, configs, migrations, docs/swagger, scripts, tests, go.mod, Makefile, .air.toml) - deploy/: Docker (Dockerfile, docker-compose*), quickstart, fluentd - docs/: project documentation + reports/ (moved from repo root) - AGENTS.md: new AI coding-agent guide at repo root Update all references to the new layout: - Dockerfile: COPY backend/go.mod, COPY backend/ (context = repo root) - docker-compose files: context ../.., dockerfile deploy/docker/Dockerfile, env_file ../../.env, volume mounts ../../backend:/app - deploy/quickstart/compose.yaml: dockerfile deploy/docker/Dockerfile - CI: working-directory: backend for go commands, file deploy/docker/Dockerfile, coverage path backend/coverage.out, health_check backend/scripts/ - backend/Makefile: docker target uses -f ../deploy/docker/Dockerfile ../ - README: architecture tree, quickstart, config paths updated Move root stray scripts (rename_models.*, run_m11_tests.sh, verify_build.sh, gorm_bool_main.go) to backend/scripts/legacy/. All moves via git mv to preserve history. Build, vet, SQLite tests, and docker compose config verified.
66 lines
2.1 KiB
Go
66 lines
2.1 KiB
Go
package v1
|
|
|
|
import (
|
|
"net/http"
|
|
"strconv"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/gochat/gochat/internal/service"
|
|
applogger "github.com/gochat/gochat/pkg/logger"
|
|
"github.com/gochat/gochat/pkg/response"
|
|
)
|
|
|
|
// RAGHandler handles Captain RAG (Retrieval-Augmented Generation) Q&A endpoints.
|
|
// Reference: M12 PRD §Captain AI — RAG Knowledge Base Q&A
|
|
type RAGHandler struct {
|
|
svc *service.RAGService
|
|
}
|
|
|
|
// NewRAGHandler creates a new RAGHandler.
|
|
func NewRAGHandler(svc *service.RAGService) *RAGHandler {
|
|
return &RAGHandler{svc: svc}
|
|
}
|
|
|
|
// Query performs a RAG query against the knowledge base.
|
|
// POST /api/v1/accounts/:account_id/captain/rag/query
|
|
func (h *RAGHandler) Query(c *gin.Context) {
|
|
accountID, err := strconv.ParseUint(c.Param("id"), 10, 64)
|
|
if err != nil {
|
|
response.AbortWithStatusError(c, http.StatusBadRequest, "invalid_account_id", "Invalid account ID")
|
|
return
|
|
}
|
|
|
|
var req service.RAGQueryRequest
|
|
if err := c.ShouldBindJSON(&req); err != nil {
|
|
response.AbortWithStatusError(c, http.StatusBadRequest, "invalid_request", err.Error())
|
|
return
|
|
}
|
|
|
|
result, err := h.svc.Query(c.Request.Context(), uint(accountID), &req)
|
|
if err != nil {
|
|
applogger.L().Errorf("RAGHandler.Query failed: %v", err)
|
|
response.AbortWithStatusError(c, http.StatusInternalServerError, "query_failed", "Failed to process RAG query")
|
|
return
|
|
}
|
|
|
|
response.OK(c, result)
|
|
}
|
|
|
|
// IndexResponse indexes an assistant response for RAG retrieval.
|
|
// POST /api/v1/accounts/:account_id/captain/rag/index/:response_id
|
|
func (h *RAGHandler) IndexResponse(c *gin.Context) {
|
|
responseID, err := strconv.ParseUint(c.Param("response_id"), 10, 64)
|
|
if err != nil {
|
|
response.AbortWithStatusError(c, http.StatusBadRequest, "invalid_response_id", "Invalid response ID")
|
|
return
|
|
}
|
|
|
|
if err := h.svc.IndexResponse(c.Request.Context(), uint(responseID)); err != nil {
|
|
applogger.L().Errorf("RAGHandler.IndexResponse failed: %v", err)
|
|
response.AbortWithStatusError(c, http.StatusInternalServerError, "index_failed", "Failed to index response for RAG")
|
|
return
|
|
}
|
|
|
|
response.OK(c, map[string]string{"status": "indexed"})
|
|
}
|