Fix lint

llmclient/llms.py

@@ -861,7 +861,7 @@ async def call_multiple(
         tool_choice: Tool | str | None = TOOL_CHOICE_REQUIRED,
         **chat_kwargs,
     ) -> list[LLMResult]:
-        if chat_kwargs.get("n", 1) == 1 or self.config.get("n", 1) == 1:
+        if 1 in {chat_kwargs.get("n", 1), self.config.get("n", 1)}:
             if (
                 chat_kwargs.get("n")
                 and self.config.get("n")

tests/test_llms.py

@@ -332,7 +332,8 @@ async def test_single_completion(self, model_name: str) -> None:
         result = await model.call(messages)
         assert isinstance(result, LLMResult)
 
-        result = await model.call(messages, n=1)
+        result = await model.call(messages, n=1)  # noqa: FURB120
+
         assert isinstance(result, LLMResult)
         assert result.messages
         assert len(result.messages) == 1
@@ -358,7 +359,7 @@ async def test_multiple_completion(self, model_name: str, request) -> None:
             with pytest.raises(litellm.BadRequestError, match="anthropic"):
                 await model.call(messages)
         else:
-            results = await model.call(messages, n=None)
+            results = await model.call(messages, n=None)  # noqa: FURB120
             assert len(results) == self.NUM_COMPLETIONS
 
             results = await model.call(messages, n=self.NUM_COMPLETIONS)