jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Update ChatAnthropicComponent configuration

Browse source
This commit is contained in:
carlosrcoelho 2024-01-10 11:52:39 -03:00
commit ee4a0da65d
1 changed files with 5 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

10
src/backend/langflow/components/llms/ChatAnthropic.py
View file

@ -1,6 +1,5 @@
from langflow import CustomComponent from langflow import CustomComponent
from langchain.tools import SecretStr
from typing import Optional, Dict, Union, Callable from typing import Optional, Dict, Union, Callable
from langflow.field_typing import BaseLanguageModel from langflow.field_typing import BaseLanguageModel
@ -13,7 +12,8 @@ class ChatAnthropicComponent(CustomComponent):
return { return {
"anthropic_api_key": { "anthropic_api_key": {
"display_name": "Anthropic API Key", "display_name": "Anthropic API Key",
"type": SecretStr, "type": str,
"password": True,
}, },
"anthropic_api_url": { "anthropic_api_url": {
"display_name": "Anthropic API URL", "display_name": "Anthropic API URL",
@ -21,7 +21,7 @@ class ChatAnthropicComponent(CustomComponent):
}, },
"model_kwargs": { "model_kwargs": {
"display_name": "Model Kwargs", "display_name": "Model Kwargs",
"type": Dict[str, Union[str, int, float, bool]], "field_type": 'dict',
"advanced": True, "advanced": True,
}, },
"temperature": { "temperature": {
@ -32,9 +32,9 @@ class ChatAnthropicComponent(CustomComponent):
def build( def build(
self, self,
anthropic_api_key: Optional[SecretStr] = None, anthropic_api_key: Optional[str] = None,
anthropic_api_url: Optional[str] = None, anthropic_api_url: Optional[str] = None,
model_kwargs: Optional[Dict[str, Union[str, int, float, bool]]] = None, model_kwargs: dict = {},
temperature: Optional[float] = None, temperature: Optional[float] = None,
) -> Union[BaseLanguageModel, Callable]: ) -> Union[BaseLanguageModel, Callable]:
from langchain.model_io.models.chat.integrations import ChatAnthropic # Importing here due to potential local scope requirements from langchain.model_io.models.chat.integrations import ChatAnthropic # Importing here due to potential local scope requirements