50 lines
1.1 KiB
Go
50 lines
1.1 KiB
Go
package handler
|
|
|
|
import (
|
|
"net/http"
|
|
"os"
|
|
|
|
"github.com/zs/InsightReply/internal/service"
|
|
)
|
|
|
|
type AIHandler struct {
|
|
svc *service.AIService
|
|
}
|
|
|
|
func NewAIHandler(svc *service.AIService) *AIHandler {
|
|
return &AIHandler{svc: svc}
|
|
}
|
|
|
|
func (h *AIHandler) Test(w http.ResponseWriter, r *http.Request) {
|
|
// ...
|
|
}
|
|
|
|
func (h *AIHandler) Generate(w http.ResponseWriter, r *http.Request) {
|
|
var body struct {
|
|
TweetContent string `json:"tweet_content"`
|
|
Strategy string `json:"strategy"`
|
|
Identity string `json:"identity"`
|
|
}
|
|
|
|
if err := json.NewDecoder(r.Body).Decode(&body); err != nil {
|
|
SendError(w, http.StatusBadRequest, 4001, "Invalid request body")
|
|
return
|
|
}
|
|
|
|
if body.TweetContent == "" {
|
|
SendError(w, http.StatusBadRequest, 4002, "Tweet content is required")
|
|
return
|
|
}
|
|
|
|
ctx := r.Context()
|
|
reply, err := h.svc.GenerateReply(ctx, body.TweetContent, body.Strategy, body.Identity)
|
|
if err != nil {
|
|
SendError(w, http.StatusBadGateway, 5002, "Failed to generate AI reply: "+err.Error())
|
|
return
|
|
}
|
|
|
|
SendSuccess(w, map[string]string{
|
|
"reply": reply,
|
|
})
|
|
}
|