perf: support GPT setting

jumpserver · Dec 11, 2023 · 64efe74 · 64efe74
1 parent c8a4a5d
commit 64efe74
Show file tree

Hide file tree

Showing 4 changed files with 78 additions and 24 deletions.
diff --git a/cmd/impl/jms.go b/cmd/impl/jms.go
@@ -363,10 +363,15 @@ func (j *JMServer) GetPublicSetting(ctx context.Context, empty *pb.Empty) (*pb.P
 		status.Err = err.Error()
 		return &pb.PublicSettingResponse{Status: &status}, nil
 	}
+	setting := j.uploader.GetTerminalSetting()
 	status.Ok = true
 	pbSetting := pb.PublicSetting{
 		XpackEnabled: data.XpackEnabled,
 		ValidLicense: data.ValidLicense,
+		GptBaseUrl:   setting.GPTBaseURL,
+		GptApiKey:    setting.GPTApiKey,
+		GptProxy:     setting.GPTProxy,
+		GptModel:     setting.GPTModel,
 	}
 	return &pb.PublicSettingResponse{Status: &status, Data: &pbSetting}, nil
 }

diff --git a/pkg/jms-sdk-go/model/terminal.go b/pkg/jms-sdk-go/model/terminal.go
@@ -15,6 +15,11 @@ type TerminalConfig struct {
 	HeartbeatDuration   int           `json:"TERMINAL_HEARTBEAT_INTERVAL"`
 	HostKey             string        `json:"TERMINAL_HOST_KEY"`
 	EnableSessionShare  bool          `json:"SECURITY_SESSION_SHARE"`
+
+	GPTBaseURL string `json:"GPT_BASE_URL"`
+	GPTApiKey  string `json:"GPT_API_KEY"`
+	GPTProxy   string `json:"GPT_PROXY"`
+	GPTModel   string `json:"GPT_MODEL"`
 }
 
 type Terminal struct {

diff --git a/protobuf-go/protobuf/common.pb.go b/protobuf-go/protobuf/common.pb.go
diff --git a/protos/common.proto b/protos/common.proto
@@ -193,6 +193,10 @@ message Forward {
 message PublicSetting {
   bool xpack_enabled = 1;
   bool valid_license = 2;
+  string gpt_base_url = 3;
+  string gpt_api_key = 4;
+  string gpt_proxy = 5;
+  string gpt_model = 6;
 }
 
 message Cookie {