Add __version__ (#103)

Add __version__ to kor public interface
eyurtsev · Apr 1, 2023 · eb35feb · eb35feb
1 parent 838f957
commit eb35feb
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 0 deletions.
diff --git a/kor/__init__.py b/kor/__init__.py
@@ -7,6 +7,7 @@
     TypeDescriptor,
     TypeScriptDescriptor,
 )
+from .version import __version__
 
 __all__ = (
     "BulletPointDescriptor",
@@ -22,4 +23,5 @@
     "TypeDescriptor",
     "TypeScriptDescriptor",
     "XMLEncoder",
+    "__version__",
 )
diff --git a/kor/version.py b/kor/version.py
@@ -0,0 +1,7 @@
+"""Get the version of the package."""
+from importlib import metadata
+
+try:
+    __version__ = metadata.version("kor")
+except metadata.PackageNotFoundError:
+    __version__ = "local"
diff --git a/tests/test_public_interface.py b/tests/test_public_interface.py
@@ -19,6 +19,7 @@ def test_kor__all__() -> None:
         "TypeDescriptor",
         "TypeScriptDescriptor",
         "XMLEncoder",
+        "__version__",
         "create_extraction_chain",
         "from_pydantic",
     ]