CreateRequest will include missing numPredict

microsoft · Jan 18, 2025 · dd7d6a5 · dd7d6a5
1 parent 7dc448d
commit dd7d6a5
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/dotnet/src/Connectors/Connectors.Ollama/Services/OllamaTextGenerationService.cs b/dotnet/src/Connectors/Connectors.Ollama/Services/OllamaTextGenerationService.cs
@@ -131,7 +131,8 @@ private static GenerateRequest CreateRequest(OllamaPromptExecutionSettings setti
                 Temperature = settings.Temperature,
                 TopP = settings.TopP,
                 TopK = settings.TopK,
-                Stop = settings.Stop?.ToArray()
+                Stop = settings.Stop?.ToArray(),
+                NumPredict = settings.NumPredict
             },
             Model = selectedModel,
             Stream = true