diff --git a/lib/sycamore/sycamore/llms/anthropic.py b/lib/sycamore/sycamore/llms/anthropic.py index 217396b26..ee48b3e50 100644 --- a/lib/sycamore/sycamore/llms/anthropic.py +++ b/lib/sycamore/sycamore/llms/anthropic.py @@ -235,7 +235,7 @@ def generate_batch(self, *, prompts: list[RenderedPrompt], llm_kwargs: Optional[ kwargs = get_generate_kwargs(p, llm_kwargs) kwargs["model"] = self.model.value kwargs["max_tokens"] = kwargs.get("max_tokens", 1024) - mparams = MessageCreateParamsNonStreaming(**kwargs) + mparams = MessageCreateParamsNonStreaming(**kwargs) # type: ignore rq = Request(custom_id=str(i), params=mparams) calls.append(rq)