diff --git a/src/main/java/ee/carlrobert/codegpt/completions/CompletionClientProvider.java b/src/main/java/ee/carlrobert/codegpt/completions/CompletionClientProvider.java index 1c11b732..8868e26d 100644 --- a/src/main/java/ee/carlrobert/codegpt/completions/CompletionClientProvider.java +++ b/src/main/java/ee/carlrobert/codegpt/completions/CompletionClientProvider.java @@ -55,9 +55,9 @@ public class CompletionClientProvider { } public static LlamaClient getLlamaClient() { - return new LlamaClient.Builder() - .setPort(LlamaSettingsState.getInstance().getServerPort()) - .build(); + var builder = new LlamaClient.Builder() + .setPort(LlamaSettingsState.getInstance().getServerPort()); + return (LlamaClient) addDefaultClientParams(builder).build(); } private static OpenAIClient.Builder getOpenAIClientBuilder() {