Fixes JSON Binding for OllamaChatMessageRole

2024-02-10 00:22:55 +00:00 · 2024-02-10 00:22:55 +00:00 · 4ca6eef8fd
commit 4ca6eef8fd
parent a635dd9be2
1 changed files with 4 additions and 2 deletions
--- a/src/main/java/io/github/amithkoujalgi/ollama4j/core/models/chat/OllamaChatMessageRole.java
+++ b/src/main/java/io/github/amithkoujalgi/ollama4j/core/models/chat/OllamaChatMessageRole.java
@ -1,14 +1,16 @@
 package io.github.amithkoujalgi.ollama4j.core.models.chat;

+import com.fasterxml.jackson.annotation.JsonValue;
+
 /**
 * Defines the possible Chat Message roles.
 */
 public enum OllamaChatMessageRole {
    SYSTEM("system"),
    USER("user"),
-    ASSISTANT("assisstant");
+    ASSISTANT("assistant");

-    @SuppressWarnings("unused")
+    @JsonValue
    private String roleName;

    private OllamaChatMessageRole(String roleName){