v0.11.6 现在 --endpoint 参数不指定 http(s) 也能正常工作了

MoonshotAI · Sep 11, 2024 · 4eeb2cc · 4eeb2cc
1 parent b0e2cfc
commit 4eeb2cc
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/endpoint_custom.go b/endpoint_custom.go
@@ -4,6 +4,7 @@
 package main
 
 import (
+	"net/url"
 	"strings"
 
 	"github.com/spf13/cobra"
@@ -18,6 +19,12 @@ func init() {
 		if !flags.Changed("endpoint") && MoonConfig.Endpoint != "" {
 			endpoint = MoonConfig.Endpoint
 		}
+		if eUrl, err := url.Parse(endpoint); err == nil {
+			if eUrl.Scheme == "" {
+				eUrl.Scheme = "https"
+			}
+			endpoint = eUrl.String()
+		}
 		endpoint = strings.TrimSuffix(endpoint, "/")
 	})
 }
diff --git a/main.go b/main.go
@@ -17,7 +17,7 @@ func init() {
 var (
 	MoonPalace = &cobra.Command{
 		Use:           "moonpalace",
-		Version:       "v0.11.5",
+		Version:       "v0.11.6",
 		Short:         "MoonPalace is a command-line tool for debugging the Moonshot AI HTTP API",
 		SilenceErrors: true,
 		SilenceUsage:  true,