Catch case where chain cannot be built from content

Fingel · Jan 9, 2025 · 28b20f5 · 28b20f5
1 parent 6dd49de
commit 28b20f5
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "django-llm-poison"
-version = "0.2.0"
+version = "0.2.1"
 description = "Add your description here"
 readme = "README.md"
 authors = [{ name = "Austin Riba", email = "austin@m51.io" }]

diff --git a/src/django_llm_poison/markov.py b/src/django_llm_poison/markov.py
@@ -31,7 +31,10 @@ def build_model_for_content(value: str):
     hash = hashlib.sha1(value.encode()).hexdigest()
     stripped = strip_tags(value)
     if not MarkovModel.objects.filter(hash=hash).exists():
-        model = markovify.Text(stripped)
+        try:
+            model = markovify.Text(stripped)
+        except Exception:
+            return None
         markov_model = MarkovModel.objects.create(hash=hash, text=model.to_json())
         cache.delete("combined_model")
         return markov_model