rogee
42b8b6c7f9
Phase 1 of AI_FEATURE_ROADMAP.md: 1. config.yaml: Add captain configuration section (llm_provider, llm_model, llm_api_key, llm_base_url, embedding_model, embedding_dims, max_tokens, temperature). Secrets injected via GOCHAT_CAPTAIN_LLM_API_KEY env var. 2. RAG route registration: - bootstrap.go: instantiate RAGService + RAGHandler, add to Handlers - router.go: register POST /captain/rag/query + /captain/rag/index/:response_id - rag_handler.go: fix account_id param name (was "id", route uses "account_id") 3. CaptainConversationService activation: - Remove "_ = captainConversationService" ignore in bootstrap.go - Create CaptainConversationHandler with BuildResponse endpoint - Register POST /captain/conversations/:conversation_id/respond route - Fix account_id param name in handler Verified: - go build ./... passes - go vet passes on all modified packages - go test passes (llm + service packages) - Server starts with captain config loaded - RAG query/index routes return proper handler responses (not 404) - Conversation respond route returns proper skip for non-pending conversations - Existing captain routes unaffected (assistants list still works)
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("account_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"})
|
|
}
|