瀏覽代碼

check other request fields before load short circuit in `/api/generate`

Jeffrey Morgan 1 年之前
父節點
當前提交
9b12a511ca
共有 1 個文件被更改,包括 2 次插入1 次删除
  1. 2 1
      server/routes.go

+ 2 - 1
server/routes.go

@@ -223,7 +223,8 @@ func GenerateHandler(c *gin.Context) {
 			ch <- r
 			ch <- r
 		}
 		}
 
 
-		if req.Prompt == "" {
+		// an empty request loads the model
+		if req.Prompt == "" && req.Template == "" && req.System == "" {
 			ch <- api.GenerateResponse{Model: req.Model, Done: true}
 			ch <- api.GenerateResponse{Model: req.Model, Done: true}
 		} else {
 		} else {
 			if err := loaded.llm.Predict(c.Request.Context(), req.Context, prompt, fn); err != nil {
 			if err := loaded.llm.Predict(c.Request.Context(), req.Context, prompt, fn); err != nil {