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

refactor: Update max_tokens default value in AnthropicLLM classes

Browse source
This commit is contained in:
Rodrigo 2024-06-14 01:12:07 -03:00
commit 750624f097
3 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

2
src/backend/base/langflow/components/model_specs/AnthropicLLMSpecs.py
View file

@ -54,7 +54,7 @@ class ChatAntropicSpecsComponent(CustomComponent):
self,
model: str,
anthropic_api_key: Optional[str] = None,
max_tokens: Optional[int] = None,
max_tokens: Optional[int] = 1000,
temperature: Optional[float] = None,
api_endpoint: Optional[str] = None,
) -> BaseLanguageModel:

2
src/backend/base/langflow/components/model_specs/ChatAnthropicSpecs.py
View file

@ -66,7 +66,7 @@ class AnthropicLLM(CustomComponent):
self,
model: str,
anthropic_api_key: Optional[str] = None,
max_tokens: Optional[int] = None,
max_tokens: Optional[int] = 1000,
temperature: Optional[float] = None,
anthropic_api_url: Optional[str] = None,
) -> BaseLanguageModel:

2
src/backend/base/langflow/components/models/AnthropicModel.py
View file

@ -82,7 +82,7 @@ class AnthropicLLM(LCModelComponent):
input_value: Text,
system_message: Optional[str] = None,
anthropic_api_key: Optional[str] = None,
max_tokens: Optional[int] = None,
max_tokens: Optional[int] = 1000,
temperature: Optional[float] = None,
anthropic_api_url: Optional[str] = None,
stream: bool = False,