Skip to content

Commit b1bad6e

Browse files
committed
adding anthropic chat model
1 parent c32400f commit b1bad6e

File tree

2 files changed

+7
-3
lines changed

2 files changed

+7
-3
lines changed

src/agentlab/llm/chat_api.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -482,13 +482,11 @@ def __init__(
482482
temperature=0.5,
483483
max_tokens=100,
484484
max_retry=4,
485-
log_probs=False,
486485
):
487486
self.model_name = model_name
488487
self.temperature = temperature
489488
self.max_tokens = max_tokens
490489
self.max_retry = max_retry
491-
self.log_probs = log_probs
492490

493491
api_key = api_key or os.getenv("ANTHROPIC_API_KEY")
494492
self.client = anthropic.Anthropic(api_key=api_key)
@@ -544,5 +542,4 @@ def make_model(self):
544542
model_name=self.model_name,
545543
temperature=self.temperature,
546544
max_tokens=self.max_new_tokens,
547-
log_probs=self.log_probs,
548545
)

src/agentlab/llm/llm_configs.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
from agentlab.llm.chat_api import (
2+
AnthropicModelArgs,
23
AzureModelArgs,
34
OpenAIModelArgs,
45
OpenRouterModelArgs,
@@ -115,6 +116,12 @@
115116
max_new_tokens=16_384,
116117
vision_support=True,
117118
),
119+
# ---------------- Anthropic ----------------#
120+
"anthropic/claude-3-7-sonnet-20250219": AnthropicModelArgs(
121+
model_name="claude-3-7-sonnet-20250219",
122+
max_new_tokens=16_384,
123+
temperature=1e-1,
124+
),
118125
# ---------------- OSS LLMs ----------------#
119126
"meta-llama/Meta-Llama-3-70B-Instruct": SelfHostedModelArgs(
120127
model_name="meta-llama/Meta-Llama-3-70B-Instruct",

0 commit comments

Comments
 (0)