We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 308cd6b commit e513763Copy full SHA for e513763
src/memos/llms/openai.py
@@ -31,6 +31,9 @@ def __init__(self, config: OpenAILLMConfig):
31
@timed_with_status(log_prefix="OpenAI LLM", log_args=["model_name_or_path"])
32
def generate(self, messages: MessageList, **kwargs) -> str:
33
"""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
+ )
37
response = self.client.chat.completions.create(
38
model=kwargs.get("model_name_or_path", self.config.model_name_or_path),
39
messages=messages,
0 commit comments