improved logs

DiverOfDark · Aug 20, 2024 · 10e11d1 · 10e11d1
1 parent 7e3b376
commit 10e11d1
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/kube/chat-summarizer.yaml b/kube/chat-summarizer.yaml
@@ -19,6 +19,7 @@ spec:
           containers:
             - name: ollama
               image: ghcr.io/diverofdark/telegramsummarizer:master
+              imagePullPolicy: Always
               volumeMounts:
                 - mountPath: /ollama/
                   name: shared

diff --git a/src/main/java/pro/kirillorlov/chat_summarizer/llama/LlamaController.java b/src/main/java/pro/kirillorlov/chat_summarizer/llama/LlamaController.java
@@ -76,9 +76,13 @@ private void downloadModel(OllamaProperties props) throws Exception {
                     String line;
                     while ((line = reader.readLine()) != null) {
                         // Process each line of the response as it arrives
-                        HashMap hashMap = Json.fromJson(line, HashMap.class);
-                        String status = (String) hashMap.get("status");
-                        logger.info("Progress: {}", status);
+                        try {
+                            HashMap hashMap = Json.fromJson(line, HashMap.class);
+                            String status = (String) hashMap.get("status");
+                            logger.info("Progress: {}", status);
+                        } catch (Exception e) {
+                            logger.error("Error parsing response: {}", line);
+                        }
                     }
 
                 } catch (StreamClosedException e) { }