diff --git a/rdagent/oai/backend/base.py b/rdagent/oai/backend/base.py index c15691f4..7a632165 100644 --- a/rdagent/oai/backend/base.py +++ b/rdagent/oai/backend/base.py @@ -242,6 +242,17 @@ class APIBackend(ABC): ) return messages + def _build_log_messages(self, messages: list[dict[str, Any]]) -> str: + log_messages = "" + for m in messages: + log_messages += ( + f"\n{LogColors.MAGENTA}{LogColors.BOLD}Role:{LogColors.END}" + f"{LogColors.CYAN}{m['role']}{LogColors.END}\n" + f"{LogColors.MAGENTA}{LogColors.BOLD}Content:{LogColors.END} " + f"{LogColors.CYAN}{m['content']}{LogColors.END}\n" + ) + return log_messages + def build_messages_and_create_chat_completion( # type: ignore[no-untyped-def] self, user_prompt: str, diff --git a/rdagent/oai/backend/deprec.py b/rdagent/oai/backend/deprec.py index 3011385f..a96d8814 100644 --- a/rdagent/oai/backend/deprec.py +++ b/rdagent/oai/backend/deprec.py @@ -283,17 +283,6 @@ class DeprecBackend(APIBackend): return [content_to_embedding_dict[content] for content in input_content_list] - def _build_log_messages(self, messages: list[dict[str, Any]]) -> str: - log_messages = "" - for m in messages: - log_messages += ( - f"\n{LogColors.MAGENTA}{LogColors.BOLD}Role:{LogColors.END}" - f"{LogColors.CYAN}{m['role']}{LogColors.END}\n" - f"{LogColors.MAGENTA}{LogColors.BOLD}Content:{LogColors.END} " - f"{LogColors.CYAN}{m['content']}{LogColors.END}\n" - ) - return log_messages - def _create_chat_completion_inner_function( # type: ignore[no-untyped-def] # noqa: C901, PLR0912, PLR0915 self, messages: list[dict[str, Any]], diff --git a/rdagent/oai/backend/litellm.py b/rdagent/oai/backend/litellm.py index c13f976e..86dd9719 100644 --- a/rdagent/oai/backend/litellm.py +++ b/rdagent/oai/backend/litellm.py @@ -85,6 +85,7 @@ class LiteLLMAPIBackend(APIBackend): f"{LogColors.GREEN}Using chat model{LogColors.END} {LITELLM_SETTINGS.chat_model}", tag="llm_messages" ) + logger.info(self._build_log_messages(messages), tag="llm_messages") if LITELLM_SETTINGS.chat_stream: logger.info(f"{LogColors.BLUE}assistant:{LogColors.END}", tag="llm_messages") content = ""