Skip to content

Commit e513763

Browse files
author
harvey_xiang
committed
feat: add model log
1 parent 308cd6b commit e513763

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

src/memos/llms/openai.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -31,6 +31,9 @@ def __init__(self, config: OpenAILLMConfig):
3131
@timed_with_status(log_prefix="OpenAI LLM", log_args=["model_name_or_path"])
3232
def generate(self, messages: MessageList, **kwargs) -> str:
3333
"""Generate a response from OpenAI LLM, optionally overriding generation params."""
34+
logger.info(
35+
f"LLM Model: {self.config.model_name_or_path} {kwargs.get('model_name_or_path')}"
36+
)
3437
response = self.client.chat.completions.create(
3538
model=kwargs.get("model_name_or_path", self.config.model_name_or_path),
3639
messages=messages,

0 commit comments

Comments
 (0)