github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-25 08:03:39 +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
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 16 additions and 0 deletions
Show Stats 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 fireworks_api_key: Optional[str] = None
max_retries: int = 20 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() @root_validator()
def validate_environment(cls, values: Dict) -> Dict: def validate_environment(cls, values: Dict) -> Dict:
"""Validate that api key in environment.""" """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 fireworks_api_key: Optional[str] = None
max_retries: int = 20 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() @root_validator()
def validate_environment(cls, values: Dict) -> Dict: def validate_environment(cls, values: Dict) -> Dict:
"""Validate that api key in environment.""" """Validate that api key in environment."""