From 4947ac29657f2c97c86a1d5551a25232c28fb1d2 Mon Sep 17 00:00:00 2001 From: Eugene Yurtsev Date: Thu, 28 Sep 2023 16:24:00 -0400 Subject: [PATCH] Add langserve version (#11195) Add langserve version --- libs/langserve/langserve/__init__.py | 11 +++++++++++ libs/langserve/tests/unit_tests/test_version.py | 3 +++ 2 files changed, 14 insertions(+) create mode 100644 libs/langserve/tests/unit_tests/test_version.py diff --git a/libs/langserve/langserve/__init__.py b/libs/langserve/langserve/__init__.py index c708c73398..581495e0bd 100644 --- a/libs/langserve/langserve/__init__.py +++ b/libs/langserve/langserve/__init__.py @@ -1,4 +1,15 @@ +"""Main entrypoint into package.""" +from importlib import metadata + from langserve.client import RemoteRunnable from langserve.server import add_routes __all__ = ["RemoteRunnable", "add_routes"] + + +try: + __version__ = metadata.version(__package__) +except metadata.PackageNotFoundError: + # Case where package metadata is not available. + __version__ = "" +del metadata # optional, avoids polluting the results of dir(__package__) diff --git a/libs/langserve/tests/unit_tests/test_version.py b/libs/langserve/tests/unit_tests/test_version.py new file mode 100644 index 0000000000..9ba15d398b --- /dev/null +++ b/libs/langserve/tests/unit_tests/test_version.py @@ -0,0 +1,3 @@ +def test_version_available() -> None: + """Test that a version is available.""" + from langserve import __version__ # noqa: F401