Update AnthropicLLM model name

This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-12-04 11:39:14 -03:00
commit 1c2d2402ba

View file

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