diff --git a/rlm/clients/openai.py b/rlm/clients/openai.py index 48e8ed37c..b84ffe1e5 100644 --- a/rlm/clients/openai.py +++ b/rlm/clients/openai.py @@ -84,7 +84,7 @@ async def acompletion( raise ValueError("Model name is required for OpenAI client.") extra_body = {} - if self.base_url == DEFAULT_PRIME_INTELLECT_BASE_URL: + if self.client.base_url == DEFAULT_PRIME_INTELLECT_BASE_URL: extra_body["usage"] = {"include": True} response = await self.async_client.chat.completions.create(