62 lines
1.9 KiB
Go
62 lines
1.9 KiB
Go
package router
|
|
|
|
import (
|
|
"net/http"
|
|
"strings"
|
|
|
|
"opencatd-open/pkg/claude"
|
|
"opencatd-open/pkg/google"
|
|
"opencatd-open/pkg/openai"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
func ChatHandler(c *gin.Context) {
|
|
var chatreq openai.ChatCompletionRequest
|
|
if err := c.ShouldBindJSON(&chatreq); err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
|
|
// if chatreq.Messages[len(chatreq.Messages)-1].Role == "user" {
|
|
// result, err := search.BingSearch(search.SearchParams{Query: string(chatreq.Messages[len(chatreq.Messages)-1].Content)})
|
|
// if err == nil {
|
|
// var msgs []openai.ChatCompletionMessage
|
|
// for i, m := range chatreq.Messages {
|
|
// var buf bytes.Buffer
|
|
// buf.WriteString("根据我提问的语言回答我,我将提供一些从搜索引擎获取的信息(以websearch:开头)。你自行判断是否使用搜索引擎获取的内容。不要原封不动照抄,根据你自己的知识库提炼信息之后回答我\n\n")
|
|
// if m.Role == "system" {
|
|
// buf.Write(m.Content)
|
|
// msgs = append(msgs, openai.ChatCompletionMessage{Role: m.Role, Content: buf.Bytes()})
|
|
// } else {
|
|
// msgs = append(msgs, openai.ChatCompletionMessage{Role: m.Role, Content: buf.Bytes()})
|
|
// }
|
|
// if i == len(chatreq.Messages)-1 {
|
|
// m.Content = append(m.Content, json.RawMessage("\n\nwebsearch:")...)
|
|
// m.Content = append(m.Content, json.RawMessage(result.(string))...)
|
|
// msgs = append(msgs, openai.ChatCompletionMessage{Role: m.Role, Content: m.Content})
|
|
// } else {
|
|
// msgs = append(msgs, openai.ChatCompletionMessage{Role: m.Role, Content: m.Content})
|
|
// }
|
|
|
|
// }
|
|
// chatreq.Messages = msgs
|
|
// }
|
|
// }
|
|
|
|
if strings.HasPrefix(chatreq.Model, "gpt") {
|
|
openai.ChatProxy(c, &chatreq)
|
|
return
|
|
}
|
|
|
|
if strings.HasPrefix(chatreq.Model, "claude") {
|
|
claude.ChatProxy(c, &chatreq)
|
|
return
|
|
}
|
|
|
|
if strings.HasPrefix(chatreq.Model, "gemini") {
|
|
google.ChatProxy(c, &chatreq)
|
|
return
|
|
}
|
|
}
|