Update AnthropicLLM model name
This commit is contained in:
parent
a632fc6bdb
commit
1c2d2402ba
1 changed files with 4 additions and 2 deletions
|
|
@ -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,
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue