From 1c2d2402ba444862e5a30c39ddb71250361803fe Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Mon, 4 Dec 2023 11:39:14 -0300
Subject: [PATCH] Update AnthropicLLM model name

---
 src/backend/langflow/components/llms/AnthropicLLM.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

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,