Set Cloud as Policy source

daxa-ai · Sep 5, 2024 · ce3b01d · ce3b01d
1 parent b842c21
commit ce3b01d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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)