diff --git a/src/memos/llms/openai.py b/src/memos/llms/openai.py index c45038e9d..f4ebf45c7 100644 --- a/src/memos/llms/openai.py +++ b/src/memos/llms/openai.py @@ -55,7 +55,7 @@ def generate(self, messages: MessageList, **kwargs) -> str: return reasoning_content + response_content return response_content - @timed(log=True, log_prefix="OpenAI LLM") + @timed(log=True, log_prefix="OpenAI LLM", log_args=["model_name_or_path"]) def generate_stream(self, messages: MessageList, **kwargs) -> Generator[str, None, None]: """Stream response from OpenAI LLM with optional reasoning support.""" if kwargs.get("tools"):