diff --git a/internal/biz/router_test.go b/internal/biz/router_test.go index 2cef402..7d75674 100644 --- a/internal/biz/router_test.go +++ b/internal/biz/router_test.go @@ -56,13 +56,15 @@ func in() *AiRouterBiz { panic("加载配置失败") } allLogger := log.DefaultLogger() - manager := tools.NewManager(configConfig) + utilOllama := utils_ollama.NewUtilOllama(configConfig, allLogger) + manager := tools.NewManager(configConfig, utilOllama) + db, _ := utils.NewGormDb(configConfig) sessionImpl := impl.NewSessionImpl(db) sysImpl := impl.NewSysImpl(db) taskImpl := impl.NewTaskImpl(db) chatImpl := impl.NewChatImpl(db) - utilOllama := utils_ollama.NewUtilOllama(configConfig, allLogger) + routerBiz := NewAiRouterBiz(manager, sessionImpl, sysImpl, taskImpl, chatImpl, configConfig, utilOllama) return routerBiz