Bläddra i källkod

Refactor code. Remove extra variable.

Vyacheslav Moskalev 9 månader sedan
förälder
incheckning
3b5210548e
1 ändrade filer med 4 tillägg och 6 borttagningar
  1. 4 6
      server/routes.go

+ 4 - 6
server/routes.go

@@ -187,7 +187,6 @@ func (s *Server) GenerateHandler(c *gin.Context) {
 			values.Messages = append(msgs, api.Message{Role: "user", Content: req.Prompt})
 			values.Messages = append(msgs, api.Message{Role: "user", Content: req.Prompt})
 		}
 		}
 
 
-		var s string
 		var b bytes.Buffer
 		var b bytes.Buffer
 		if err := tmpl.Execute(&b, values); err != nil {
 		if err := tmpl.Execute(&b, values); err != nil {
 			c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
 			c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
@@ -195,16 +194,15 @@ func (s *Server) GenerateHandler(c *gin.Context) {
 		}
 		}
 
 
 		if req.Context != nil {
 		if req.Context != nil {
-			prev, err := r.Detokenize(c.Request.Context(), req.Context)
+			s, err := r.Detokenize(c.Request.Context(), req.Context)
 			if err != nil {
 			if err != nil {
 				c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
 				c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
 				return
 				return
 			}
 			}
-			s += prev
+			prompt = s + b.String()
+		} else {
+			prompt = b.String();
 		}
 		}
-
-		s += b.String();
-		prompt = s
 	}
 	}
 
 
 	slog.Debug("generate request", "prompt", prompt, "images", images)
 	slog.Debug("generate request", "prompt", prompt, "images", images)