re-run ruff (ill add a hook for this later)

Author: alexzhang13

rlm/clients/azure_openai.py

@@ -66,9 +66,7 @@ def __init__(
         self.model_output_tokens: dict[str, int] = defaultdict(int)
         self.model_total_tokens: dict[str, int] = defaultdict(int)
 
-    def completion(
-        self, prompt: str | list[dict[str, Any]], model: str | None = None
-    ) -> str:
+    def completion(self, prompt: str | list[dict[str, Any]], model: str | None = None) -> str:
         if isinstance(prompt, str):
             messages = [{"role": "user", "content": prompt}]
         elif isinstance(prompt, list) and all(isinstance(item, dict) for item in prompt):
@@ -139,4 +137,3 @@ def get_last_usage(self) -> ModelUsageSummary:
             total_input_tokens=self.last_prompt_tokens,
             total_output_tokens=self.last_completion_tokens,
         )
-

rlm/core/types.py

@@ -2,7 +2,9 @@
 from types import ModuleType
 from typing import Any, Literal
 
-ClientBackend = Literal["openai", "portkey", "openrouter", "vllm", "litellm", "anthropic", "azure_openai"]
+ClientBackend = Literal[
+    "openai", "portkey", "openrouter", "vllm", "litellm", "anthropic", "azure_openai"
+]
 EnvironmentType = Literal["local", "prime", "modal"]