Minor fixes (#1)

* fixed logo in readme

* fixed  type validation

Author: n-splv

README.md

@@ -1,7 +1,7 @@
 <picture>
   <source media="(prefers-color-scheme: dark)" srcset="docs/images/logo.dark.png" width="300">
   <source media="(prefers-color-scheme: light)" srcset="docs/images/logo.light.png" width="300">
-  <img alt="logo" src="docs/images/logo.light.png" width="300">
+  <img alt="logo" src="https://github.com/n-splv/promptsub-py/raw/main/docs/images/logo.light.png" width="300">
 </picture>
 
 # Prompt substitution for humans

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "promptsub"
-version = "0.1.0"
+version = "0.1.1"
 description = "Creating parametrized prompts for language models"
 authors = ["n-splv <[email protected]>"]
 license = "MIT"

src/promptsub/prompt_api.py

@@ -1,3 +1,4 @@
+from __future__ import annotations
 from functools import cached_property
 
 from attrs import (
@@ -16,10 +17,18 @@
 )
 
 
-template_factory = Factory(
-    lambda self: Template(input_text=self.template_text),
-    takes_self=True
-)
+def _init_template(self: Prompt) -> Template:
+    """
+    Manual type check because `attrs.validators` is a bit broken:
+    https://github.com/python-attrs/attrs/issues/1237
+
+    Can be replaced with a simple lambda if it gets fixed.
+    """
+    if not isinstance(self.template_text, str):
+        err_message = "Template text must be a string"
+        raise TypeError(err_message)
+
+    return Template(input_text=self.template_text)
 
 
 @frozen
@@ -43,7 +52,9 @@ class Prompt:
     template_text: str = field(validator=validators.instance_of(str))
 
     _template: Template = field(
-        init=False, repr=False, default=template_factory
+        init=False,
+        repr=False,
+        default=Factory(_init_template, takes_self=True)
     )
 
     def substitute(self,

tests/test_errors.py

@@ -7,11 +7,11 @@
 from promptsub import special_chatacters as sc
 
 
-def raises(exception):
+def raises(exception, match: str = None):
     def decorator(func):
         @wraps(func)
         def wrapper(*args, **kwargs):
-            with pytest.raises(exception):
+            with pytest.raises(exception, match=match):
                 func(*args, **kwargs)
         return wrapper
     return decorator
@@ -75,7 +75,7 @@ def test_component_not_closed(self, template):
     def test_empty_template(self, template):
         Prompt(template)
 
-    @raises(TypeError)
+    @raises(TypeError, match="Template text must be a string")
     @pytest.mark.parametrize("template", (
         None,
         type("Foo", (), {})(),