fix num_keep

681f3c4c · Michael Yang · 5d3f314b · 681f3c4c
Commit 681f3c4c authored Sep 03, 2023 by Michael Yang
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

server/routes.go server/routes.go +2 -1

No files found.
--- a/server/routes.go
+++ b/server/routes.go
@@ -117,12 +117,13 @@ func load(ctx context.Context, model *Model, reqOpts map[string]interface{}, ses
 			if err != nil {
 				return err
 			}
+
 			tokensNoSystem, err := llmModel.Encode(ctx, promptNoSystem)
 			if err != nil {
 				return err
 			}

-			opts.NumKeep = len(tokensWithSystem) - len(tokensNoSystem) + 1
+			opts.NumKeep = len(tokensWithSystem) - len(tokensNoSystem)

 			llmModel.SetOptions(opts)
 		}