diff --git a/libs/community/langchain_community/chains/pebblo_retrieval/base.py b/libs/community/langchain_community/chains/pebblo_retrieval/base.py index a29bda567c2..ee595061d95 100644 --- a/libs/community/langchain_community/chains/pebblo_retrieval/base.py +++ b/libs/community/langchain_community/chains/pebblo_retrieval/base.py @@ -6,6 +6,7 @@ against a vector database. import datetime import inspect import logging +from importlib.metadata import version from typing import Any, Dict, List, Optional from langchain.chains.base import Chain @@ -27,6 +28,7 @@ from langchain_community.chains.pebblo_retrieval.models import ( App, AuthContext, ChainInfo, + Framework, Model, SemanticContext, VectorDB, @@ -327,6 +329,10 @@ class PebbloRetrievalQA(Chain): framework=framework, chains=chains, plugin_version=PLUGIN_VERSION, + client_version=Framework( + name="langchain_community", + version=version("langchain_community"), + ), ) return app diff --git a/libs/community/langchain_community/chains/pebblo_retrieval/models.py b/libs/community/langchain_community/chains/pebblo_retrieval/models.py index d5693404214..315905d18dd 100644 --- a/libs/community/langchain_community/chains/pebblo_retrieval/models.py +++ b/libs/community/langchain_community/chains/pebblo_retrieval/models.py @@ -123,6 +123,7 @@ class App(BaseModel): framework: Framework chains: List[ChainInfo] plugin_version: str + client_version: Framework class Context(BaseModel): diff --git a/libs/community/langchain_community/document_loaders/pebblo.py b/libs/community/langchain_community/document_loaders/pebblo.py index becb5d54859..bcf1cde050b 100644 --- a/libs/community/langchain_community/document_loaders/pebblo.py +++ b/libs/community/langchain_community/document_loaders/pebblo.py @@ -3,6 +3,7 @@ import logging import os import uuid +from importlib.metadata import version from typing import Dict, Iterator, List, Optional from langchain_core.documents import Document @@ -12,6 +13,7 @@ from langchain_community.utilities.pebblo import ( BATCH_SIZE_BYTES, PLUGIN_VERSION, App, + Framework, IndexedDocument, PebbloLoaderAPIWrapper, generate_size_based_batches, @@ -175,6 +177,10 @@ class PebbloSafeLoader(BaseLoader): runtime=runtime, framework=framework, plugin_version=PLUGIN_VERSION, + client_version=Framework( + name="langchain_community", + version=version("langchain_community"), + ), ) return app diff --git a/libs/community/langchain_community/utilities/pebblo.py b/libs/community/langchain_community/utilities/pebblo.py index 7f0dcfc5343..eacf90ed8aa 100644 --- a/libs/community/langchain_community/utilities/pebblo.py +++ b/libs/community/langchain_community/utilities/pebblo.py @@ -129,6 +129,8 @@ class App(BaseModel): """Framework details of the app.""" plugin_version: str """Plugin version used for the app.""" + client_version: Framework + """Client version used for the app.""" class Doc(BaseModel):