updates

2060-io · Nov 9, 2024 · 9b87090 · 9b87090
1 parent 2fd890d
commit 9b87090
Showing 1 changed file with 20 additions and 1 deletion.
diff --git a/src/main/java/io/twentysixty/ollama/hologram/chatbot/svc/OllamaService.java b/src/main/java/io/twentysixty/ollama/hologram/chatbot/svc/OllamaService.java
@@ -77,7 +77,26 @@ public List<OllamaChatMessage> getMessagesFromHistory(UUID connectionId) {
 			OllamaChatMessage m = new OllamaChatMessage();
 			OllamaChatMessageRole role;
 			try {
-				role = OllamaChatMessageRole.getRole(h.getRole().toString());
+
+				/*switch (h.getRole()) {
+					
+					case ASSISTANT: {
+						
+					}
+					case USER: {
+						
+					}
+					case TOOL: {
+						
+					}
+					default:
+					case SYSTEM: {
+						
+					}
+						
+				}*/
+
+				role = OllamaChatMessageRole.getRole(h.getRole().toString().toLowerCase());
 			} catch (RoleNotFoundException e) {
 				role = OllamaChatMessageRole.SYSTEM;
 			}