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

Update AnthropicLLM model name

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-12-04 11:39:14 -03:00
commit 1c2d2402ba
1 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
src/backend/langflow/components/llms/AnthropicLLM.py
View file

@ -1,8 +1,10 @@
from typing import Optional from typing import Optional
from langflow import CustomComponent
from langchain.chat_models.anthropic import ChatAnthropic from langchain.chat_models.anthropic import ChatAnthropic
from langchain.llms.base import BaseLLM from langchain.llms.base import BaseLLM
from langflow import CustomComponent
class AnthropicLLM(CustomComponent): class AnthropicLLM(CustomComponent):
display_name: str = "AnthropicLLM" display_name: str = "AnthropicLLM"
@ -60,7 +62,7 @@ class AnthropicLLM(CustomComponent):
try: try:
output = ChatAnthropic( output = ChatAnthropic(
model=model, model_name=model,
anthropic_api_key=anthropic_api_key, anthropic_api_key=anthropic_api_key,
max_tokens_to_sample=max_tokens, max_tokens_to_sample=max_tokens,
temperature=temperature, temperature=temperature,