Add serialisable support for Replicate (#8525)

langchain-ai · Aug 11, 2023 · 6d03f8b · 6d03f8b
1 parent 16af5f8
commit 6d03f8b
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/libs/langchain/langchain/llms/replicate.py b/libs/langchain/langchain/llms/replicate.py
@@ -46,6 +46,14 @@ class Config:
 
         extra = Extra.forbid
 
+    @property
+    def lc_secrets(self) -> Dict[str, str]:
+        return {"replicate_api_token": "REPLICATE_API_TOKEN"}
+
+    @property
+    def lc_serializable(self) -> bool:
+        return True
+
     @root_validator(pre=True)
     def build_extra(cls, values: Dict[str, Any]) -> Dict[str, Any]:
         """Build extra kwargs from additional params that were passed in."""