姚章浩
40 lines
881 B
Go
40 lines
881 B
Go
package main
|
|
|
|
import (
|
|
"os"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestMainWiresReasoningModeRoutingAndCompletion(t *testing.T) {
|
|
mainSource, err := os.ReadFile("main.go")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
wiringSource, err := os.ReadFile("reasoning_wiring.go")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
mainText := string(mainSource)
|
|
for _, required := range []string{
|
|
`os.Getenv("DOBOTAI_QUICK_AGENT_ID")`,
|
|
`os.Getenv("DOBOTAI_REASONING_AGENT_ID")`,
|
|
`reasoningmode.NewService`,
|
|
`qa.WithReasoningModeFinalizer`,
|
|
`ReasoningRouter:`,
|
|
} {
|
|
if !strings.Contains(mainText, required) {
|
|
t.Errorf("main reasoning wiring missing %q", required)
|
|
}
|
|
}
|
|
for _, required := range []string{
|
|
"dingTalkReasoningRouter",
|
|
"ReasoningMessage",
|
|
"service.Prepare",
|
|
} {
|
|
if !strings.Contains(string(wiringSource), required) {
|
|
t.Errorf("reasoning adapter missing %q", required)
|
|
}
|
|
}
|
|
}
|