From 7205057c3e12c3fa2e3424c9837c86753637af6f Mon Sep 17 00:00:00 2001
From: Dristy Srivastava <58721149+dristysrivastava@users.noreply.github.com>
Date: Mon, 26 Aug 2024 18:17:48 +0530
Subject: [PATCH] [Community][minor]: Added langchain_version while calling
 discover API (#24428)

- **Description:** Added langchain version while calling discover API
during both ingestion and retrieval
- **Issue:** NA
- **Dependencies:** NA
- **Tests:** NA
- **Docs** NA

---------

Co-authored-by: dristy.cd <dristy@clouddefense.io>
---
 .../langchain_community/chains/pebblo_retrieval/base.py     | 6 ++++++
 .../langchain_community/chains/pebblo_retrieval/models.py   | 1 +
 .../langchain_community/document_loaders/pebblo.py          | 6 ++++++
 libs/community/langchain_community/utilities/pebblo.py      | 2 ++
 4 files changed, 15 insertions(+)

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):