Create handler/chat.go

internal/handler/chat.go

@@ -0,0 +1,48 @@
+package handler
+
+import (
+	"net/http"
+
+	"github.com/gin-gonic/gin"
+	"zencoder-2api/internal/model"
+	"zencoder-2api/internal/service"
+)
+
+type ChatHandler struct {
+	svc *service.APIService
+}
+
+func NewChatHandler() *ChatHandler {
+	return &ChatHandler{svc: service.NewAPIService()}
+}
+
+func (h *ChatHandler) ChatCompletions(c *gin.Context) {
+	var req model.ChatCompletionRequest
+	if err := c.ShouldBindJSON(&req); err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
+		return
+	}
+
+	if req.Stream {
+		h.handleStream(c, &req)
+		return
+	}
+
+	resp, err := h.svc.Chat(&req)
+	if err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
+		return
+	}
+
+	c.JSON(http.StatusOK, resp)
+}
+
+func (h *ChatHandler) handleStream(c *gin.Context, req *model.ChatCompletionRequest) {
+	c.Writer.Header().Set("Content-Type", "text/event-stream")
+	c.Writer.Header().Set("Cache-Control", "no-cache")
+	c.Writer.Header().Set("Connection", "keep-alive")
+
+	if err := h.svc.ChatStream(req, c.Writer); err != nil {
+		c.SSEvent("error", err.Error())
+	}
+}