fixed syste msg reset

kspviswa · Dec 11, 2024 · 44f5576 · 44f5576
1 parent e6e05c3
commit 44f5576
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/bin/lpw_ollamaClient.py b/bin/lpw_ollamaClient.py
@@ -28,9 +28,15 @@ def check_system_message(self) -> bool:
             return False
 
     def set_system_message(self, system_message:str) -> None:
+        if self.check_system_message():
+            self.edit_system_message(system_message)
+        else:
+            self.create_system_message(system_message)
+
+    def create_system_message(self, system_message:str) -> None:
         sMessage = dict({'role' : 'system', 'content' : system_message})
         self.messages.append(sMessage)
-    
+
     def edit_system_message(self, system_message:str) -> None:
         for m in self.messages:
             if m['role'] == 'system':