fix: resolve retired/preview gemini model routing and test configuration errors
This commit is contained in:
@@ -0,0 +1,31 @@
|
||||
package server
|
||||
|
||||
import "testing"
|
||||
|
||||
func TestIsSoftwareDevelopment(t *testing.T) {
|
||||
tests := []struct {
|
||||
message string
|
||||
expected bool
|
||||
}{
|
||||
{"can you check the logs? it looks like a lot of requests are being routed to kimi-k2.7-code when they don't need to be", false},
|
||||
{"it looks like its still routing to kimi when it shouldn't need to", false},
|
||||
{"Write a python script to parse logs", true},
|
||||
{"Search the web for weather in New York", false},
|
||||
{"How to build a compiler in Go", true},
|
||||
{"Check my vscode config", false},
|
||||
{"Let's decode this barcode", false},
|
||||
{"go ahead and email both docxs and the ppt to kayla\n\n<memory-context>... HELPipedia_Telethon_Speaker_Script.docx ... mental health program specialist ...", false},
|
||||
{"academic program development", false},
|
||||
{"write a script for the video presentation", false},
|
||||
{"How to write a bash script", true},
|
||||
{"refactor this sql query", true},
|
||||
{"can you organize Downloads and Documents real quick?\n\n<memory-context>\nCLI wrapper: ~/Projects/rag-engine/rag (bash script activating .venv). Python venv@~/Projects/rag-engine/.venv/.\n</memory-context>", false},
|
||||
}
|
||||
|
||||
for _, tt := range tests {
|
||||
result := isSoftwareDevelopment(tt.message)
|
||||
if result != tt.expected {
|
||||
t.Errorf("isSoftwareDevelopment(%q) = %v; expected %v", tt.message, result, tt.expected)
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user