github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-13 21:47:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add serialisation props to Fireworks and ChatFireworks (#12255)

Browse Source
This commit is contained in:
David Duong
2023-10-25 12:41:33 +02:00
committed by GitHub
parent fd5f549a9e
commit c25b174db5
2 changed files with 16 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
libs/langchain/langchain/chat_models/fireworks.py
View File

@@ -90,6 +90,14 @@ class ChatFireworks(BaseChatModel):
fireworks_api_key: Optional[str] = None
max_retries: int = 20
@property
def lc_secrets(self) -> Dict[str, str]:
return {"fireworks_api_key": "FIREWORKS_API_KEY"}
@classmethod
def is_lc_serializable(cls) -> bool:
return True
@root_validator()
def validate_environment(cls, values: Dict) -> Dict:
"""Validate that api key in environment."""

8
libs/langchain/langchain/llms/fireworks.py
View File

@@ -37,6 +37,14 @@ class Fireworks(LLM):
fireworks_api_key: Optional[str] = None
max_retries: int = 20
@property
def lc_secrets(self) -> Dict[str, str]:
return {"fireworks_api_key": "FIREWORKS_API_KEY"}
@classmethod
def is_lc_serializable(cls) -> bool:
return True
@root_validator()
def validate_environment(cls, values: Dict) -> Dict:
"""Validate that api key in environment."""