From 6d03f8b5d82781f4c013682d21405ed27dbfc504 Mon Sep 17 00:00:00 2001 From: David Duong Date: Fri, 11 Aug 2023 20:35:21 +0200 Subject: [PATCH] Add serialisable support for Replicate (#8525) --- libs/langchain/langchain/llms/replicate.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/libs/langchain/langchain/llms/replicate.py b/libs/langchain/langchain/llms/replicate.py index 5ae4638d501..c8706a6f8a8 100644 --- a/libs/langchain/langchain/llms/replicate.py +++ b/libs/langchain/langchain/llms/replicate.py @@ -46,6 +46,14 @@ class Replicate(LLM): 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."""