Merge pull request #164 from kthcloud/dev

hotfixes
kthcloud · Sep 4, 2023 · a7bca31 · a7bca31
2 parents 5ca0bb6 + 8ac4908
commit a7bca31
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/models/sys/vm/gpu/client.go b/models/sys/vm/gpu/client.go
@@ -23,6 +23,14 @@ func New() *Client {
 }
 
 func NewWithExclusion(excludedHosts []string, excludedGPUs []string) *Client {
+	if excludedHosts == nil {
+		excludedHosts = make([]string, 0)
+	}
+
+	if excludedGPUs == nil {
+		excludedGPUs = make([]string, 0)
+	}
+
 	return &Client{
 		ExcludedHosts: excludedHosts,
 		ExcludedGPUs:  excludedGPUs,

diff --git a/routers/api/v1/v1_deployment/logs.go b/routers/api/v1/v1_deployment/logs.go
@@ -65,7 +65,9 @@ func GetLogs(c *gin.Context) {
 				case websocket.CloseNormalClosure,
 					websocket.CloseGoingAway,
 					websocket.CloseNoStatusReceived:
-					logContext.Done()
+					if logContext != nil {
+						logContext.Done()
+					}
 					log.Println("websocket closed for deployment ", requestURI.DeploymentID)
 					return
 				}