mirror of
https://github.com/CJackHwang/ds2api.git
synced 2026-05-05 08:55:28 +08:00
42 lines
1.0 KiB
Go
42 lines
1.0 KiB
Go
package gemini
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"github.com/go-chi/chi/v5"
|
|
|
|
"ds2api/internal/util"
|
|
)
|
|
|
|
var writeJSON = util.WriteJSON
|
|
|
|
type Handler struct {
|
|
Store ConfigReader
|
|
Auth AuthResolver
|
|
DS DeepSeekCaller
|
|
OpenAI OpenAIChatRunner
|
|
}
|
|
|
|
//nolint:unused // used by native Gemini stream/non-stream runtime helpers.
|
|
func (h *Handler) compatStripReferenceMarkers() bool {
|
|
if h == nil || h.Store == nil {
|
|
return true
|
|
}
|
|
return h.Store.CompatStripReferenceMarkers()
|
|
}
|
|
|
|
func RegisterRoutes(r chi.Router, h *Handler) {
|
|
r.Post("/v1beta/models/{model}:generateContent", h.GenerateContent)
|
|
r.Post("/v1beta/models/{model}:streamGenerateContent", h.StreamGenerateContent)
|
|
r.Post("/v1/models/{model}:generateContent", h.GenerateContent)
|
|
r.Post("/v1/models/{model}:streamGenerateContent", h.StreamGenerateContent)
|
|
}
|
|
|
|
func (h *Handler) GenerateContent(w http.ResponseWriter, r *http.Request) {
|
|
h.handleGenerateContent(w, r, false)
|
|
}
|
|
|
|
func (h *Handler) StreamGenerateContent(w http.ResponseWriter, r *http.Request) {
|
|
h.handleGenerateContent(w, r, true)
|
|
}
|