From 135bc0f0782f061138fe9cc6a8025c65206b9acf Mon Sep 17 00:00:00 2001 From: "C.C" Date: Wed, 6 Mar 2024 17:35:17 +0800 Subject: [PATCH] Update serve.go --- cli/serve.go | 21 +++++---------------- 1 file changed, 5 insertions(+), 16 deletions(-) diff --git a/cli/serve.go b/cli/serve.go index 234f3f385..b582f7371 100644 --- a/cli/serve.go +++ b/cli/serve.go @@ -117,32 +117,21 @@ func registerAIProvider(aiConfig *ai.Config) error { for name, provider := range aiConfig.Providers { switch name { case "azopenai": - err := ai.RegisterProvider(azopenai.NewProvider( + ai.RegisterProvider(azopenai.NewProvider( provider["api_key"], provider["api_endpoint"], provider["deployment_id"], provider["api_version"], )) - if err != nil { - return fmt.Errorf("failed to register azopenai provider: %w", err) - } - log.InfoStatusEvent(os.Stdout, "registered [%s] AI provider", name) case "gemini": - err := ai.RegisterProvider(gemini.NewProvider(provider["api_key"])) - if err != nil { - return fmt.Errorf("failed to register gemini provider: %w", err) - } - log.InfoStatusEvent(os.Stdout, "registered [%s] AI provider", name) + ai.RegisterProvider(gemini.NewProvider(provider["api_key"])) case "openai": - err := ai.RegisterProvider(openai.NewProvider(provider["api_key"], provider["model"])) - if err != nil { - return fmt.Errorf("failed to register openai provider: %w", err) - } - log.InfoStatusEvent(os.Stdout, "registered [%s] AI provider", name) + ai.RegisterProvider(openai.NewProvider(provider["api_key"], provider["model"])) default: - log.Warnf("unknown provider: %s", name) + log.WarningStatusEvent(os.Stdout, "unknown provider: %s", name) } } + log.InfoStatusEvent(os.Stdout, "registered [%s] AI provider", name) return nil }