From ed9397c3dde571cf5703f7be4032e82a3613bd49 Mon Sep 17 00:00:00 2001 From: Carl-Robert Linnupuu Date: Sun, 21 Apr 2024 23:25:53 +0300 Subject: [PATCH] fix: llama server success callback trigger --- .../carlrobert/codegpt/completions/llama/LlamaServerAgent.java | 3 ++- .../codegpt/completions/llama/LlamaServerMessage.java | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/src/main/java/ee/carlrobert/codegpt/completions/llama/LlamaServerAgent.java b/src/main/java/ee/carlrobert/codegpt/completions/llama/LlamaServerAgent.java index a837150a..ee3854ed 100644 --- a/src/main/java/ee/carlrobert/codegpt/completions/llama/LlamaServerAgent.java +++ b/src/main/java/ee/carlrobert/codegpt/completions/llama/LlamaServerAgent.java @@ -157,7 +157,8 @@ public final class LlamaServerAgent implements Disposable { try { var serverMessage = objectMapper.readValue(event.getText(), LlamaServerMessage.class); - if ("HTTP server listening".equals(serverMessage.message())) { + // hack + if ("HTTP server listening".equals(serverMessage.msg())) { LOG.info("Server up and running!"); LlamaSettings.getCurrentState().setServerPort(port); diff --git a/src/main/java/ee/carlrobert/codegpt/completions/llama/LlamaServerMessage.java b/src/main/java/ee/carlrobert/codegpt/completions/llama/LlamaServerMessage.java index 0eec2be9..b3f683ab 100644 --- a/src/main/java/ee/carlrobert/codegpt/completions/llama/LlamaServerMessage.java +++ b/src/main/java/ee/carlrobert/codegpt/completions/llama/LlamaServerMessage.java @@ -3,5 +3,5 @@ package ee.carlrobert.codegpt.completions.llama; import com.fasterxml.jackson.annotation.JsonIgnoreProperties; @JsonIgnoreProperties(ignoreUnknown = true) -public record LlamaServerMessage(String level, String message) { +public record LlamaServerMessage(String level, String msg) { }