[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>
This commit is contained in:
Dristy Srivastava 2024-08-26 18:17:48 +05:30 committed by GitHub
parent fbb4761199
commit 7205057c3e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
4 changed files with 15 additions and 0 deletions

View File

@ -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

View File

@ -123,6 +123,7 @@ class App(BaseModel):
framework: Framework
chains: List[ChainInfo]
plugin_version: str
client_version: Framework
class Context(BaseModel):

View File

@ -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

View File

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