lisone/ds2api
1
0
Fork 0
mirror of https://github.com/CJackHwang/ds2api.git synced 2026-05-05 00:45:29 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor: introduce a public app package to expose the internal/server router, resolving internal package import restrictions.

Browse Source
This commit is contained in:
CJACK
2026-02-16 19:11:55 +08:00
parent f82a7e3e3c
commit 190881f13a
2 changed files with 19 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
internal/adapter/claude/handler.go
View File

@@ -232,17 +232,18 @@ func (h *Handler) writeClaudeStream(w http.ResponseWriter, r *http.Request, mode
w.Header().Set("Content-Type", "text/event-stream")
w.Header().Set("Cache-Control", "no-cache")
w.Header().Set("Connection", "keep-alive")
flusher, ok := w.(http.Flusher)
if !ok {
writeJSON(w, http.StatusInternalServerError, map[string]any{"error": map[string]any{"type": "api_error", "message": "streaming unsupported"}})
return
flusher, hasFlusher := w.(http.Flusher)
if !hasFlusher {
config.Logger.Warn("[claude_stream] response writer does not support flush; falling back to buffered SSE")
}
send := func(v any) {
b, _ := json.Marshal(v)
_, _ = w.Write([]byte("data: "))
_, _ = w.Write(b)
_, _ = w.Write([]byte("\n\n"))
flusher.Flush()
if hasFlusher {
flusher.Flush()
}
}
messageID := fmt.Sprintf("msg_%d", time.Now().UnixNano())
inputTokens := util.EstimateTokens(fmt.Sprintf("%v", messages))

21
internal/adapter/openai/handler.go
View File

@@ -193,10 +193,9 @@ func (h *Handler) handleStream(w http.ResponseWriter, r *http.Request, resp *htt
w.Header().Set("Content-Type", "text/event-stream")
w.Header().Set("Cache-Control", "no-cache")
w.Header().Set("Connection", "keep-alive")
flusher, ok := w.(http.Flusher)
if !ok {
writeOpenAIError(w, http.StatusInternalServerError, "streaming unsupported")
return
flusher, hasFlusher := w.(http.Flusher)
if !hasFlusher {
config.Logger.Warn("[stream] response writer does not support flush; falling back to buffered SSE")
}
lines := make(chan []byte, 128)
@@ -233,11 +232,15 @@ func (h *Handler) handleStream(w http.ResponseWriter, r *http.Request, resp *htt
_, _ = w.Write([]byte("data: "))
_, _ = w.Write(b)
_, _ = w.Write([]byte("\n\n"))
flusher.Flush()
if hasFlusher {
flusher.Flush()
}
}
sendDone := func() {
_, _ = w.Write([]byte("data: [DONE]\n\n"))
flusher.Flush()
if hasFlusher {
flusher.Flush()
}
}
finalize := func(finishReason string) {
@@ -313,8 +316,10 @@ func (h *Handler) handleStream(w http.ResponseWriter, r *http.Request, resp *htt
finalize("stop")
return
}
_, _ = w.Write([]byte(": keep-alive\n\n"))
flusher.Flush()
if hasFlusher {
_, _ = w.Write([]byte(": keep-alive\n\n"))
flusher.Flush()
}
case line, ok := <-lines:
if !ok {
// Ensure scanner completion is observed only after all queued