File tree Expand file tree Collapse file tree 2 files changed +7
-3
lines changed
Expand file tree Collapse file tree 2 files changed +7
-3
lines changed Original file line number Diff line number Diff 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 )
Original file line number Diff line number Diff line change 11from agentlab .llm .chat_api import (
2+ AnthropicModelArgs ,
23 AzureModelArgs ,
34 OpenAIModelArgs ,
45 OpenRouterModelArgs ,
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" ,
You can’t perform that action at this time.
0 commit comments