reface to openteam

internal/controller/proxy/chat_proxy.go

@@ -0,0 +1,60 @@
+package controller
+
+import (
+	"fmt"
+	"net/http"
+	"opencatd-open/internal/dto"
+	"opencatd-open/llm"
+	"opencatd-open/llm/claude/v2"
+	"opencatd-open/llm/google/v2"
+	"opencatd-open/llm/openai_compatible"
+
+	"github.com/gin-gonic/gin"
+)
+
+func (h *Proxy) ChatHandler(c *gin.Context) {
+	var chatreq llm.ChatRequest
+	if err := c.ShouldBindJSON(&chatreq); err != nil {
+		c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
+		return
+	}
+
+	err := h.SelectApiKey(chatreq.Model)
+	if err != nil {
+		dto.WrapErrorAsOpenAI(c, 500, err.Error())
+		return
+	}
+
+	var llm llm.LLM
+	switch *h.apikey.ApiType {
+	case "claude":
+		llm, err = claude.NewClaude(h.apikey)
+	case "gemini":
+		llm, err = google.NewGemini(c, h.apikey)
+	case "openai", "azure", "github":
+		fallthrough
+	default:
+		llm, err = openai_compatible.NewOpenAICompatible(h.apikey)
+		if err != nil {
+			dto.WrapErrorAsOpenAI(c, 500, fmt.Errorf("create llm client error: %w", err).Error())
+			return
+		}
+	}
+
+	if !chatreq.Stream {
+		resp, err := llm.Chat(c, chatreq)
+		if err != nil {
+			dto.WrapErrorAsOpenAI(c, 500, err.Error())
+		}
+		c.JSON(http.StatusOK, resp)
+
+	} else {
+		datachan, err := llm.StreamChat(c, chatreq)
+		if err != nil {
+			dto.WrapErrorAsOpenAI(c, 500, err.Error())
+		}
+		for data := range datachan {
+			c.SSEvent("", data)
+		}
+	}
+}