diff --git a/opencompass/models/claude_sdk_api.py b/opencompass/models/claude_sdk_api.py index 173047f15..23ef4939f 100644 --- a/opencompass/models/claude_sdk_api.py +++ b/opencompass/models/claude_sdk_api.py @@ -35,6 +35,7 @@ def __init__( temperature: Optional[float] = 0.0, thinking: Optional[Dict] = None, retry: int = 2, + extra_headers: Optional[Dict] = None, ): super().__init__(path=path, max_seq_len=max_seq_len, @@ -47,7 +48,7 @@ def __init__( raise ImportError('Import anthropic failed. Please install it ' 'with "pip install anthropic" and try again.') - self.anthropic = Anthropic(api_key=key) + self.anthropic = Anthropic(api_key=key, default_headers=extra_headers or {}) self.model = path self.temperature = temperature self.thinking = thinking