152 lines
7.0 KiB
Go
152 lines
7.0 KiB
Go
package config
|
|
|
|
import (
|
|
"testing"
|
|
)
|
|
|
|
func TestLoadOllamaPoolAndGLPIKB(t *testing.T) {
|
|
t.Setenv("BRAIN_DATA_DIR", t.TempDir())
|
|
t.Setenv("OLLAMA_URLS", "http://gpu-a:11434,http://gpu-b:11434")
|
|
t.Setenv("OLLAMA_NODE_NAMES", "a,b")
|
|
t.Setenv("OLLAMA_NODE_WEIGHTS", "1,3")
|
|
t.Setenv("OLLAMA_ROUTING_MODE", "weighted")
|
|
t.Setenv("GLPI_KB_ENABLED", "true")
|
|
t.Setenv("GLPI_URL", "http://glpi.test")
|
|
t.Setenv("GLPI_ALLOW_INSECURE_HTTP", "true")
|
|
t.Setenv("GLPI_CLIENT_ID", "id")
|
|
t.Setenv("GLPI_CLIENT_SECRET", "secret")
|
|
t.Setenv("GLPI_USERNAME", "user")
|
|
t.Setenv("GLPI_PASSWORD", "password")
|
|
cfg, err := Load()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(cfg.OllamaURLs) != 2 || cfg.OllamaNodeNames[1] != "b" || cfg.OllamaNodeWeights[1] != 3 || cfg.OllamaRoutingMode != "weighted" {
|
|
t.Fatalf("unexpected pool config: %+v", cfg)
|
|
}
|
|
if !cfg.GLPIKBEnabled || cfg.GLPIURL != "http://glpi.test" {
|
|
t.Fatalf("unexpected GLPI config: %+v", cfg)
|
|
}
|
|
}
|
|
|
|
func TestLoadRejectsMismatchedPoolMetadata(t *testing.T) {
|
|
t.Setenv("BRAIN_DATA_DIR", t.TempDir())
|
|
t.Setenv("OLLAMA_URLS", "http://gpu-a:11434,http://gpu-b:11434")
|
|
t.Setenv("OLLAMA_NODE_NAMES", "only-one")
|
|
if _, err := Load(); err == nil {
|
|
t.Fatal("expected pool metadata validation error")
|
|
}
|
|
}
|
|
|
|
func TestLoadRuntimeControlDefaultsAndFilters(t *testing.T) {
|
|
t.Setenv("BRAIN_DATA_DIR", t.TempDir())
|
|
t.Setenv("BRAIN_LEARNING_ENABLED", "false")
|
|
t.Setenv("BRAIN_THINKING_ENABLED", "false")
|
|
t.Setenv("BRAIN_LEARNING_SOURCES", "GLPI Knowledge Base,internal-kb")
|
|
t.Setenv("BRAIN_DISPLAY_SOURCES", "GLPI Knowledge Base")
|
|
t.Setenv("BRAIN_THINKING_SOURCES", "internal-kb")
|
|
t.Setenv("BRAIN_DEFAULT_VIEW", "constellation")
|
|
t.Setenv("BRAIN_MAX_DISPLAY_NODES", "12000")
|
|
t.Setenv("BRAIN_LOW_POWER_MODE", "true")
|
|
cfg, err := Load()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if cfg.LearningEnabled || cfg.ThinkingEnabled || cfg.DefaultView != "constellation" || cfg.MaxDisplayNodes != 12000 || !cfg.LowPowerMode {
|
|
t.Fatalf("unexpected runtime defaults: %+v", cfg)
|
|
}
|
|
if len(cfg.LearningSources) != 2 || len(cfg.DisplaySources) != 1 || len(cfg.ThinkingSources) != 1 {
|
|
t.Fatalf("unexpected source defaults: %+v", cfg)
|
|
}
|
|
}
|
|
|
|
func TestLoadRejectsInvalidMaxDisplayNodes(t *testing.T) {
|
|
t.Setenv("BRAIN_DATA_DIR", t.TempDir())
|
|
t.Setenv("BRAIN_MAX_DISPLAY_NODES", "500001")
|
|
if _, err := Load(); err == nil {
|
|
t.Fatal("expected max display nodes validation error")
|
|
}
|
|
}
|
|
|
|
func TestLoadArticleSynthesisQualitySettings(t *testing.T) {
|
|
t.Setenv("BRAIN_DATA_DIR", t.TempDir())
|
|
t.Setenv("BRAIN_ARTICLE_SYNTHESIS_ENABLED", "true")
|
|
t.Setenv("BRAIN_ARTICLE_MIN_SOURCES", "4")
|
|
t.Setenv("BRAIN_ARTICLE_MAX_SOURCES", "7")
|
|
t.Setenv("BRAIN_ARTICLE_MIN_PRODUCTION_RATIO", "0.8")
|
|
t.Setenv("BRAIN_ARTICLE_MAX_GENERATION_DEPTH", "2")
|
|
t.Setenv("BRAIN_ARTICLE_MIN_CONFIDENCE", "0.81")
|
|
t.Setenv("BRAIN_ARTICLE_MIN_TEXT_CHARS", "220")
|
|
t.Setenv("BRAIN_ARTICLE_MIN_ANSWER_CHARS", "500")
|
|
t.Setenv("BRAIN_ARTICLE_LANGUAGE", "en-US")
|
|
cfg, err := Load()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if !cfg.ArticleSynthesisEnabled || cfg.ArticleMinSources != 4 || cfg.ArticleMaxSources != 7 || cfg.ArticleMinProductionRatio != 0.8 || cfg.ArticleMaxGenerationDepth != 2 || cfg.ArticleMinConfidence != 0.81 || cfg.ArticleMinTextChars != 220 || cfg.ArticleMinAnswerChars != 500 || cfg.ArticleLanguage != "en-US" {
|
|
t.Fatalf("unexpected article synthesis config: %+v", cfg)
|
|
}
|
|
}
|
|
|
|
func TestLoadIterativeResearchSettings(t *testing.T) {
|
|
t.Setenv("BRAIN_DATA_DIR", t.TempDir())
|
|
t.Setenv("BRAIN_ARTICLE_MAX_RESEARCH_QUERIES", "5")
|
|
t.Setenv("BRAIN_ARTICLE_RESEARCH_RESULTS", "10")
|
|
t.Setenv("BRAIN_ARTICLE_RESEARCH_ROUNDS", "4")
|
|
t.Setenv("BRAIN_ARTICLE_RESEARCH_FETCH_RESULTS", "3")
|
|
t.Setenv("BRAIN_ARTICLE_RESEARCH_EXPLORATION_RESULTS", "2")
|
|
t.Setenv("BRAIN_ARTICLE_RESEARCH_PREFETCH_MIN_RELEVANCE", "0.3")
|
|
t.Setenv("BRAIN_ARTICLE_RESEARCH_MIN_RELEVANCE", "0.7")
|
|
t.Setenv("BRAIN_ARTICLE_RESEARCH_MIN_QUALITY", "0.6")
|
|
t.Setenv("BRAIN_ARTICLE_RESEARCH_PAGE_MAX_BYTES", "1048576")
|
|
t.Setenv("BRAIN_ARTICLE_RESEARCH_PAGE_MAX_CHARS", "9000")
|
|
t.Setenv("BRAIN_ARTICLE_RESEARCH_FETCH_TIMEOUT", "15s")
|
|
t.Setenv("BRAIN_ARTICLE_RESEARCH_ALLOW_PRIVATE", "true")
|
|
t.Setenv("BRAIN_RESEARCH_DEDUPE_THRESHOLD", "0.94")
|
|
t.Setenv("BRAIN_RESEARCH_DEDUPE_TTL", "30m")
|
|
t.Setenv("BRAIN_RESEARCH_OLLAMA_MAX_INFLIGHT", "3")
|
|
t.Setenv("BRAIN_RESEARCH_OLLAMA_QUEUE_SIZE", "24")
|
|
cfg, err := Load()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if cfg.ArticleMaxResearchQueries != 5 || cfg.ArticleResearchResults != 10 || cfg.ArticleResearchRounds != 4 || cfg.ArticleResearchFetchResults != 3 || cfg.ArticleResearchExplorationResults != 2 || cfg.ArticleResearchPrefetchMinRelevance != .3 || cfg.ArticleResearchMinRelevance != .7 || cfg.ArticleResearchMinQuality != .6 || cfg.ArticleResearchPageMaxBytes != 1048576 || cfg.ArticleResearchPageMaxChars != 9000 || cfg.ArticleResearchFetchTimeout.String() != "15s" || !cfg.ArticleResearchAllowPrivate || cfg.ResearchDedupeThreshold != .94 || cfg.ResearchDedupeTTL.String() != "30m0s" || cfg.ResearchOllamaMaxInflight != 3 || cfg.ResearchOllamaQueueSize != 24 {
|
|
t.Fatalf("unexpected iterative research config: %+v", cfg)
|
|
}
|
|
}
|
|
|
|
func TestLoadAutonomousResearchSettings(t *testing.T) {
|
|
t.Setenv("BRAIN_DATA_DIR", t.TempDir())
|
|
t.Setenv("SEARXNG_URL", "http://searxng:8080")
|
|
t.Setenv("BRAIN_AUTONOMOUS_RESEARCH_ENABLED", "true")
|
|
t.Setenv("BRAIN_AUTONOMOUS_RESEARCH_IDLE_ONLY", "false")
|
|
t.Setenv("BRAIN_AUTONOMOUS_RESEARCH_INTERVAL", "12m")
|
|
t.Setenv("BRAIN_AUTONOMOUS_RESEARCH_TASKS_PER_CYCLE", "2")
|
|
t.Setenv("BRAIN_AUTONOMOUS_RESEARCH_MAX_TASKS_PER_DAY", "18")
|
|
t.Setenv("BRAIN_AUTONOMOUS_RESEARCH_MAX_QUERIES_PER_TASK", "7")
|
|
t.Setenv("BRAIN_AUTONOMOUS_RESEARCH_MAX_PAGES_PER_TASK", "9")
|
|
t.Setenv("BRAIN_AUTONOMOUS_RESEARCH_MAX_ROUNDS", "4")
|
|
t.Setenv("BRAIN_AUTONOMOUS_RESEARCH_MIN_PRIORITY", "0.72")
|
|
t.Setenv("BRAIN_AUTONOMOUS_RESEARCH_COOLDOWN", "120h")
|
|
t.Setenv("BRAIN_AUTONOMOUS_RESEARCH_LEASE", "35m")
|
|
t.Setenv("BRAIN_AUTONOMOUS_RESEARCH_MAX_ATTEMPTS", "4")
|
|
t.Setenv("BRAIN_AUTONOMOUS_RESEARCH_QUERY_TRIGGERS", "false")
|
|
t.Setenv("BRAIN_AUTONOMOUS_RESEARCH_OPPORTUNITY_LIMIT", "11")
|
|
cfg, err := Load()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if !cfg.AutonomousResearchEnabled || cfg.AutonomousResearchIdleOnly || cfg.AutonomousResearchInterval.String() != "12m0s" || cfg.AutonomousResearchTasksPerCycle != 2 || cfg.AutonomousResearchMaxTasksPerDay != 18 || cfg.AutonomousResearchMaxQueriesPerTask != 7 || cfg.AutonomousResearchMaxPagesPerTask != 9 || cfg.AutonomousResearchMaxRounds != 4 || cfg.AutonomousResearchMinPriority != .72 || cfg.AutonomousResearchCooldown.String() != "120h0m0s" || cfg.AutonomousResearchLease.String() != "35m0s" || cfg.AutonomousResearchMaxAttempts != 4 || cfg.AutonomousResearchQueryTriggers || cfg.AutonomousResearchOpportunityLimit != 11 {
|
|
t.Fatalf("unexpected autonomous research config: %+v", cfg)
|
|
}
|
|
}
|
|
|
|
func TestLoadRejectsAutonomousResearchWithoutSearXNG(t *testing.T) {
|
|
t.Setenv("BRAIN_DATA_DIR", t.TempDir())
|
|
t.Setenv("BRAIN_AUTONOMOUS_RESEARCH_ENABLED", "true")
|
|
t.Setenv("SEARXNG_URL", "")
|
|
if _, err := Load(); err == nil {
|
|
t.Fatal("expected SEARXNG_URL validation error")
|
|
}
|
|
}
|