fix crash on ps when multiple models are loaded at same time (#24)

paulrobello · Aug 11, 2024 · 8f96569 · 8f96569
1 parent 4976332
commit 8f96569
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -27,6 +27,8 @@
     * [Where we are](#where-we-are)
     * [Where we're going](#where-were-going)
 * [What's new](#whats-new)
+    * [v0.3.7](#v036)
+    * [v0.3.6](#v036)
     * [v0.3.5](#v035)
     * [v0.3.4](#v034)
     * [v0.3.3](#v033)
@@ -324,6 +326,9 @@ if anything remains to be fixed before the commit is allowed.
 
 ## What's new
 
+### v0.3.7
+* Fix for possible crash when there is more than one model loaded into ollama
+
 ### v0.3.6
 * Added option to save chat input history and set its length
 * Fixed tab switch issue on startup

diff --git a/parllama/__init__.py b/parllama/__init__.py
@@ -7,7 +7,7 @@
 __credits__ = ["Paul Robello"]
 __maintainer__ = "Paul Robello"
 __email__ = "probello@gmail.com"
-__version__ = "0.3.6"
+__version__ = "0.3.7"
 __licence__ = "MIT"
 __application_title__ = "PAR LLAMA"
 __application_binary__ = "parllama"

diff --git a/parllama/utils.py b/parllama/utils.py
@@ -660,6 +660,8 @@ def output_to_dicts(output: str) -> list[dict[str, Any]]:
     for model in reader:
         mod = {}
         for key, value in model.items():
+            if not isinstance(key, str):
+                continue
             mod[key.strip().lower()] = value.strip()
         ret.append(mod)
     return ret