From 0bc6d5a8ecc3835ed6371b4ade810d1e3dd2f16d Mon Sep 17 00:00:00 2001 From: Rajendra Kadam Date: Tue, 3 Sep 2024 18:52:11 +0530 Subject: [PATCH] Set Cloud as Policy source --- .../langchain_community/chains/pebblo_retrieval/base.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/libs/community/langchain_community/chains/pebblo_retrieval/base.py b/libs/community/langchain_community/chains/pebblo_retrieval/base.py index c914bb1e046e3..50fecff9d1dd2 100644 --- a/libs/community/langchain_community/chains/pebblo_retrieval/base.py +++ b/libs/community/langchain_community/chains/pebblo_retrieval/base.py @@ -36,6 +36,7 @@ from langchain_community.chains.pebblo_retrieval.utilities import ( PLUGIN_VERSION, PebbloRetrievalAPIWrapper, + PolicySource, get_runtime, ) @@ -255,6 +256,7 @@ def from_chain_type( classifier_location=classifier_location, classifier_url=classifier_url, app_name=app_name, + policy_source=PolicySource.CLOUD, ) # send app discovery request pb_client.send_app_discover(app)