From 356a139b0a6c1ac5a0eff95c265ce226a91838bb Mon Sep 17 00:00:00 2001 From: Eugene Yurtsev Date: Mon, 1 Apr 2024 15:34:38 -0400 Subject: [PATCH] cli[minor]: Add __version__ to integration package template (#19876) Packages should export __version__ --- .../integration_template/__init__.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/libs/cli/langchain_cli/integration_template/integration_template/__init__.py b/libs/cli/langchain_cli/integration_template/integration_template/__init__.py index eb8632da86..60d8dac5d0 100644 --- a/libs/cli/langchain_cli/integration_template/integration_template/__init__.py +++ b/libs/cli/langchain_cli/integration_template/integration_template/__init__.py @@ -1,11 +1,21 @@ +from importlib import metadata + from __module_name__.chat_models import Chat__ModuleName__ from __module_name__.embeddings import __ModuleName__Embeddings from __module_name__.llms import __ModuleName__LLM from __module_name__.vectorstores import __ModuleName__VectorStore +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__) + __all__ = [ - "__ModuleName__LLM", "Chat__ModuleName__", + "__ModuleName__LLM", "__ModuleName__VectorStore", "__ModuleName__Embeddings", + "__version__", ]