mirror of
https://github.com/BaaRai/ARTEX.git
synced 2026-11-13 11:40:53 +08:00
GRAB Racem
a6e3fd0a2e
FromEnv compared the raw value, so "Ollama" or "OpenAI" matched no case and fell through to the anthropic default. With ARTEX_LLM_PROVIDER=Ollama and an ANTHROPIC_API_KEY set, the Anthropic key was sent to the Ollama base URL.
38 lines
1.0 KiB
Go
38 lines
1.0 KiB
Go
package agent
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/Autumn-27/norma/llm"
|
|
)
|
|
|
|
// ARTEX_LLM_PROVIDER is matched regardless of case and surrounding spaces. A
|
|
// miss used to fall through to anthropic: "Ollama" then sent ANTHROPIC_API_KEY
|
|
// to the Ollama base URL.
|
|
func TestFromEnvProviderIsCaseInsensitive(t *testing.T) {
|
|
t.Setenv("ANTHROPIC_API_KEY", "sk-ant-secret")
|
|
t.Setenv("OPENAI_API_KEY", "sk-oai")
|
|
t.Setenv("ARTEX_LLM_BASE_URL", "")
|
|
t.Setenv("ARTEX_LLM_MODEL", "")
|
|
for _, tc := range []struct {
|
|
env string
|
|
want llm.Format
|
|
key string
|
|
}{
|
|
{"Ollama", FormatOllama, ""},
|
|
{" OLLAMA ", FormatOllama, ""},
|
|
{"OpenAI", llm.FormatOpenAI, "sk-oai"},
|
|
{"OpenAI-Responses", llm.FormatOpenAIResponses, "sk-oai"},
|
|
{"Anthropic", llm.FormatAnthropic, "sk-ant-secret"},
|
|
} {
|
|
t.Setenv("ARTEX_LLM_PROVIDER", tc.env)
|
|
c, ok := FromEnv()
|
|
if !ok {
|
|
t.Fatalf("%q: FromEnv returned ok=false", tc.env)
|
|
}
|
|
if c.Format != tc.want || c.APIKey != tc.key {
|
|
t.Fatalf("%q: format=%v key=%q, want format=%v key=%q", tc.env, c.Format, c.APIKey, tc.want, tc.key)
|
|
}
|
|
}
|
|
}
|