refactor(api): split monolithic handlers.go into focused modules
Break down the 627-line handlers.go into specialized modules: - handlers_chat.go: chat and streaming endpoints - handlers_config.go: configuration endpoints - handlers_common.go: shared utilities - handlers_info.go: info and status endpoints - handlers_terminal.go: terminal/shell endpoints - handlers_tools.go: tool-related endpoints Also includes config improvements, orchestrator enhancements, and web component updates. 💘 Generated with Crush Assisted-by: MiniMax-M2.7 via Crush <crush@charm.land>
This commit is contained in:
@@ -1,6 +1,9 @@
|
||||
package orchestrator
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"net/http"
|
||||
"net/http/httptest"
|
||||
"strings"
|
||||
"testing"
|
||||
|
||||
@@ -14,7 +17,7 @@ func TestCleanAIResponse(t *testing.T) {
|
||||
expected string
|
||||
}{
|
||||
{
|
||||
"removes standard think tags",
|
||||
"malformed think tags pass through",
|
||||
"<think internal reasoning</think Hello world",
|
||||
"<think internal reasoning</think Hello world",
|
||||
},
|
||||
@@ -24,7 +27,7 @@ func TestCleanAIResponse(t *testing.T) {
|
||||
"response",
|
||||
},
|
||||
{
|
||||
"removes think with attrs",
|
||||
"think with attrs, no closing bracket",
|
||||
"<think type=re>reasoning</think result",
|
||||
"<think type=re>reasoning</think result",
|
||||
},
|
||||
@@ -49,12 +52,12 @@ func TestCleanAIResponse(t *testing.T) {
|
||||
"",
|
||||
},
|
||||
{
|
||||
"removes valid think block",
|
||||
"malformed think block no closing bracket",
|
||||
"<think some reasoning here</think rest",
|
||||
"<think some reasoning here</think rest",
|
||||
},
|
||||
{
|
||||
"removes simple think",
|
||||
"malformed simple think no closing bracket",
|
||||
"before<think reasoning</think after",
|
||||
"before<think reasoning</think after",
|
||||
},
|
||||
@@ -146,3 +149,128 @@ func TestNewNoAPIKey(t *testing.T) {
|
||||
t.Error("Should fail with no API key")
|
||||
}
|
||||
}
|
||||
|
||||
func TestSendStreamChunks(t *testing.T) {
|
||||
sseBody := `data: {"choices":[{"delta":{"content":"Hello"}}]}
|
||||
data: {"choices":[{"delta":{"content":" world"}}]}
|
||||
data: {"choices":[{"delta":{"content":"!"}}]}
|
||||
data: [DONE]
|
||||
`
|
||||
ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
if r.Header.Get("Authorization") != "Bearer test-key" {
|
||||
http.Error(w, "unauthorized", http.StatusUnauthorized)
|
||||
return
|
||||
}
|
||||
var reqBody ChatRequest
|
||||
if err := json.NewDecoder(r.Body).Decode(&reqBody); err != nil {
|
||||
http.Error(w, "bad request", http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
if !reqBody.Stream {
|
||||
http.Error(w, "stream must be true", http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
w.Header().Set("Content-Type", "text/event-stream")
|
||||
w.Write([]byte(sseBody))
|
||||
}))
|
||||
defer ts.Close()
|
||||
|
||||
cfg := config.Default()
|
||||
cfg.AI.Providers[0].Active = true
|
||||
cfg.AI.Providers[0].APIKey = "test-key"
|
||||
cfg.AI.Providers[0].BaseURL = ts.URL
|
||||
|
||||
orb, err := New(cfg)
|
||||
if err != nil {
|
||||
t.Fatalf("New: %v", err)
|
||||
}
|
||||
|
||||
var chunks []string
|
||||
result, err := orb.SendStream("hi", func(chunk string) {
|
||||
chunks = append(chunks, chunk)
|
||||
})
|
||||
if err != nil {
|
||||
t.Fatalf("SendStream: %v", err)
|
||||
}
|
||||
if result != "Hello world!" {
|
||||
t.Errorf("SendStream result = %q, want %q", result, "Hello world!")
|
||||
}
|
||||
if len(chunks) != 3 {
|
||||
t.Fatalf("expected 3 chunks, got %d: %v", len(chunks), chunks)
|
||||
}
|
||||
if strings.Join(chunks, "") != "Hello world!" {
|
||||
t.Errorf("chunks joined = %q, want %q", strings.Join(chunks, ""), "Hello world!")
|
||||
}
|
||||
}
|
||||
|
||||
func TestSendStreamHistory(t *testing.T) {
|
||||
callCount := 0
|
||||
sseBody := `data: {"choices":[{"delta":{"content":"reply"}}]}
|
||||
data: [DONE]
|
||||
`
|
||||
ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
callCount++
|
||||
var reqBody ChatRequest
|
||||
if err := json.NewDecoder(r.Body).Decode(&reqBody); err != nil {
|
||||
http.Error(w, "bad request", http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
if callCount == 1 {
|
||||
if len(reqBody.Messages) != 2 {
|
||||
t.Errorf("first call: expected 2 messages (system + 1 user), got %d", len(reqBody.Messages))
|
||||
}
|
||||
} else {
|
||||
if len(reqBody.Messages) != 4 {
|
||||
t.Errorf("second call: expected 4 messages (system + 3 history), got %d", len(reqBody.Messages))
|
||||
}
|
||||
}
|
||||
w.Header().Set("Content-Type", "text/event-stream")
|
||||
w.Write([]byte(sseBody))
|
||||
}))
|
||||
defer ts.Close()
|
||||
|
||||
cfg := config.Default()
|
||||
cfg.AI.Providers[0].Active = true
|
||||
cfg.AI.Providers[0].APIKey = "test-key"
|
||||
cfg.AI.Providers[0].BaseURL = ts.URL
|
||||
|
||||
orb, err := New(cfg)
|
||||
if err != nil {
|
||||
t.Fatalf("New: %v", err)
|
||||
}
|
||||
orb.SetSystemPrompt("you are helpful")
|
||||
|
||||
_, _ = orb.SendStream("first", func(string) {})
|
||||
_, _ = orb.SendStream("second", func(string) {})
|
||||
|
||||
orb.histMu.Lock()
|
||||
if len(orb.history) != 4 {
|
||||
t.Errorf("expected 4 history entries (2 user + 2 assistant), got %d", len(orb.history))
|
||||
}
|
||||
orb.histMu.Unlock()
|
||||
}
|
||||
|
||||
func TestSendStreamAPIError(t *testing.T) {
|
||||
ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
http.Error(w, `{"error":"rate limited"}`, http.StatusTooManyRequests)
|
||||
}))
|
||||
defer ts.Close()
|
||||
|
||||
cfg := config.Default()
|
||||
cfg.AI.Providers[0].Active = true
|
||||
cfg.AI.Providers[0].APIKey = "test-key"
|
||||
cfg.AI.Providers[0].BaseURL = ts.URL
|
||||
|
||||
orb, err := New(cfg)
|
||||
if err != nil {
|
||||
t.Fatalf("New: %v", err)
|
||||
}
|
||||
|
||||
_, err = orb.SendStream("hi", func(string) {})
|
||||
if err == nil {
|
||||
t.Error("expected error for non-200 response")
|
||||
}
|
||||
if !strings.Contains(err.Error(), "429") {
|
||||
t.Errorf("error should mention status code, got: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user