diff --git a/src/backend/langflow/components/llms/AnthropicLLM.py b/src/backend/langflow/components/llms/AnthropicLLM.py index ef7cced37..ff6a24a0e 100644 --- a/src/backend/langflow/components/llms/AnthropicLLM.py +++ b/src/backend/langflow/components/llms/AnthropicLLM.py @@ -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,