mirror of
https://github.com/hwchase17/langchain
synced 2024-11-10 01:10:59 +00:00
[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:
parent
fbb4761199
commit
7205057c3e
@ -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
|
||||
|
||||
|
@ -123,6 +123,7 @@ class App(BaseModel):
|
||||
framework: Framework
|
||||
chains: List[ChainInfo]
|
||||
plugin_version: str
|
||||
client_version: Framework
|
||||
|
||||
|
||||
class Context(BaseModel):
|
||||
|
@ -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
|
||||
|
||||
|
@ -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):
|
||||
|
Loading…
Reference in New Issue
Block a user