Archives/langchain
2
0
Fork 1
mirror of https://github.com/hwchase17/langchain synced 2024-11-06 03:20:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix secrets serialisation for ChatAnthropic (#7300)

Browse Source
This commit is contained in:
David Duong 2023-07-06 22:57:12 +02:00 committed by GitHub
parent cb9ff6efb8
commit 12d14f8947
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 10 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
langchain/chat_models/anthropic.py
View File

@ -34,6 +34,10 @@ class ChatAnthropic(BaseChatModel, _AnthropicCommon):
model = ChatAnthropic(model="<model_name>", anthropic_api_key="my-api-key") model = ChatAnthropic(model="<model_name>", anthropic_api_key="my-api-key")
""" """
@property
def lc_secrets(self) -> Dict[str, str]:
return {"anthropic_api_key": "ANTHROPIC_API_KEY"}
@property @property
def _llm_type(self) -> str: def _llm_type(self) -> str:
"""Return type of chat model.""" """Return type of chat model."""

12
langchain/llms/anthropic.py
View File

@ -50,11 +50,11 @@ class _AnthropicCommon(BaseModel):
@root_validator() @root_validator()
def validate_environment(cls, values: Dict) -> Dict: def validate_environment(cls, values: Dict) -> Dict:
"""Validate that api key and python package exists in environment.""" """Validate that api key and python package exists in environment."""
anthropic_api_key = get_from_dict_or_env( values["anthropic_api_key"] = get_from_dict_or_env(
values, "anthropic_api_key", "ANTHROPIC_API_KEY" values, "anthropic_api_key", "ANTHROPIC_API_KEY"
) )
# Get custom api url from environment. # Get custom api url from environment.
anthropic_api_url = get_from_dict_or_env( values["anthropic_api_url"] = get_from_dict_or_env(
values, values,
"anthropic_api_url", "anthropic_api_url",
"ANTHROPIC_API_URL", "ANTHROPIC_API_URL",
@ -72,13 +72,13 @@ class _AnthropicCommon(BaseModel):
f"`pip install -U anthropic`" f"`pip install -U anthropic`"
) )
values["client"] = anthropic.Anthropic( values["client"] = anthropic.Anthropic(
base_url=anthropic_api_url, base_url=values["anthropic_api_url"],
api_key=anthropic_api_key, api_key=values["anthropic_api_key"],
timeout=values["default_request_timeout"], timeout=values["default_request_timeout"],
) )
values["async_client"] = anthropic.AsyncAnthropic( values["async_client"] = anthropic.AsyncAnthropic(
base_url=anthropic_api_url, base_url=values["anthropic_api_url"],
api_key=anthropic_api_key, api_key=values["anthropic_api_key"],
timeout=values["default_request_timeout"], timeout=values["default_request_timeout"],
) )
values["HUMAN_PROMPT"] = anthropic.HUMAN_PROMPT values["HUMAN_PROMPT"] = anthropic.HUMAN_PROMPT