From 6fa1eec71c566fea153d43e3fed131de02672545 Mon Sep 17 00:00:00 2001 From: Algo Kamishiro Date: Sat, 24 May 2025 07:04:00 +0900 Subject: [PATCH] =?UTF-8?q?Anthropic=E3=81=AEAPI=E3=82=AD=E3=83=BC?= =?UTF-8?q?=E6=B8=A1=E3=81=97=E7=AD=89=E3=82=B3=E3=83=BC=E3=83=89=E4=BF=AE?= =?UTF-8?q?=E6=AD=A3?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- llm_node.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/llm_node.py b/llm_node.py index 5f32d28..f828ca2 100644 --- a/llm_node.py +++ b/llm_node.py @@ -6,7 +6,7 @@ class UniversalLLMNode: return { "required": { "provider": (["openai", "anthropic", "google", "groq", "mistral"],), - "model": ("STRING", {"multiline": False}), + "model": ("STRING", {"default": "claude-3-sonnet-20240229"}), "prompt": ("STRING", {"multiline": True}), "max_tokens": ("INT", {"default": 300, "min": 50, "max": 4096}), } @@ -30,11 +30,12 @@ class UniversalLLMNode: elif provider == "anthropic": import anthropic - client = anthropic.Anthropic(api_key=os.getenv("ANTHROPIC_API_KEY")) + api_key = os.getenv("ANTHROPIC_API_KEY") + client = anthropic.Anthropic(api_key=api_key) completion = client.messages.create( model=model, max_tokens=max_tokens, - messages=[{"role": "user", "content": prompt}], + messages=[{"role": "user", "content": prompt}] ) return (completion.content[0].text,)